| from sqlalchemy import create_engine | |
| from llama_index.core import SQLDatabase | |
| import pandas as pd | |
| db_user = "shenzhen_ai_for_vibemate_eson" | |
| db_password = "dBsnc7OrM0MVi0FEhiHe2y" | |
| db_host = "192.168.1.99" | |
| db_port = 3306 | |
| db_name = "hytto_surfease" | |
| engine = create_engine(f"mysql+pymysql://{db_user}:{db_password}@{db_host}:{db_port}/{db_name}") | |
| sql_db = SQLDatabase(engine) | |
| tables = sql_db.get_usable_table_names() | |
| csv_dir = "/home/purui/projects/chatbot/data/csv" | |
| for table in tables: | |
| query = f"SELECT * FROM {table}" | |
| df = pd.read_sql_query(query, engine) | |
| csv_path = csv_dir+f"/{table}.csv" | |
| df.to_csv(csv_path, index=False) |