问题处理

zhaoqingang

2024-12-06 9e51ec752c93d850db15cd61a7a3463d8fe94344

 app/models/base_model.py

@@ -1,10 +1,12 @@
from sqlalchemy import create_engine
import os

from sqlalchemy import create_engine, text
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, Session

from app.config.config import settings
# from app.models.agent_model import AgentModel

DATABASE_URL = settings.database_url
DATABASE_URL = os.getenv('DATABASE_URL') or settings.database_url

engine = create_engine(DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@@ -14,6 +16,12 @@

# 创建所有表（如果有新的模型类，会自动创建相应的表）
def init_db():
    try:
        sql = text('DROP TABLE IF EXISTS agent;')
        engine.execute(sql)
    except Exception as  e:
        print(e)

    Base.metadata.create_all(bind=engine)


@@ -22,4 +30,9 @@
    try:
        yield db
    finally:
        db.close()
        db.close()


# def drop_table():
#     sql = text('DROP TABLE IF EXISTS agent;')
#     engine.execute(sql)