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

pum/hook.py

@@ -59,8 +59,9 @@ class HookBase(abc.ABC):
             connection: The database connection.
             sql: The SQL statement to execute or a path to a SQL file..
         """
+        parameters_literals = SqlContent.prepare_parameters(self._parameters)
         SqlContent(sql).execute(
-            connection=self._connection, parameters=self._parameters, commit=False
+            connection=self._connection, parameters=parameters_literals, commit=False
         )
 
     execute.__isfinal__ = True
@@ -191,13 +192,15 @@ class HookHandler:
             f"Executing hook from file: {self.file} or SQL code with parameters: {parameters}",
         )
 
+        parameters_literals = SqlContent.prepare_parameters(parameters)
+
         if self.file is None and self.code is None:
             raise ValueError("No file or SQL code specified for the migration hook.")
 
         if self.file:
             if self.file.suffix == ".sql":
                 SqlContent(self.file).execute(
-                    connection=connection, commit=False, parameters=parameters
+                    connection=connection, commit=False, parameters=parameters_literals
                 )
             elif self.file.suffix == ".py":
                 for parameter_arg in self.parameter_args:
@@ -208,9 +211,10 @@ class HookHandler:
                         )
 
                 _hook_parameters = {}
-                for key, value in parameters.items():
-                    if key in self.parameter_args:
-                        _hook_parameters[key] = value
+                if parameters:
+                    for key, value in parameters.items():
+                        if key in self.parameter_args:
+                            _hook_parameters[key] = value
                 self.hook_instance._prepare(connection=connection, parameters=parameters)
                 try:
                     if _hook_parameters:
@@ -225,7 +229,7 @@ class HookHandler:
                     f"Unsupported file type for migration hook: {self.file.suffix}. Only .sql and .py files are supported."
                 )
         elif self.code:
-            SqlContent(self.code).execute(connection, parameters=parameters, commit=False)
+            SqlContent(self.code).execute(connection, parameters=parameters_literals, commit=False)
 
         if commit:
             connection.commit()

pum/parameter.py

@@ -11,6 +11,7 @@ class ParameterType(Enum):
         INTEGER (str): Represents an integer parameter type.
         TEXT (str): Represents a text parameter type.
         DECIMAL (str): Represents a decimal parameter type.
+        PATH (str): Represents a path parameter type.
 
     """
 
@@ -18,6 +19,7 @@ class ParameterType(Enum):
     INTEGER = "integer"
     TEXT = "text"
     DECIMAL = "decimal"
+    PATH = "path"
 
 
 class ParameterDefinition:

pum/sql_content.py

@@ -263,3 +263,20 @@ class SqlContent:
                 ) from e
 
         return [format_sql(statement, parameters) for statement in sql_code]
+
+    @staticmethod
+    def prepare_parameters(parameters: dict | None):
+        """
+        Prepares a dictionary of parameters for use in SQL queries by converting each value to a psycopg.sql.Literal.
+
+        Args:
+            parameters: A dictionary of parameters to be converted, or None.
+
+        Returns:
+            dict: A new dictionary with the same keys as `parameters`, where each value is wrapped in psycopg.sql.Literal.
+        """
+        parameters_literals = {}
+        if parameters:
+            for key, value in parameters.items():
+                parameters_literals[key] = psycopg.sql.Literal(value)
+        return parameters_literals

pum/upgrader.py

@@ -4,7 +4,6 @@ import logging
 import packaging
 import packaging.version
 import psycopg
-import copy
 
 from .pum_config import PumConfig
 from .exceptions import PumException
@@ -76,8 +75,6 @@ class Upgrader:
             commit:
                 If True, the changes will be committed to the database.
         """
-        parameters_literals = self._prepare_parameters(parameters)
-
         if demo_data and demo_data not in self.config.demo_data():
             raise PumException(
                 f"Demo data '{demo_data}' not found in the configuration. Available demo data: {self.config.demo_data()}"
@@ -98,8 +95,9 @@ class Upgrader:
             )
 
         for pre_hook in self.config.pre_hook_handlers():
-            pre_hook.execute(connection=connection, commit=False, parameters=parameters_literals)
+            pre_hook.execute(connection=connection, commit=False, parameters=parameters)
 
+        parameters_literals = SqlContent.prepare_parameters(parameters)
         last_changelog = None
         for changelog in self.config.changelogs(max_version=max_version):
             last_changelog = changelog
@@ -117,7 +115,7 @@ class Upgrader:
             )
 
         for post_hook in self.config.post_hook_handlers():
-            post_hook.execute(connection=connection, commit=False, parameters=parameters_literals)
+            post_hook.execute(connection=connection, commit=False, parameters=parameters)
 
         logger.info(
             "Installed %s.pum_migrations table and applied changelogs up to version %s",
@@ -148,16 +146,15 @@ class Upgrader:
         if name not in self.config.demo_data():
             raise PumException(f"Demo data '{name}' not found in the configuration.")
 
-        parameters_literals = self._prepare_parameters(parameters)
-
         demo_data_file = self.config.base_path / self.config.demo_data()[name]
         logger.info("Installing demo data from %s", demo_data_file)
 
         for pre_hook in self.config.pre_hook_handlers():
-            pre_hook.execute(connection=connection, commit=False, parameters=parameters_literals)
+            pre_hook.execute(connection=connection, commit=False, parameters=parameters)
 
         connection.commit()
 
+        parameters_literals = SqlContent.prepare_parameters(parameters)
         SqlContent(sql=demo_data_file).execute(
             connection=connection,
             commit=False,
@@ -167,22 +164,6 @@ class Upgrader:
         connection.commit()
 
         for post_hook in self.config.post_hook_handlers():
-            post_hook.execute(connection=connection, commit=False, parameters=parameters_literals)
+            post_hook.execute(connection=connection, commit=False, parameters=parameters)
 
         logger.info("Demo data '%s' installed successfully.", name)
-
-    @staticmethod
-    def _prepare_parameters(parameters: dict | None):
-        """
-        Prepares a dictionary of parameters for use in SQL queries by converting each value to a psycopg.sql.Literal.
-
-        Args:
-            parameters: A dictionary of parameters to be converted, or None.
-
-        Returns:
-            dict: A new dictionary with the same keys as `parameters`, where each value is wrapped in psycopg.sql.Literal.
-        """
-        parameters_literals = copy.deepcopy(parameters) if parameters else {}
-        for key, value in parameters_literals.items():
-            parameters_literals[key] = psycopg.sql.Literal(value)
-        return parameters_literals

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

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

@@ -6,18 +6,18 @@ 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/hook.py,sha256=5MrVa6Xr0o28RfsXylGDatlM_vOKfKtGJmhYx8crC94,9541
 pum/info.py,sha256=VSCUZJJ_ae-khKaudwbgqszZXBMKB_yskuQo5Mc1PgY,1024
-pum/parameter.py,sha256=e9f80kMZpART9laeImW_YECeTvwDyDSmZlTeJGvpS_8,2449
+pum/parameter.py,sha256=rx8Wsm4wjm7YLVDqAx_NBIvkKB_ljEIGR88l17aaGbM,2521
 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/sql_content.py,sha256=KoAa3CNVH_Hh6Z3FxpCYJCsG1NEiCRpfzlfP8wPFZHE,10414
+pum/upgrader.py,sha256=HrFy-ev55t_nfaMYDOVRd44ACvb56-f4r4rv7m1UBVg,6387
 pum/conf/pum_config_example.yaml,sha256=_nwV_7z6S_Se-mejh_My0JFLY-A0Q4nigeLGPZAfcqg,424
-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,,
+pum-1.1.6.dist-info/licenses/LICENSE,sha256=2ylvL381vKOhdO-w6zkrOxe9lLNBhRQpo9_0EbHC_HM,18046
+pum-1.1.6.dist-info/METADATA,sha256=adZCP888gUAgoCaPs0bkgMOb7Qd0hvM9PYIKnljXD4M,3138
+pum-1.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pum-1.1.6.dist-info/entry_points.txt,sha256=U6dmxSpKs1Pe9vWiR29VPhJMDjrmZeJCSxvfLGR8BD4,36
+pum-1.1.6.dist-info/top_level.txt,sha256=ddiI4HLBhY6ql-NNm0Ez0JhoOHdWDIzrHeCdHmmagcc,4
+pum-1.1.6.dist-info/RECORD,,