fix StrEnum in integration tests; Leave windows on 3.10

.github/workflows/build_and_test.yml

@@ -109,15 +109,15 @@ jobs:
  runs-on: windows-latest
  strategy:
  matrix:
- python-version: ["3.11"]
+ python-version: ["3.10"]
  needs: check-code-quality
  steps:
  - uses: actions/checkout@v3
  - name: Set up Python ${{ matrix.python-version }}
  uses: actions/setup-python@v3
  with:
  python-version: ${{ matrix.python-version }}
- # Run Playwright web UI tests only run once per workflow (Python 3.11).
+ # Run Playwright web UI tests only run once per workflow (Python 3.10).
  - name: Set up Node
  uses: actions/setup-node@v3
  with:

mage_integrations/mage_integrations/connections/mysql/__init__.py

@@ -6,8 +6,8 @@
 from mysql.connector import connect
 from sshtunnel import SSHTunnelForwarder
 
-from mage_ai.shared.enum import StrEnum
 from mage_integrations.connections.sql.base import Connection
+from mage_integrations.utils.enum import StrEnum
 
 
 class ConnectionMethod(StrEnum):

mage_integrations/mage_integrations/sources/couchbase/__init__.py

@@ -2,7 +2,6 @@
 
 from singer import catalog
 
-from mage_ai.shared.enum import StrEnum
 from mage_integrations.connections.couchbase import Couchbase as CouchbaseConnection
 from mage_integrations.sources.base import main
 from mage_integrations.sources.catalog import Catalog
@@ -16,6 +15,7 @@
  wrap_column_in_quotes,
 )
 from mage_integrations.sources.sql.base import Source
+from mage_integrations.utils.enum import StrEnum
 
 DEFAULT_COLUMN_NAME = '_document'
 

mage_integrations/mage_integrations/sources/sql/constants.py

@@ -1,4 +1,4 @@
-from mage_ai.shared.enum import StrEnum
+from mage_integrations.utils.enum import StrEnum
 
 
 class PredicateOperator(StrEnum):

mage_integrations/mage_integrations/utils/enum.py

@@ -0,0 +1,12 @@
+try:
+ from enum import IntEnum, StrEnum
+except ImportError:
+ from enum import Enum, IntEnum
+
+ class StrEnum(str, Enum):
+ pass
+
+__all__ = [
+ "IntEnum",
+ "StrEnum",
+]