pum 1.1.18__tar.gz → 1.2.0__tar.gz

{pum-1.1.18 → pum-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.1.18
+Version: 1.2.0
 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.18 → pum-1.2.0}/pum/config_model.py

@@ -132,11 +132,22 @@ class DemoDataModel(PumCustomBaseModel):
     DemoDataModel represents a configuration for demo data.
 
     Attributes:
-        files: Optional list of named demo data files.
+        name: Name of the demo data.
+        file: Optional path to a single demo data file.
+        files: Optional list of paths to multiple demo data files.
     """
 
     name: str = Field(..., description="Name of the demo data.")
-    file: str = Field(..., description="Path to the demo data file.")
+
+    file: Optional[str] = None
+    files: Optional[List[str]] = None
+
+    @model_validator(mode="after")
+    def validate_args(self):
+        file, files = self.file, self.files
+        if (file and files) or (not file and not files):
+            raise PumConfigError("Exactly one of 'file' or 'files' must be set in a demo data set.")
+        return self
 
 
 class DependencyModel(PumCustomBaseModel):

{pum-1.1.18 → pum-1.2.0}/pum/pum_config.py

@@ -256,9 +256,12 @@ class PumConfig:
             else []
         )
 
-    def demo_data(self) -> dict[str, str]:
+    def demo_data(self) -> dict[str, list[str]]:
         """Return a dictionary of demo data files defined in the configuration."""
-        return {dm.name: dm.file for dm in self.config.demo_data}
+        demo_data_files = {}
+        for dm in self.config.demo_data:
+            demo_data_files[dm.name] = dm.files or [dm.file]
+        return demo_data_files
 
     def __del__(self):
         # Cleanup temporary directories and sys.path modifications

{pum-1.1.18 → pum-1.2.0}/pum/schema_migrations.py

@@ -177,9 +177,9 @@ class SchemaMigrations:
         """
         if isinstance(version, packaging.version.Version):
             version = str(version)
-        pattern = re.compile(r"^\d+\.\d+\.\d+$")
+        pattern = re.compile(r"^\d+\.\d+(\.\d+)?$")
         if not re.match(pattern, version):
-            raise ValueError(f"Wrong version format: {version}. Must be x.x.x")
+            raise ValueError(f"Wrong version format: {version}. Must be x.y or x.y.z")
 
         current = self.baseline(connection=connection)
         if current and current >= version:

{pum-1.1.18 → pum-1.2.0}/pum/upgrader.py

@@ -144,8 +144,7 @@ class Upgrader:
         if name not in self.config.demo_data():
             raise PumException(f"Demo data '{name}' not found in the configuration.")
 
-        demo_data_file = self.config.base_path / self.config.demo_data()[name]
-        logger.info("Installing demo data from %s", demo_data_file)
+        logger.info(f"Installing demo data {name}")
 
         for pre_hook in self.config.pre_hook_handlers():
             pre_hook.execute(connection=connection, commit=False, parameters=parameters)
@@ -153,11 +152,13 @@ class Upgrader:
         connection.commit()
 
         parameters_literals = SqlContent.prepare_parameters(parameters)
-        SqlContent(sql=demo_data_file).execute(
-            connection=connection,
-            commit=False,
-            parameters=parameters_literals,
-        )
+        for demo_data_file in self.config.demo_data()[name]:
+            demo_data_file = self.config.base_path / demo_data_file
+            SqlContent(sql=demo_data_file).execute(
+                connection=connection,
+                commit=False,
+                parameters=parameters_literals,
+            )
 
         connection.commit()
 

{pum-1.1.18 → pum-1.2.0}/pum.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.1.18
+Version: 1.2.0
 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.18 → pum-1.2.0}/test/test_upgrader.py

@@ -407,6 +407,24 @@ class TestUpgrader(unittest.TestCase):
             count = cursor.fetchone()[0]
             self.assertEqual(count, 4)
 
+    def test_demo_data_multi(self) -> None:
+        """Test the installation of demo data."""
+        test_dir = Path("test") / "data" / "demo_data_multi"
+        cfg = PumConfig.from_yaml(test_dir / ".pum.yaml")
+        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)
+            with self.assertRaises(PumException):
+                upgrader.install_demo_data(connection=conn, name="nope, nothing here fella")
+            upgrader.install_demo_data(connection=conn, name="some cool demo dataset")
+            self.assertTrue(sm.exists(conn))
+            cursor = conn.cursor()
+            cursor.execute("SELECT COUNT(*) FROM pum_test_data.some_table;")
+            count = cursor.fetchone()[0]
+            self.assertEqual(count, 4)
+
     def test_dependencies(self) -> None:
         """Test the installation of dependencies."""
         test_dir = Path("test") / "data" / "dependencies"