codefuse-chatbot/dev_opsgpt/orm/__init__.py

23 lines
517 B
Python
Raw Normal View History

2023-09-28 10:58:58 +08:00
from .db import _engine, Base
from loguru import logger
__all__ = [
]
def create_tables():
Base.metadata.create_all(bind=_engine)
def reset_tables():
Base.metadata.drop_all(bind=_engine)
create_tables()
def check_tables_exist(table_name) -> bool:
table_exist = _engine.dialect.has_table(_engine.connect(), table_name, schema=None)
return table_exist
def table_init():
if (not check_tables_exist("knowledge_base")) or (not check_tables_exist ("knowledge_file")):
create_tables()