copywriting_production/utils/sql_engine.py

from config.settings import OUTPUT_DIR, WORK_DIR, DB_URL
from datetime import datetime
from sqlmodel import Field, SQLModel, create_engine, Session, select
from sqlalchemy import text  # Add this import


engine = create_engine(DB_URL, echo=False)


def create_db_and_tables():
    SQLModel.metadata.create_all(engine)

def drop_table(model: SQLModel):
    """删除SearchResult表以便重新创建"""
    with engine.connect() as conn:
        # Use text() to create an executable SQL statement
        conn.execute(text(f"DROP TABLE IF EXISTS {model.__tablename__}"))
        conn.commit()
    SQLModel.metadata.create_all(engine)