sij/sijapi
1
1
Fork 0
Code Issues 3 Pull requests Projects Releases Packages Wiki Activity Actions

Auto-update: Mon Jul 29 18:22:58 PDT 2024

Browse source
This commit is contained in:
sanj 2024-07-29 18:22:58 -07:00
parent 7dcb71d6ad
commit 2756be327f
1 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
sijapi/classes.py
View file

@ -570,12 +570,16 @@ class APIConfig(BaseModel):
for change in changes: for change in changes:
columns = change.keys() columns = change.keys()
values = [change[col] for col in columns] values = [change[col] for col in columns]
await remote_conn.execute(f""" placeholders = [f'${i+1}' for i in range(len(columns))]
insert_query = f"""
INSERT INTO "{table_name}" ({', '.join(columns)}) INSERT INTO "{table_name}" ({', '.join(columns)})
VALUES ({', '.join(f'${i+1}' for i in range(len(columns)))}) VALUES ({', '.join(placeholders)})
ON CONFLICT (id) DO UPDATE SET ON CONFLICT (id) DO UPDATE SET
{', '.join(f"{col} = EXCLUDED.{col}" for col in columns if col != 'id')} {', '.join(f"{col} = EXCLUDED.{col}" for col in columns if col != 'id')}
""", *values) """
await remote_conn.execute(insert_query, *values)
if changes: if changes:
await self.update_sync_status(table_name, pool_entry['ts_id'], changes[-1]['version']) await self.update_sync_status(table_name, pool_entry['ts_id'], changes[-1]['version'])
@ -583,13 +587,16 @@ class APIConfig(BaseModel):
info(f"Successfully pushed changes to {pool_entry['ts_id']}") info(f"Successfully pushed changes to {pool_entry['ts_id']}")
except Exception as e: except Exception as e:
err(f"Error pushing changes to {pool_entry['ts_id']}: {str(e)}") err(f"Error pushing changes to {pool_entry['ts_id']}: {str(e)}")
err(f"Traceback: {traceback.format_exc()}")
async def get_last_synced_version(self, table_name, server_id): async def get_last_synced_version(self, table_name, server_id):
async with self.get_connection() as conn: async with self.get_connection() as conn:
return await conn.fetchval(""" result = await conn.fetchval("""
SELECT last_synced_version FROM sync_status SELECT last_synced_version FROM sync_status
WHERE table_name = $1 AND server_id = $2 WHERE table_name = $1 AND server_id = $2
""", table_name, server_id) or 0 """, table_name, server_id)
return result if result is not None else 0
async def update_sync_status(self, table_name, server_id, version): async def update_sync_status(self, table_name, server_id, version):