lbxtoolkit 2.3.0__tar.gz → 2.3.1__tar.gz

{lbxtoolkit-2.3.0 → lbxtoolkit-2.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lbxtoolkit
-Version: 2.3.0
+Version: 2.3.1
 Summary: 
 Author: Cristiano P. Ferrari
 Author-email: cristiano.ferrari@lbx.com.br

{lbxtoolkit-2.3.0 → lbxtoolkit-2.3.1}/lbxtoolkit/postgresql.py

@@ -170,7 +170,7 @@ conexao.close()
             raise
         #
         #
-    def db_insert_df(self, DataFrame, Tabela, Schema=None, Colunas=None, OnConflict=None, TypeCast=None, Returning=None): # Insere os dados de um dataframe em uma tabela equivalente no banco (exige mesma estrutura de colunas)   
+    def db_insert_df(self, DataFrame, Tabela, Schema=None, Colunas=None, OnConflict=None, TypeCast=None, Returning=None, Truncar=False): # Insere os dados de um dataframe em uma tabela equivalente no banco (exige mesma estrutura de colunas)   
         """
         Insere um DataFrame no PostgreSQL usando psycopg2 e execute_values().
         
@@ -223,6 +223,15 @@ conexao.close()
         try:
             self.Cursor = self.Conexao.cursor()
 
+            if Truncar:
+                try:
+                    self.Cursor.execute(f"truncate table {Schema}.{Tabela} restart identity cascade;")
+                    self.Conexao.commit()
+                except Exception as Err:                    
+                    self.logger.erro(f'{__name__}:Erro [{Err}] ao TRUNCAR a tabela {Schema}.{Tabela}.')
+                    self.Conexao.rollback()
+                    raise
+
             if Returning:
                 Retorno = []
                 for valor in Valores:
@@ -233,7 +242,7 @@ conexao.close()
 
             self.Conexao.commit()        
         except Exception as Err:
-            self.logger.erro(f'{__name__}: Erro [{Err}] ao inserir dados na tabela {Schema}.{Tabela}!')    
+            self.logger.erro(f'{__name__}: Erro [{Err}] ao INSERIR dados na tabela {Schema}.{Tabela}!')    
             self.Conexao.rollback()
         finally:        
             self.Cursor.close()
@@ -263,7 +272,7 @@ conexao.close()
         #
     def db_update(self, Query): # Retorna um cursor à partir de um select
         UpdRows = 0
-        BlackList = ['INSERT ', 'SELECT ', 'CREATE ', 'DROP ', 'MERGE ', 'REPLACE ', 'CALL ', 'EXECUTE ']
+        BlackList = ['INSERT ', 'SELECT ', 'CREATE ', 'DROP ', 'MERGE ', 'REPLACE ' ] ## 'CALL ', 'EXECUTE ' ]
         if any(element in Query.upper() for element in BlackList):
             BlackListed = [element for element in BlackList if element in Query.upper()]          
             self.logger.erro(f'{__name__}: Este método permite apenas updates e deletes. A query informada possui as seguintes palavras reservadas não aceitas: {BlackListed} e não foi executada!')

{lbxtoolkit-2.3.0 → lbxtoolkit-2.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lbxtoolkit"
-version = "2.3.0"
+version = "2.3.1"
 description = ""
 authors = ["Cristiano P. Ferrari <cristiano.ferrari@lbx.com.br>"]
 readme = "README.md"