pum 1.1.6__tar.gz → 1.1.8__tar.gz

{pum-1.1.6/pum.egg-info → pum-1.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.1.6
+Version: 1.1.8
 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.6 → pum-1.1.8}/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.6 → pum-1.1.8}/pum/parameter.py

@@ -1,5 +1,4 @@
 from enum import Enum
-import psycopg
 
 
 class ParameterType(Enum):
@@ -55,7 +54,7 @@ class ParameterDefinition:
                 raise ValueError(f"Parameter '{name}' has an invalid type: {type}. ") from None
         else:
             raise TypeError("type must be a str or ParameterType")
-        self.default = psycopg.sql.Literal(default)
+        self.default = default
         self.description = description
 
     def __repr__(self) -> str:

{pum-1.1.6 → pum-1.1.8}/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.6 → pum-1.1.8/pum.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.1.6
+Version: 1.1.8
 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.6 → pum-1.1.8}/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)",