pum 1.1.7__tar.gz → 1.1.9__tar.gz

{pum-1.1.7/pum.egg-info → pum-1.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.1.7
+Version: 1.1.9
 Summary: Pum stands for "Postgres Upgrades Manager". It is a Database migration management tool very similar to flyway-db or Liquibase, based on metadata tables.
 Author-email: Denis Rouzaud <denis@opengis.ch>
 License-Expression: GPL-2.0-or-later

{pum-1.1.7 → pum-1.1.9}/pum/cli.py

@@ -201,6 +201,11 @@ def create_parser() -> argparse.ArgumentParser:
     parser_install.add_argument(
         "-d", "--demo-data", help="Load demo data with the given name", type=str, default=None
     )
+    parser_install.add_argument(
+        "--beta-testing",
+        help="This will install the module in beta testing, meaning that it will not be possible to receive any future updates.",
+        action="store_true",
+    )
 
     # Role management parser
     parser_role = subparsers.add_parser("role", help="manage roles in the database")
@@ -347,6 +352,7 @@ def cli() -> int:  # noqa: PLR0912
                 roles=args.roles,
                 grant=args.grant,
                 demo_data=args.demo_data,
+                beta_testing=args.beta_testing,
             )
             conn.commit()
             if args.demo_data:

{pum-1.1.7 → pum-1.1.9}/pum/info.py

@@ -3,9 +3,11 @@ import sys
 import psycopg
 
 from .pum_config import PumConfig
-from .schema_migrations import SchemaMigrations
+from .schema_migrations import SchemaMigrations, MIGRATION_TABLE_NAME
 
 logger = logging.getLogger(__name__)
+# set to info here
+logger.setLevel(logging.INFO)
 
 
 def run_info(connection: psycopg.Connection, config: PumConfig) -> None:
@@ -17,14 +19,20 @@ def run_info(connection: psycopg.Connection, config: PumConfig) -> None:
 
     """
     try:
-        schema_migrations = SchemaMigrations(connection, config)
-        if not schema_migrations.exists():
+        schema_migrations = SchemaMigrations(config=config)
+        if not schema_migrations.exists(connection=connection):
             logger.info(
-                f"No migrations found in {config.pum.migration_table_schema}.{config.pum.migration_table_name}."
+                f"No migrations found in {config.config.pum.migration_table_schema}.{MIGRATION_TABLE_NAME}."
             )
         else:
             # Add your logic for when migrations exist; for now, we simply print a message.
             logger.info("Migrations found.")
+            logger.info(
+                f"Schema migrations table: {config.config.pum.migration_table_schema}.{MIGRATION_TABLE_NAME}"
+            )
+            logger.info(
+                f"Version: {schema_migrations.migration_details(connection=connection)['version']}"
+            )
     except Exception:
         logger.exception("An error occurred while checking for migrations.")
         sys.exit(1)

{pum-1.1.7 → pum-1.1.9}/pum/upgrader.py

@@ -52,6 +52,7 @@ class Upgrader:
         roles: bool = False,
         grant: bool = False,
         demo_data: str | None = None,
+        beta_testing: bool = False,
         commit: bool = False,
     ) -> None:
         """Installs the given module
@@ -72,6 +73,10 @@ class Upgrader:
                 If True, permissions will be granted to the roles.
             demo_data:
                 The name of the demo data to load. If None, no demo data is loaded.
+            beta_testing:
+                If True, the module is installed in beta testing mode.
+                This means that the module will not be able to receive any future updates.
+                We strongly discourage using this for production.
             commit:
                 If True, the changes will be committed to the database.
         """
@@ -108,7 +113,7 @@ class Upgrader:
             self.schema_migrations.set_baseline(
                 connection=connection,
                 version=changelog.version,
-                beta_testing=False,
+                beta_testing=beta_testing,
                 commit=False,
                 changelog_files=changelog_files,
                 parameters=parameters,

{pum-1.1.7 → pum-1.1.9/pum.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.1.7
+Version: 1.1.9
 Summary: Pum stands for "Postgres Upgrades Manager". It is a Database migration management tool very similar to flyway-db or Liquibase, based on metadata tables.
 Author-email: Denis Rouzaud <denis@opengis.ch>
 License-Expression: GPL-2.0-or-later

{pum-1.1.7 → pum-1.1.9}/test/test_upgrader.py

@@ -63,11 +63,28 @@ class TestUpgrader(unittest.TestCase):
             self.assertEqual(sm.baseline(conn), "1.2.3")
             self.assertEqual(sm.migration_details(conn), sm.migration_details(conn, "1.2.3"))
             self.assertEqual(sm.migration_details(conn)["version"], "1.2.3")
+            self.assertEqual(sm.migration_details(conn)["beta_testing"], False)
             self.assertEqual(
                 sm.migration_details(conn)["changelog_files"],
                 [str(changelog_file)],
             )
 
+    def test_install_beta_testing(self) -> None:
+        """Test the installation as beta testing."""
+        test_dir = Path("test") / "data" / "single_changelog"
+        cfg = PumConfig(test_dir)
+        sm = SchemaMigrations(cfg)
+        with psycopg.connect(f"service={self.pg_service}") as conn:
+            self.assertFalse(sm.exists(conn))
+            upgrader = Upgrader(
+                config=cfg,
+            )
+            upgrader.install(connection=conn, beta_testing=True)
+            self.assertTrue(sm.exists(conn))
+            self.assertEqual(sm.baseline(conn), "1.2.3")
+            self.assertEqual(sm.migration_details(conn), sm.migration_details(conn, "1.2.3"))
+            self.assertEqual(sm.migration_details(conn)["beta_testing"], True)
+
     @unittest.skipIf(
         os.name == "nt" and os.getenv("CI") == "true",
         "Test not supported on Windows CI (postgis not installed)",