dbhydra 2.2.17__py3-none-any.whl → 2.3.1__py3-none-any.whl

dbhydra/src/abstract_table.py

@@ -78,9 +78,10 @@ class AbstractSelectable:
             columns = self.columns
         elif column_string.find(",") == -1:
             assert column_string.count("select")<=1 #assume there are no table columns containing "select" substring
-            columns = [column_string.replace("select","").strip()]
+            columns = [column_string.replace("select","").replace(self.db1.identifier_quote,"").strip()]
         else:
-            columns = [x.strip() for x in column_string.split(",")]
+            assert column_string.count("select")<=1
+            columns = [x.replace("select","").replace(self.db1.identifier_quote,"").strip() for x in column_string.split(",")]
         return(columns)
 
 
@@ -208,7 +209,7 @@ class AbstractSelectable:
             queries, queries_destinations=self._build_queries_from_blocks()
             self._query_blocks=[]
             for i,query in enumerate(queries):
-                self.db1.execute()
+                self.db1.execute(query)
                 
                 if "SELECT" in query:
                     selected_columns=self._get_selected_columns(query)
@@ -268,10 +269,17 @@ class AbstractTable(AbstractJoinable, abc.ABC):
 
     def update(self, variable_assign, where=None, debug_mode = False):
         quote = self.db1.identifier_quote
-        if where is None:
-            query = f"UPDATE {quote}{self.name}{quote} SET {quote}{variable_assign}{quote}"
-        else:
-            query = f"UPDATE {quote}{self.name}{quote} SET {quote}{variable_assign}{quote} WHERE {quote}{where}{quote}"
+
+        query = f"UPDATE {quote}{self.name}{quote} SET {variable_assign}"
+        
+        if where:
+            query += f" WHERE {where}"
+
+        #Old broken implementation it gives `` for the SQL syntax query
+        # if where is None:
+        #     query = f"UPDATE {quote}{self.name}{quote} SET {quote}{variable_assign}{quote}"
+        # else:
+        #     query = f"UPDATE {quote}{self.name}{quote} SET {quote}{variable_assign}{quote} WHERE {quote}{where}{quote}"
 
         if debug_mode:
             print(query)

dbhydra/src/tables.py

@@ -38,7 +38,6 @@ PYTHON_TO_MYSQL_DATA_MAPPING = {
 
 def save_migration(function, *args, **kw):  # decorator
     def new_function(instance, *args, **kw):
-        print("TOTO TU")
         print(instance)
         print(*args)
         command = function.__name__
@@ -63,9 +62,12 @@ def save_migration(function, *args, **kw):  # decorator
             print(migration_dict)
         # TODO: add other methods
 
-        migrator = instance.db1.migrator
-        migrator.migration_list.append(migration_dict)
-        # migrator.migration_list_to_json()
+        if hasattr(instance.db1, 'migrator'):
+            migrator = instance.db1.migrator
+            migrator.migration_list.append(migration_dict)
+            # migrator.migration_list_to_json()
+        else:
+            print(f"[save_migration] WARNING: db1 object of type {type(instance.db1)} has no 'migrator' attribute. Migration not saved.")
         function(instance, *args, **kw)
 
     return (new_function)

dbhydra/tests/test_mysql_ddl.py

@@ -0,0 +1,91 @@
+##### DDL (data definition language) tests for MySQL #####
+
+import os
+import pytest
+import random
+import string
+import dbhydra.dbhydra_core as dh
+
+def random_table_name(prefix="test_table_"):
+    return prefix + ''.join(random.choices(string.ascii_lowercase + string.digits, k=8))
+
+# Rename mysqldb fixture and all references to db1
+@pytest.fixture(scope="module")
+def db1():
+    # Get the directory of this test file, then go up one level to the project root
+    root_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+    config_path = os.path.join(root_dir, "config-mysql.ini")
+    return dh.MysqlDb(config_file=config_path)
+
+@pytest.fixture(scope="function")
+def temp_mysql_table(db1):
+    table_name = random_table_name()
+    columns = ["id", "name"]
+    types = ["int", "varchar(255)"]
+    table = dh.MysqlTable(db1, table_name, columns, types)
+    with db1.connect_to_db():
+        table.create()
+        table_dict = db1.generate_table_dict()
+        assert table_name in table_dict, f"Temp table {table_name} was not created!"
+    yield table
+    # Cleanup
+    try:
+        with db1.connect_to_db():
+            table.drop()
+            table_dict = db1.generate_table_dict()
+            assert table_name not in table_dict, f"Temp table {table_name} was not dropped!"
+    except Exception as e:
+        print(f"[CLEANUP] Could not drop table {table_name}: {e}")
+        raise
+
+def test_mysql_create_and_drop_table(db1):
+    table_name = random_table_name()
+    columns = ["id", "name"]
+    types = ["int", "varchar(255)"]
+    table = dh.MysqlTable(db1, table_name, columns, types)
+    with db1.connect_to_db():
+        table.create()
+        table_dict = db1.generate_table_dict()
+        assert table_name in table_dict, f"Table {table_name} was not created!"
+        table.drop()
+        table_dict = db1.generate_table_dict()
+        assert table_name not in table_dict, f"Table {table_name} was not dropped!"
+
+def test_mysql_add_column(temp_mysql_table, db1):
+    table = temp_mysql_table
+    column = "age"
+    type = "int"
+    with db1.connect_to_db():
+        table.add_column(column, type)
+        df = table.select_to_df()
+        columns = df.columns.tolist()
+        assert column in columns, f"Column '{column}' was not added! Columns: {columns}"
+
+def test_mysql_drop_column(temp_mysql_table, db1):
+    table = temp_mysql_table
+    column = "age"
+    type = "int"
+    with db1.connect_to_db():
+        table.add_column(column, type)
+        df = table.select_to_df()
+        columns = df.columns.tolist()
+        assert column in columns, f"Column '{column}' was not added! Columns: {columns}"
+        table.drop_column(column)
+        df = table.select_to_df()
+        columns = df.columns.tolist()
+        assert column not in columns, f"Column '{column}' was not dropped! Columns: {columns}"
+
+def test_mysql_modify_column(temp_mysql_table, db1):
+    table = temp_mysql_table
+    column = "age"
+    type = "int"
+    with db1.connect_to_db():
+        table.add_column(column, type)
+        df = table.select_to_df()
+        columns = df.columns.tolist()
+        assert column in columns, f"Column '{column}' was not added! Columns: {columns}"
+        # Modify column type
+        type2 = "varchar(100)"
+        table.modify_column(column, type2)
+        types = table.get_all_types()
+        assert any("varchar" in t for t in types), f"Column '{column}' was not modified to varchar! Types: {types}" 

{dbhydra-2.2.17.dist-info → dbhydra-2.3.1.dist-info}/METADATA

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

{dbhydra-2.2.17.dist-info → dbhydra-2.3.1.dist-info}/RECORD

@@ -3,23 +3,24 @@ dbhydra/dbhydra_core.py,sha256=jn0VC3LkawR2P0Yd_oZNEXqP9o2BPEtOBWGzgFwfJyA,2537
 dbhydra/test_migrator.py,sha256=e3Nnb2mCd3CfjhjSexNg1tXVJMjkl5cCoYcuhbfZ4pM,803
 dbhydra/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbhydra/src/abstract_db.py,sha256=7Rv2XgdaKiQ5yBnU0EDh5uT8XbB3JI5Pis_in6iHn2c,7172
-dbhydra/src/abstract_table.py,sha256=UAjiejG8z7VVCWZ6bjMDkbEl_w8shVAX2euGytCgYWQ,18947
+dbhydra/src/abstract_table.py,sha256=5sTNPpUD31l4UbZ0at5CMg-AbLlOYodpq_sHmT1Lrb8,19331
 dbhydra/src/bigquery_db.py,sha256=77XsgvYbANlvYaJnuVve-kz-PNBx_CHoYCL-eYnA8e4,1834
 dbhydra/src/migrator.py,sha256=QzaODEFfraD9_6HN_Osaidaj-nLYQryCYYWwJtUu3n8,18931
 dbhydra/src/mongo_db.py,sha256=mP48zRjI7mXKpm45R8prroZI-Eo7JKf0KJqGX-oTy3w,1922
 dbhydra/src/mysql_db.py,sha256=lNWaS3YHtQE1Xii-ErWUDhxeo3b6HWgXxQAMJfgiGEE,3583
 dbhydra/src/postgres_db.py,sha256=L7MaBq_6ArwDSP_5LaEqK58oLxZ1X7FgIokcDOSB7wk,1805
 dbhydra/src/sqlserver_db.py,sha256=9Xi3NAliqM79MTV8fpNQb0nWMH8Bqjl1leJSEqgyT94,3611
-dbhydra/src/tables.py,sha256=oRbSTIHjeSc5ZCHmn74ZqdOvtBZDy6t_fp2KO-Z_Y-8,47612
+dbhydra/src/tables.py,sha256=Xt0c8o93ZthxTF6YF9TGS3Ud719qXSD3iF7-wGbckW4,47799
 dbhydra/src/xlsx_db.py,sha256=glFF-0dQK0GGinAqc2zvf9JQ9iAH9YkDAYd4Jb3oimQ,3616
 dbhydra/src/errors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbhydra/src/errors/exceptions.py,sha256=LVpfbTd3NHfQIM-D5TFAU6hOZwGQ3b5DwFD4B6vtf2U,149
 dbhydra/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbhydra/tests/test_cases.py,sha256=eAFGaHaIaab3md3HHm2_ryb_HHfObtcXDAEzLh4qWx8,508
 dbhydra/tests/test_mongo.py,sha256=M8TD72M0iQAk7ZcLTWwLmcmmF_zwALnYEGTWjhQlq0s,1979
+dbhydra/tests/test_mysql_ddl.py,sha256=-gncZE_HjTOAI1yxOn8u5TZkoqoRdShLMF469Trbb9U,3531
 dbhydra/tests/test_sql.py,sha256=aPFXyA0jh8o9VG3B5f9fNz7qDbuVPZ9TcE2twn5dAeQ,3126
-dbhydra-2.2.17.dist-info/LICENSE,sha256=k49Yga8CP889JJaHlOpGFzr_be2nqMoep2chYeIDctk,1091
-dbhydra-2.2.17.dist-info/METADATA,sha256=z0FPGa_MnMZbpQag6Ieff2Pqqq8VowEKwtfD6bpO-_Y,2299
-dbhydra-2.2.17.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-dbhydra-2.2.17.dist-info/top_level.txt,sha256=oO4Gf1T8_txIsIlp11GI0k7PtBIMb9GRwb5ObF4MLVg,8
-dbhydra-2.2.17.dist-info/RECORD,,
+dbhydra-2.3.1.dist-info/LICENSE,sha256=k49Yga8CP889JJaHlOpGFzr_be2nqMoep2chYeIDctk,1091
+dbhydra-2.3.1.dist-info/METADATA,sha256=Mwdf3U9g9K4GXTATZYRdO9qu1TZIYBREDdBp39YgSgk,2298
+dbhydra-2.3.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+dbhydra-2.3.1.dist-info/top_level.txt,sha256=oO4Gf1T8_txIsIlp11GI0k7PtBIMb9GRwb5ObF4MLVg,8
+dbhydra-2.3.1.dist-info/RECORD,,

{dbhydra-2.2.17.dist-info → dbhydra-2.3.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.38.4)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any