pum 1.1.2__tar.gz → 1.1.4__tar.gz

{pum-1.1.2/pum.egg-info → pum-1.1.4}/PKG-INFO

@@ -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 → pum-1.1.4}/docs/docs/index.md

@@ -8,7 +8,7 @@ PUM (PostgreSQL Upgrades Manager) is a robust database migration management tool
 
 # Key Features
 
-- **Command-line and Python Integration**: Use PUM as a standalone [CLI tool](cli.md) or integrate it into your Python project using the [API](api/api.md).
+- **Command-line and Python Integration**: Use PUM as a standalone [CLI tool](cli.md) or integrate it into your Python project using the [API](api/pum_config.md).
 - **Database Versioning**: Automatically manage database versioning with a metadata table.
 - **Changelog Management**: Apply and track SQL delta files for database upgrades.
 - **Migration Hooks**: Define custom hooks to execute additional SQL or Python code before or after migrations. This feature allows you to isolate data (table) code from application code (such as views and triggers), ensuring a clear separation of concerns and more maintainable database structures.

{pum-1.1.2 → pum-1.1.4}/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-1.1.2 → pum-1.1.4}/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 → pum-1.1.4/pum.egg-info}/PKG-INFO

@@ -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 → pum-1.1.4}/pum.egg-info/SOURCES.txt

@@ -19,7 +19,6 @@ docs/docs/getting_started.md
 docs/docs/hooks.md
 docs/docs/index.md
 docs/docs/roles.md
-docs/docs/api/api.md
 docs/docs/api/changelog.md
 docs/docs/api/dump_format.md
 docs/docs/api/dumper.md

pum-1.1.2/docs/docs/api/api.md

@@ -1,11 +0,0 @@
-# PUM API
-
-- [Changelog](api/changelog.md)
-- [Hook](api/hook.md)
-- [Hook](api/hook_handler.md)
-- [ParameterDefinition](api/parameter_definition.md)
-- [ParameterType](api/parameter_type.md)
-- [PumConfig](api/pum_config.md)
-- [SchemaMigrations](api/schema_migrations.md)
-- [Upgrader](api/upgrader.md)
-- [Exceptions](api/exceptions.md)