import pymysql
import sys

DB_CONFIG = dict(
    host="localhost",
    port=3306,
    user="root",
    password="",
    db="cems_db",
    charset="utf8mb4",
)

def main():
    try:
        conn = pymysql.connect(**DB_CONFIG)
        with conn.cursor() as cur:
            cur.execute("ALTER TABLE project_category_tb DROP COLUMN icon;")
        conn.commit()
        print("Successfully dropped 'icon' column from 'project_category_tb'.")
    except Exception as e:
        print(f"Failed to drop column: {e}")

if __name__ == "__main__":
    main()
