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 = os.getenv('DATABASE_URL') or settings.database_url

engine = create_engine(DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

Base = declarative_base()


# 创建所有表（如果有新的模型类，会自动创建相应的表）
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)


def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()


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