pgbelt 0.6.1__tar.gz → 0.6.2__tar.gz

{pgbelt-0.6.1 → pgbelt-0.6.2}/PKG-INFO

@@ -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 → pgbelt-0.6.2}/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 → pgbelt-0.6.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pgbelt"
-version = "0.6.1"
+version = "0.6.2"
 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"
@@ -18,7 +18,7 @@ tabulate = "^0.9.0"
 typer = "^0.9.0"
 
 [tool.poetry.dev-dependencies]
-black = "~23.12.1"
+black = "~24.1.1"
 pre-commit = "~3.6.0"
 flake8 = "^7.0.0"
 pytest-cov = "~4.1.0"
@@ -26,7 +26,7 @@ pytest = "^7.4.4"
 coverage = {extras = ["toml"], version = "^7.4"}
 safety = "^2.3.1"
 mypy = "^1.8"
-xdoctest = {extras = ["colors"], version = "^1.1.2"}
+xdoctest = {extras = ["colors"], version = "^1.1.3"}
 flake8-bandit = "~4.1.1"
 flake8-bugbear = ">=21.9.2"
 flake8-docstrings = "^1.6.0"
@@ -38,7 +38,7 @@ pre-commit-hooks = "^4.5.0"
 Pygments = "^2.17.2"
 pyupgrade = "^3.15.0"
 pylint = "^3.0.3"
-pytest-asyncio = "~0.23.3"
+pytest-asyncio = "~0.23.4"
 
 [build-system]
 requires = ["poetry-core>=1.0.0", "setuptools"]