pum 1.1.2__py3-none-any.whl → 1.1.4__py3-none-any.whl

pum/config_model.py

@@ -26,6 +26,12 @@ class ParameterDefinitionModel(PumCustomBaseModel):
     default: Optional[Any] = None
     description: Optional[str] = None
 
+    @model_validator(mode="before")
+    def validate_default(cls, values):
+        if values.get("type") == ParameterType.BOOLEAN:
+            values["default"] = values.get("default", False) in (1, "1", "true", "TRUE", True)
+        return values
+
 
 class HookModel(PumCustomBaseModel):
     """

pum/pum_config.py

@@ -18,7 +18,6 @@ from .config_model import ConfigModel
 from .hook import HookHandler
 import tempfile
 import sys
-import atexit
 
 
 try:
@@ -26,7 +25,7 @@ try:
 except importlib.metadata.PackageNotFoundError:
     # Fallback: try to read from pum-*.dist-info/METADATA
     dist_info_dirs = glob.glob(os.path.join(os.path.dirname(__file__), "..", "pum-*.dist-info"))
-    version = None
+    versions = []
     for dist_info in dist_info_dirs:
         metadata_path = os.path.join(dist_info, "METADATA")
         if os.path.isfile(metadata_path):
@@ -34,11 +33,11 @@ except importlib.metadata.PackageNotFoundError:
                 for line in f:
                     if line.startswith("Version:"):
                         version = line.split(":", 1)[1].strip()
+                        versions.append(version)
                         break
-        if version:
-            break
-    if version:
-        PUM_VERSION = packaging.version.Version(version)
+    if versions:
+        # Pick the highest version
+        PUM_VERSION = max((packaging.version.Version(v) for v in versions))
     else:
         PUM_VERSION = packaging.version.Version("0.0.0")
 
@@ -138,6 +137,17 @@ class PumConfig:
         """Return the base path used for configuration and changelogs."""
         return self._base_path
 
+    def parameters(self) -> list[ParameterDefinition]:
+        """Return a list of migration parameters.
+
+        Returns:
+            list[ParameterDefinition]: A list of migration parameter definitions.
+
+        """
+        return [
+            ParameterDefinition(**parameter.model_dump()) for parameter in self.config.parameters
+        ]
+
     def parameter(self, name: str) -> ParameterDefinition:
         """Get a specific migration parameter by name.
 
@@ -250,6 +260,15 @@ class PumConfig:
         """Return a dictionary of demo data files defined in the configuration."""
         return {dm.name: dm.file for dm in self.config.demo_data}
 
+    def __del__(self):
+        # Cleanup temporary directories and sys.path modifications
+        if self.dependency_path:
+            # Remove from sys.path if present
+            sys.path = [p for p in sys.path if p != str(self.dependency_path)]
+            # Remove the directory if it exists and is a TemporaryDirectory
+            if hasattr(self, "_temp_dir") and self._temp_dir:
+                self.dependency_tmp.cleanup()
+
     def validate(self, install_dependencies: bool = False) -> None:
         """Validate the changelogs and hooks.
 
@@ -258,16 +277,10 @@ class PumConfig:
         """
 
         if install_dependencies and self.config.dependencies:
-            temp_dir = tempfile.TemporaryDirectory()
-            self.dependency_path = Path(temp_dir.name)
+            self.dependency_tmp = tempfile.TemporaryDirectory()
+            self.dependency_path = Path(self.dependency_tmp.name)
             sys.path.insert(0, str(self.dependency_path))
 
-            def cleanup():
-                sys.path = [p for p in sys.path if p != str(self.dependency_path)]
-                temp_dir.cleanup()
-
-            atexit.register(cleanup)
-
         parameter_defaults = {}
         for parameter in self.config.parameters:
             parameter_defaults[parameter.name] = psycopg.sql.Literal(parameter.default)

{pum-1.1.2.dist-info → pum-1.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pum
-Version: 1.1.2
+Version: 1.1.4
 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.2.dist-info → pum-1.1.4.dist-info}/RECORD

@@ -2,22 +2,22 @@ pum/__init__.py,sha256=P-NHd6_SYpk9aypefLI62QCZ3f5APOMCwSzrFFCKAew,759
 pum/changelog.py,sha256=yDc5swmMd5gb2vCEAlenoq5gs-ZEGc4uXicBtiGxkOk,3692
 pum/checker.py,sha256=GT2v7793HP1g94dv0mL6CHtQfblQwAyeFHEWCy44lkc,14379
 pum/cli.py,sha256=nnuApRDXgWaiBZXw2F3iwUxI5nzfIqyMdwhTXS-Jde8,14045
-pum/config_model.py,sha256=8FREAJpnm6-4Mby_udKpNo4WPbgn-4ZL1e-02a2FUFY,6147
+pum/config_model.py,sha256=piSQBYp29gnzqmtN50_OUw5o9d7Q5dEzj4q6G7cIFik,6396
 pum/dependency_handler.py,sha256=34wvDGWlI-vsMFm1z0XcSDN2cnL_VFAh61GWmeYEzk4,3841
 pum/dumper.py,sha256=EJZ8T44JM0GKgdqw1ENOfhZ-RI89OQ4DNdoTZKtLdEw,3404
 pum/exceptions.py,sha256=xyzzY4ht1nKfrVt59Giulflpmu83nJhxoTygrqiqPlw,1137
 pum/hook.py,sha256=L4Cnr34zrgPzxso9CdsUYWmtuOXRmFccQZ9Lp4IYCBM,9326
 pum/info.py,sha256=VSCUZJJ_ae-khKaudwbgqszZXBMKB_yskuQo5Mc1PgY,1024
 pum/parameter.py,sha256=e9f80kMZpART9laeImW_YECeTvwDyDSmZlTeJGvpS_8,2449
-pum/pum_config.py,sha256=7rGPaLreOOLl3kuBK_FzYxW02PiWHT4ayenWzOiJWJA,10566
+pum/pum_config.py,sha256=xTEdYh-628pnkoqH56NoPIuhIQauc1d6hksM8hcCCU4,11232
 pum/role_manager.py,sha256=yr-fmytflGqANY3IZIpgJBoMOK98ynTWfemIBhAy79A,10131
 pum/schema_migrations.py,sha256=FiaqAbhFX7vd3Rk_R43kd7-QWfil-Q5587EU8xSLBkA,10504
 pum/sql_content.py,sha256=gwgvcdXOXxNz3RvLtL8Bqr5WO3KKq3sluhbj4OAEnQs,9756
 pum/upgrader.py,sha256=jvl6vmpgxGyYiw8rrWC_bDC7Zd4wHJqGLXCK8EMt9wY,7109
 pum/conf/pum_config_example.yaml,sha256=_nwV_7z6S_Se-mejh_My0JFLY-A0Q4nigeLGPZAfcqg,424
-pum-1.1.2.dist-info/licenses/LICENSE,sha256=2ylvL381vKOhdO-w6zkrOxe9lLNBhRQpo9_0EbHC_HM,18046
-pum-1.1.2.dist-info/METADATA,sha256=TcW5O8n3DvCKSUPPiIyCk3zu-7cZJrIJuHyZ4RGtHzg,3138
-pum-1.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pum-1.1.2.dist-info/entry_points.txt,sha256=U6dmxSpKs1Pe9vWiR29VPhJMDjrmZeJCSxvfLGR8BD4,36
-pum-1.1.2.dist-info/top_level.txt,sha256=ddiI4HLBhY6ql-NNm0Ez0JhoOHdWDIzrHeCdHmmagcc,4
-pum-1.1.2.dist-info/RECORD,,
+pum-1.1.4.dist-info/licenses/LICENSE,sha256=2ylvL381vKOhdO-w6zkrOxe9lLNBhRQpo9_0EbHC_HM,18046
+pum-1.1.4.dist-info/METADATA,sha256=eZOs5rKpdCVJJR1BSQKeTjMdhneE57kZMrNaTnv-US8,3138
+pum-1.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pum-1.1.4.dist-info/entry_points.txt,sha256=U6dmxSpKs1Pe9vWiR29VPhJMDjrmZeJCSxvfLGR8BD4,36
+pum-1.1.4.dist-info/top_level.txt,sha256=ddiI4HLBhY6ql-NNm0Ez0JhoOHdWDIzrHeCdHmmagcc,4
+pum-1.1.4.dist-info/RECORD,,