dbhydra 2.2.1__tar.gz → 2.2.5__tar.gz

{dbhydra-2.2.1 → dbhydra-2.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbhydra
-Version: 2.2.1
+Version: 2.2.5
 Summary: Data science friendly ORM combining Python
 Home-page: https://github.com/DovaX/dbhydra
 Author: DovaX

{dbhydra-2.2.1 → dbhydra-2.2.5}/dbhydra/src/abstract_table.py

@@ -102,7 +102,7 @@ class AbstractSelectable:
         return (results)
 
 
-    def select(self, query, flattening_of_results=False):
+    def select(self, query, flattening_of_results=False, debug_mode = False):
         """given SELECT query returns Python list"""
         """Columns give the number of selected columns"""
         
@@ -113,7 +113,8 @@ class AbstractSelectable:
             return(self) #to enable chaining
         
         else:
-            print(query)
+            if debug_mode:
+                print(query)
             self.db1.cursor.execute(query)
             
             columns_count=len(self._get_selected_columns(query))
@@ -131,18 +132,18 @@ class AbstractSelectable:
             return(rows)
             
 
-    def select_all(self):
+    def select_all(self, debug_mode = False):
         quote = self.db1.identifier_quote
         all_cols_query = ""
         for col in self.columns:
             all_cols_query = all_cols_query + quote + col + quote + ","
         if all_cols_query[-1] == ",":
             all_cols_query = all_cols_query[:-1]
-        list1 = self.select(f"SELECT {all_cols_query} FROM {quote}{self.name}{quote};")
+        list1 = self.select(f"SELECT {all_cols_query} FROM {quote}{self.name}{quote};", debug_mode = debug_mode)
         return (list1)
 
-    def select_to_df(self):
-        rows = self.select_all()
+    def select_to_df(self, debug_mode = False):
+        rows = self.select_all(debug_mode = debug_mode)
         if self.query_building_enabled:
             self.to_df()
             df=None
@@ -257,21 +258,23 @@ class AbstractTable(AbstractJoinable, abc.ABC):
         return cls(db1, name, columns, types, id_column_name=id_column_name)
     
     
-    def drop(self):
+    def drop(self, debug_mode = False):
         query = "DROP TABLE " + self.name
-        print(query)
+        if debug_mode:
+            print(query)
         self.execute(query)
 
-    def update(self, variable_assign, where=None):
+    def update(self, variable_assign, where=None, debug_mode = False):
         if where is None:
             query = "UPDATE " + self.name + " SET " + variable_assign
         else:
             query = "UPDATE " + self.name + " SET " + variable_assign + " WHERE " + where
-        print(query)
+        if debug_mode:
+            print(query)
         return self.execute(query)
 
     def update_from_df(
-            self, update_df: pd.DataFrame, where_column: Optional[str] = None, where_value: Any = None) -> None:
+            self, update_df: pd.DataFrame, where_column: Optional[str] = None, where_value: Any = None, debug_mode = False) -> None:
         """Build UPDATE SQL query from a dataframe and execute it.
 
         :param update_df: Dataframe with updated values - MUST only hold a single row
@@ -314,14 +317,15 @@ class AbstractTable(AbstractJoinable, abc.ABC):
                 raise AttributeError(f"Unknown column type '{column_type}'")
 
         column_value_string = column_value_string.rstrip(", ")
-        sql_query = f"UPDATE {self.name} SET {column_value_string}"
+        quote = self.db1.identifier_quote
+        sql_query = f"UPDATE {quote}{self.name}{quote} SET {column_value_string}"
 
         if where_column is not None and where_value is not None:
             sql_query += f" WHERE {where_column} = {where_value};"
         else:
             sql_query += ";"
-
-        print(sql_query)
+        if debug_mode:
+            print(sql_query)
         self.execute(sql_query)
 
     def _adjust_df(self, df: pd.DataFrame, debug_mode=False) -> pd.DataFrame:

{dbhydra-2.2.1 → dbhydra-2.2.5}/dbhydra/src/tables.py

@@ -141,7 +141,7 @@ class PostgresTable(AbstractTable):
         return (cls(db1, name, columns, types))
 
     # @save_migration
-    def create(self, foreign_keys=None):
+    def create(self, foreign_keys=None, debug_mode = False):
         assert len(self.columns) == len(self.types)
         assert self.columns[0] == self.id_column_name
         assert self.types[0].lower() == "int" or self.types[0].lower() == "integer"
@@ -151,7 +151,8 @@ class PostgresTable(AbstractTable):
 
         query = query[:-1]
         query += ");"
-        print(query)
+        if debug_mode:
+            print(query)
         try:
             self.db1.execute(query)
         except Exception as e:
@@ -495,7 +496,7 @@ class SqlServerTable(AbstractTable):
         types = information_schema_table.select(query)
         return (types)
 
-    def create(self):
+    def create(self, debug_mode = False):
         assert len(self.columns) == len(self.types)
         assert self.columns[0] == self.id_column_name
         assert self.types[0].lower() == "int"
@@ -503,8 +504,8 @@ class SqlServerTable(AbstractTable):
         for i in range(1, len(self.columns)):
             query += self.columns[i] + " " + self.types[i] + ","
         query += "PRIMARY KEY("+self.id_column_name+"))"
-
-        print(query)
+        if debug_mode:
+            print(query)
         try:
             self.db1.execute(query)
         except Exception as e:
@@ -729,13 +730,14 @@ class MysqlTable(AbstractTable):
 
         return num_of_records[0][0]
 
-    def drop(self):
+    def drop(self, debug_mode = False):
         query = "DROP TABLE `" + self.name + "`;"
-        print(query)
+        if debug_mode:
+            print(query)
         self.db1.execute(query)
 
     # @save_migration #TODO: Uncomment
-    def create(self, foreign_keys=None):
+    def create(self, foreign_keys=None, debug_mode = False):
         assert len(self.columns) == len(self.types)
         assert self.columns[0] == self.id_column_name
         assert self.types[0].lower() == "int"
@@ -745,8 +747,8 @@ class MysqlTable(AbstractTable):
             [f"`{column}` {type_.upper()}" for column, type_ in column_type_pairs]
         )
         query = f"CREATE TABLE `{self.name}` ({self.id_column_name} INT UNSIGNED AUTO_INCREMENT PRIMARY KEY, {fields})"
-
-        print(query)
+        if debug_mode:
+            print(query)
         try:
             self.db1.execute(query)
         except Exception as e:

{dbhydra-2.2.1 → dbhydra-2.2.5}/dbhydra.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbhydra
-Version: 2.2.1
+Version: 2.2.5
 Summary: Data science friendly ORM combining Python
 Home-page: https://github.com/DovaX/dbhydra
 Author: DovaX

{dbhydra-2.2.1 → dbhydra-2.2.5}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
     
 setuptools.setup(
     name='dbhydra',
-    version='2.2.1',
+    version='2.2.5',
     author='DovaX',
     author_email='dovax.ai@gmail.com',
     description='Data science friendly ORM combining Python',