pum 1.2.1__tar.gz → 1.2.3__tar.gz

{pum-1.2.1 → pum-1.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.2.1
+Version: 1.2.3
 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.2.1 → pum-1.2.3}/pum/cli.py

@@ -279,6 +279,11 @@ def create_parser() -> argparse.ArgumentParser:
     parser_baseline.add_argument(
         "-b", "--baseline", help="Set baseline in the format x.x.x", required=True
     )
+    parser_baseline.add_argument(
+        "--create-table",
+        help="Create the pum_migrations table if it does not exist",
+        action="store_true",
+    )
 
     # Parser for the "upgrade" command
     parser_upgrade = subparsers.add_parser("upgrade", help="upgrade db")
@@ -307,10 +312,13 @@ def cli() -> int:  # noqa: PLR0912
         parser.print_help()
         parser.exit()
 
+    validate = args.command not in ("info", "baseline")
     if args.config_file:
-        config = PumConfig.from_yaml(args.config_file, install_dependencies=True)
+        config = PumConfig.from_yaml(args.config_file, validate=validate, install_dependencies=True)
     else:
-        config = PumConfig.from_yaml(Path(args.dir) / ".pum.yaml", install_dependencies=True)
+        config = PumConfig.from_yaml(
+            Path(args.dir) / ".pum.yaml", validate=validate, install_dependencies=True
+        )
 
     with psycopg.connect(f"service={args.pg_service}") as conn:
         # Check if the connection is successful
@@ -391,6 +399,17 @@ def cli() -> int:  # noqa: PLR0912
         elif args.command == "restore":
             pum.run_restore(args.pg_service, args.file, args.x, args.exclude_schema)
         elif args.command == "baseline":
+            sm = SchemaMigrations(config=config)
+            if not sm.exists(connection=conn):
+                if args.create_table:
+                    sm.create(connection=conn)
+                    logger.info("Created pum_migrations table.")
+                else:
+                    logger.error(
+                        "pum_migrations table does not exist. Use --create-table to create it."
+                    )
+                    exit_code = 1
+                    return exit_code
             SchemaMigrations(config=config).set_baseline(connection=conn, version=args.baseline)
 
         elif args.command == "upgrade":

{pum-1.2.1 → pum-1.2.3}/pum/config_model.py

@@ -72,7 +72,7 @@ class PumModel(PumCustomBaseModel):
     PumModel holds some PUM specifics.
 
     Attributes:
-        migration_table_schema: Name of schema for the migration table.
+        migration_table_schema: Name of schema for the migration table. The table will always be named `pum_migrations`.
         minimum_version: Minimum required version of PUM.
     """
 

{pum-1.2.1 → pum-1.2.3}/pum/pum_config.py

@@ -231,7 +231,7 @@ class PumConfig:
         """Return a RoleManager instance based on the roles defined in the configuration."""
         if not self.config.roles:
             logger.warning("No roles defined in the configuration. Returning an empty RoleManager.")
-            return RoleManager()
+            return RoleManager([])
         return RoleManager([role.model_dump() for role in self.config.roles])
 
     def pre_hook_handlers(self) -> list[HookHandler]:

{pum-1.2.1 → pum-1.2.3}/pum/upgrader.py

@@ -133,6 +133,7 @@ class Upgrader:
         name: str,
         *,
         parameters: dict | None = None,
+        grant: bool = True,
     ) -> None:
         """Install demo data for the module.
 
@@ -140,6 +141,7 @@ class Upgrader:
             connection: The database connection to use.
             name: The name of the demo data to install.
             parameters: The parameters to pass to the demo data SQL.
+            grant: If True, grant permissions to the roles after installing the demo data. Default is True.
         """
         if name not in self.config.demo_data():
             raise PumException(f"Demo data '{name}' not found in the configuration.")
@@ -167,4 +169,9 @@ class Upgrader:
 
         connection.commit()
 
+        if grant:
+            self.config.role_manager().grant_permissions(connection=connection, commit=False)
+
+        connection.commit()
+
         logger.info("Demo data '%s' installed successfully.", name)

{pum-1.2.1 → pum-1.2.3}/pum.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.2.1
+Version: 1.2.3
 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