zhang_crawl_bio/database/sqlite_engine.py

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


sqlite_file_name = OUTPUT_DIR / "database.db"
sqlite_url = f"sqlite:///{sqlite_file_name}"
engine = create_engine(sqlite_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)