pgbelt 0.7.8__tar.gz → 0.7.9__tar.gz

{pgbelt-0.7.8 → pgbelt-0.7.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pgbelt
-Version: 0.7.8
+Version: 0.7.9
 Summary: A CLI tool used to manage Postgres data migrations from beginning to end, for a single database or a fleet, leveraging pglogical replication.
 Author: Varjitt Jeeva
 Author-email: varjitt.jeeva@autodesk.com

{pgbelt-0.7.8 → pgbelt-0.7.9}/pgbelt/util/postgres.py

@@ -106,17 +106,22 @@ async def compare_data(
         full_table_name = f'{schema}."{table}"'
 
         logger.debug(f"Validating table {full_table_name}...")
-        order_by_pkeys = ",".join(pkeys_dict[table])
 
-        src_rows = await src_pool.fetch(
-            query.format(table=full_table_name, order_by_pkeys=order_by_pkeys)
+        # Have to wrap each pkey in double quotes due to capitalization issues.
+        order_by_pkeys = ""
+        for pkey in pkeys_dict[table]:
+            order_by_pkeys += f'"{pkey}", '
+        order_by_pkeys = order_by_pkeys[:-2]
+
+        filled_query = query.format(
+            table=full_table_name, order_by_pkeys=order_by_pkeys
         )
 
+        src_rows = await src_pool.fetch(filled_query)
+
         # There is a chance tables are empty...
         if len(src_rows) == 0:
-            dst_rows = await dst_pool.fetch(
-                query.format(table=full_table_name, order_by_pkeys=order_by_pkeys)
-            )
+            dst_rows = await dst_pool.fetch(filled_query)
             if len(dst_rows) != 0:
                 raise AssertionError(
                     f"Table {full_table_name} has 0 rows in source but nonzero rows in target... Big problem. Please investigate."
@@ -144,7 +149,7 @@ async def compare_data(
         dst_query = f"SELECT * FROM {full_table_name} WHERE "
 
         for k, v in pkey_vals_dict.items():
-            dst_query = dst_query + f"{k} IN ({v}) AND "
+            dst_query = dst_query + f'"{k}" IN ({v}) AND '
 
         # SELECT * FROM <table> WHERE <pkey1> IN (1,2,3,4,5,6...) AND <pkey2> IN ('a','b','c',...);
         comparison_query = dst_query[:-5] + f" ORDER BY {order_by_pkeys};"

{pgbelt-0.7.8 → pgbelt-0.7.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pgbelt"
-version = "0.7.8"
+version = "0.7.9"
 description = "A CLI tool used to manage Postgres data migrations from beginning to end, for a single database or a fleet, leveraging pglogical replication."
 authors = ["Varjitt Jeeva <varjitt.jeeva@autodesk.com>"]
 readme = "README.md"