excel_tool/utils/db_engine.py

from sqlite3 import connect
from utils.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 get_engine():
    return engine

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)