Merge pull request #2 from Widen/row-keys

locked sqlalchemy version
Widen · Jan 19, 2024 · 29ac603 · 29ac603
2 parents d1dee00 + f61ade4
commit 29ac603
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 3 deletions.
diff --git a/meltano.yml b/meltano.yml
@@ -35,7 +35,7 @@ plugins:
           - db_schema1
         sql: SELECT * FROM db_schema1.table1 LIMIT 5
     select:
-      - '*-storage_snapshot.*'
+      - '*-example_query_name.*'
   loaders:
   - name: target-jsonl
     variant: andyh1203

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,7 @@ python = "<3.12,>=3.7.1"
 singer-sdk = { version="^0.34.0" }
 fs-s3fs = { version = "^1.1.1", optional = true }
 pymysql = "^1.0.3"
+sqlalchemy = "^2.0.25"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.2.1"

diff --git a/tap_mysql/client.py b/tap_mysql/client.py
@@ -261,7 +261,7 @@ def discover_catalog_entries(self) -> list[dict]:
                         custom_result = connection.execute(query)
                         custom_rec = custom_result.fetchone()
                         # inject the table_schema into the list of columns
-                        custom_rec_keys = list(custom_rec.keys()) + ["mysql_schema"]
+                        custom_rec_keys = list(custom_rec._fields) + ["mysql_schema"]
 
                         # note that all columns are forced to be strings to avoid
                         # the complexity of inferring their data types. Warning this