From 3861fc0114dcad4ef7133094f5bb85e6c61002f8 Mon Sep 17 00:00:00 2001
From: sanj <67624670+iodrift@users.noreply.github.com>
Date: Tue, 30 Jul 2024 16:51:19 -0700
Subject: [PATCH] Auto-update: Tue Jul 30 16:51:19 PDT 2024

---
 sijapi/classes.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/sijapi/classes.py b/sijapi/classes.py
index 0ba16b3..f82794c 100644
--- a/sijapi/classes.py
+++ b/sijapi/classes.py
@@ -637,8 +637,11 @@ class APIConfig(BaseModel):
                     for table in tables:
                         table_name = table['tablename']
                         try:
+                            debug(f"Processing table: {table_name}")
                             has_primary_key = await self.ensure_sync_columns(dest_conn, table_name)
+                            debug(f"Table {table_name} has primary key: {has_primary_key}")
                             last_synced_version = await self.get_last_synced_version(dest_conn, table_name, source_id)
+                            debug(f"Last synced version for {table_name}: {last_synced_version}")
                             
                             changes = await source_conn.fetch(f"""
                                 SELECT * FROM "{table_name}"
@@ -647,6 +650,8 @@ class APIConfig(BaseModel):
                                 LIMIT $3
                             """, last_synced_version, source_id, batch_size)
                             
+                            debug(f"Number of changes for {table_name}: {len(changes)}")
+                            
                             if changes:
                                 changes_count = await self.apply_batch_changes(dest_conn, table_name, changes, has_primary_key)
                                 total_changes += changes_count
@@ -712,12 +717,15 @@ class APIConfig(BaseModel):
 
             debug(f"Generated insert query for {table_name}: {insert_query}")
 
+            # Prepare the statement
+            stmt = await conn.prepare(insert_query)
+
             affected_rows = 0
             for change in tqdm(changes, desc=f"Syncing {table_name}", unit="row"):
                 values = [change[col] for col in columns]
                 debug(f"Executing query for {table_name} with values: {values}")
-                result = await conn.execute(insert_query, *values)
-                affected_rows += int(result.split()[-1])
+                result = await stmt.execute(*values)
+                affected_rows += 1  # Since we're executing one at a time, each successful execution affects one row
 
             return affected_rows
 
@@ -728,6 +736,7 @@ class APIConfig(BaseModel):
 
 
 
+
     async def sync_spatial_ref_sys(self, source_conn, dest_conn):
         try:
             # Get all entries from the source