first_project

alter_table_for_user_request.py

@@ -0,0 +1,61 @@
+
+import mysql.connector
+
+def get_connection_details():
+    details = {}
+    with open('setting.txt', 'r') as f:
+        for line in f:
+            key, value = line.strip().split(':', 1)
+            details[key.strip()] = value.strip()
+    return details
+
+def alter_table_columns():
+    try:
+        config = get_connection_details()
+        conn = mysql.connector.connect(
+            host=config.get('Host'),
+            port=config.get('Port'),
+            user=config.get('Username'),
+            password=config.get('Password'),
+            database=config.get('Database')
+        )
+        cursor = conn.cursor()
+
+        # Check and rename 'unit' to 'plant'
+        cursor.execute("SHOW COLUMNS FROM extension_data LIKE 'unit'")
+        if cursor.fetchone():
+            print("Renaming 'unit' to 'plant'...")
+            cursor.execute("ALTER TABLE extension_data CHANGE COLUMN unit plant VARCHAR(255)")
+            print("Column 'unit' renamed to 'plant' successfully.")
+        else:
+            print("Column 'unit' does not exist or already renamed.")
+
+        # Check and drop 'source_file'
+        cursor.execute("SHOW COLUMNS FROM extension_data LIKE 'source_file'")
+        if cursor.fetchone():
+            print("Dropping 'source_file' column...")
+            cursor.execute("ALTER TABLE extension_data DROP COLUMN source_file")
+            print("Column 'source_file' dropped successfully.")
+        else:
+            print("Column 'source_file' does not exist or already dropped.")
+
+        # Check and drop 'sheet'
+        cursor.execute("SHOW COLUMNS FROM extension_data LIKE 'sheet'")
+        if cursor.fetchone():
+            print("Dropping 'sheet' column...")
+            cursor.execute("ALTER TABLE extension_data DROP COLUMN sheet")
+            print("Column 'sheet' dropped successfully.")
+        else:
+            print("Column 'sheet' does not exist or already dropped.")
+
+        conn.commit()
+
+    except Exception as e:
+        print(f"An error occurred: {e}")
+    finally:
+        if 'conn' in locals() and conn.is_connected():
+            cursor.close()
+            conn.close()
+
+if __name__ == "__main__":
+    alter_table_columns()