maisaedu-poormans-dms 1.2.31__tar.gz → 1.2.32__tar.gz

{maisaedu_poormans_dms-1.2.31 → maisaedu_poormans_dms-1.2.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: maisaedu_poormans_dms
-Version: 1.2.31
+Version: 1.2.32
 Summary: A library for making database migration tasks, for +A Education
 Home-page: UNKNOWN
 Author: A+ Educação

{maisaedu_poormans_dms-1.2.31 → maisaedu_poormans_dms-1.2.32}/maisaedu_poormans_dms/postgres_migration/Migrator.py

@@ -47,6 +47,7 @@ class Migrator(MigratorInterface):
         write_batch_size: int = 10000,
         copy_config: object = None,
         ignore_columns: object = [],
+        use_copy: bool = True,
     ) -> None:
         self.__create_source_connection()
         self.__create_target_connection()
@@ -61,6 +62,7 @@ class Migrator(MigratorInterface):
             write_batch_size,
             copy_config,
             ignore_columns,
+            use_copy,
         )
         self.__close_source_connection()
         self.__close_target_connection()

{maisaedu_poormans_dms-1.2.31 → maisaedu_poormans_dms-1.2.32}/maisaedu_poormans_dms/postgres_migration/MigratorRow/MigratorRow.py

@@ -37,6 +37,7 @@ class MigratorRow(MigratorRowInterface):
         write_batch_size: int,
         copy_config: object,
         ignore_columns: object,
+        use_copy: bool = True,
     ):
         self.source_table_name = source_table_name
         self.target_table_name = target_table_name
@@ -47,6 +48,7 @@ class MigratorRow(MigratorRowInterface):
         self.write_batch_size = write_batch_size
         self.copy_config = copy_config
         self.ignore_columns = ignore_columns
+        self.use_copy = use_copy
 
     def __init_dependencies(self):
         self.migrator_reader = MigratorReader(
@@ -65,6 +67,7 @@ class MigratorRow(MigratorRowInterface):
             self.on_conflict,
             self.copy_config,
             self.ignore_columns,
+            self.use_copy,
         )
 
     def __migrate_without_incremental_column(self):
@@ -122,6 +125,7 @@ class MigratorRow(MigratorRowInterface):
         write_batch_size: int,
         copy_config: object,
         ignore_columns: object,
+        use_copy: bool = True,
     ) -> None:
         self.__set_configs(
             source_table_name,
@@ -133,6 +137,7 @@ class MigratorRow(MigratorRowInterface):
             write_batch_size,
             copy_config,
             ignore_columns,
+            use_copy,
         )
         self.__init_dependencies()
 

{maisaedu_poormans_dms-1.2.31 → maisaedu_poormans_dms-1.2.32}/maisaedu_poormans_dms/postgres_migration/MigratorRow/Writer.py

@@ -17,6 +17,7 @@ class MigratorWriter(MigratorRowWriterInterface):
     on_conflict: str
     copy_config: object
     ignore_columns: object
+    use_copy: bool = True
 
     def __init__(
         self,
@@ -26,6 +27,7 @@ class MigratorWriter(MigratorRowWriterInterface):
         on_conflict: str,
         copy_config: object,
         ignore_columns: object,
+        use_copy: bool = True
     ):
         self.target_conn = target_conn
         self.target_table_name = target_table_name
@@ -33,6 +35,7 @@ class MigratorWriter(MigratorRowWriterInterface):
         self.on_conflict = on_conflict
         self.copy_config = copy_config
         self.ignore_columns = ignore_columns
+        self.use_copy = use_copy
 
     def __bind_on_conflict(self):
         if self.on_conflict is True:
@@ -93,7 +96,7 @@ class MigratorWriter(MigratorRowWriterInterface):
                 insert_list.append(insert_row)
 
                 if len(insert_list) > self.write_batch_size:
-                    if self.on_conflict is None:
+                    if self.on_conflict is None and self.use_copy:
                         copy(
                             self.target_conn,
                             self.target_table_name,
@@ -102,6 +105,7 @@ class MigratorWriter(MigratorRowWriterInterface):
                             default_commit=default_commit,
                         )
                     else:
+                        on_conflict = '' if on_conflict is None else on_conflict
                         insert_batch(
                             self.target_conn,
                             insert_list,
@@ -114,7 +118,7 @@ class MigratorWriter(MigratorRowWriterInterface):
                     insert_list = []
 
             if len(insert_list) > 0:
-                if self.on_conflict is None:
+                if self.on_conflict is None and self.use_copy:
                     copy(
                         self.target_conn,
                         self.target_table_name,
@@ -123,6 +127,7 @@ class MigratorWriter(MigratorRowWriterInterface):
                         default_commit=default_commit,
                     )
                 else:
+                    on_conflict = '' if on_conflict is None else on_conflict
                     insert_batch(
                         self.target_conn,
                         insert_list,

{maisaedu_poormans_dms-1.2.31 → maisaedu_poormans_dms-1.2.32}/maisaedu_poormans_dms.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: maisaedu-poormans-dms
-Version: 1.2.31
+Version: 1.2.32
 Summary: A library for making database migration tasks, for +A Education
 Home-page: UNKNOWN
 Author: A+ Educação

{maisaedu_poormans_dms-1.2.31 → maisaedu_poormans_dms-1.2.32}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="maisaedu_poormans_dms",
-    version="1.2.31",
+    version="1.2.32",
     description="A library for making database migration tasks, for +A Education",
     license="MIT License",
     author="A+ Educação",