pgbelt 0.6.1__py3-none-any.whl → 0.6.2__py3-none-any.whl

pgbelt/util/pglogical.py

@@ -97,17 +97,25 @@ async def configure_replication_set(
     """
     Add each table in the given list to the default replication set
     """
-    logger.info(f"Configuring default replication set with tables: {tables}")
+    logger.info("Creating new replication set 'pgbelt'")
+    async with pool.acquire() as conn:
+        try:
+            await conn.execute("SELECT pglogical.create_replication_set('pgbelt');")
+            logger.debug("Created the 'pgbelt' replication set")
+        except Exception as e:
+            logger.debug(f"Could not create replication set 'pgbelt': {e}")
+
+    logger.info(f"Configuring 'pgbelt' replication set with tables: {tables}")
     for table in tables:
         async with pool.acquire() as conn:
             async with conn.transaction():
                 try:
                     await conn.execute(
-                        f"SELECT pglogical.replication_set_add_table('default', '\"{table}\"');"
+                        f"SELECT pglogical.replication_set_add_table('pgbelt', '\"{table}\"');"
                     )
-                    logger.debug(f"{table} added to default replication set")
+                    logger.debug(f"Table '{table}' added to 'pgbelt' replication set")
                 except UniqueViolationError:
-                    logger.debug(f"{table} already in default replication set")
+                    logger.debug(f"Table '{table}' already in 'pgbelt' replication set")
 
 
 async def configure_node(pool: Pool, name: str, dsn: str, logger: Logger) -> None:
@@ -145,6 +153,7 @@ async def configure_subscription(
                 await conn.execute(
                     f"""SELECT pglogical.create_subscription(
                         subscription_name:='{name}',
+                        replication_sets:='{{pgbelt}}',
                         provider_dsn:='{provider_dsn}',
                         synchronize_structure:=false,
                         synchronize_data:={'true' if name.startswith('pg1') else 'false'},
@@ -197,8 +206,8 @@ async def teardown_replication_set(pool: Pool, logger: Logger) -> None:
     async with pool.acquire() as conn:
         async with conn.transaction():
             try:
-                await conn.execute("SELECT pglogical.drop_replication_set('default');")
-                logger.debug("Replication set 'default' dropped")
+                await conn.execute("SELECT pglogical.drop_replication_set('pgbelt');")
+                logger.debug("Replication set 'pgbelt' dropped")
             except (
                 InvalidSchemaNameError,
                 UndefinedFunctionError,

{pgbelt-0.6.1.dist-info → pgbelt-0.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pgbelt
-Version: 0.6.1
+Version: 0.6.2
 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.6.1.dist-info → pgbelt-0.6.2.dist-info}/RECORD

@@ -18,10 +18,10 @@ pgbelt/util/__init__.py,sha256=-6KkvVMz-yGNQfeoo4CZZrgWKXYmFd4CMyoiao8OnFE,40
 pgbelt/util/asyncfuncs.py,sha256=7i_GpBmUNNZ8RUGvU-q5nclsoaCm6Lx8jLP8usYvmZc,583
 pgbelt/util/dump.py,sha256=p6M26s3FXnSCTmee4v7X0-azYvmcVz6DkZf8J8YxU0s,14357
 pgbelt/util/logs.py,sha256=l2jT-WKZ-33eNDw4S4W1_eE4ISo4rtDRXYLVf4QTV4Y,1699
-pgbelt/util/pglogical.py,sha256=JwIKI9pP8WDLa_32-2jnQuzyGC4oSeFYo0iP7ljAJvI,12620
+pgbelt/util/pglogical.py,sha256=URi4L_4PWYIz0iWw7HcXNfjzvWXPjQCm71cAOqnt3Qk,13059
 pgbelt/util/postgres.py,sha256=9AgnmHdYpWVypyBNTEQEpd4COoa1JEQXoELOR8-HrzQ,13668
-pgbelt-0.6.1.dist-info/LICENSE,sha256=FQ5cFkW02dKK3LmKH8z-rwn93tWSCh7lsxfNUiWcFsg,10758
-pgbelt-0.6.1.dist-info/METADATA,sha256=sRQAbecXNJwIUJJDjNu762VtktWR90HIQNmMBMokPag,2826
-pgbelt-0.6.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-pgbelt-0.6.1.dist-info/entry_points.txt,sha256=SCz_poPjkaVnWpJ-CeytAnDzbVc6l0WalOwitIqW_3g,40
-pgbelt-0.6.1.dist-info/RECORD,,
+pgbelt-0.6.2.dist-info/LICENSE,sha256=FQ5cFkW02dKK3LmKH8z-rwn93tWSCh7lsxfNUiWcFsg,10758
+pgbelt-0.6.2.dist-info/METADATA,sha256=7bEElDGURHPEE3cVWbF19aPGhAXNQ0_YxPRpdZK_Fc8,2826
+pgbelt-0.6.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+pgbelt-0.6.2.dist-info/entry_points.txt,sha256=SCz_poPjkaVnWpJ-CeytAnDzbVc6l0WalOwitIqW_3g,40
+pgbelt-0.6.2.dist-info/RECORD,,