生成数据库

lyg

2 天以前 22f370322412074174cde20ecfd14ec03657ab63

 knowledgebase/db/doc_db_models.py

@@ -6,6 +6,7 @@
# @description: 文档数据库模型

from sqlalchemy import create_engine, Column, DateTime, Integer, Text, ForeignKey
from sqlalchemy.dialects.mysql import LONGTEXT
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship
from sqlalchemy.orm import sessionmaker, scoped_session
@@ -22,6 +23,7 @@
    """
    __tablename__ = 't_paragraphs'
    id = Column(Integer, primary_key=True)
    # text = Column(LONGTEXT)
    text = Column(Text)
    title_level = Column(Integer)
    title_num = Column(Text)
@@ -139,17 +141,19 @@
#     is_del = Column(Integer)


def init_doc_db():
def init_doc_db(project_path):
    """
    初始化文档数据库
    :return: 数据库session实例
    """
    # mysql
    Log.info("连接并初始化文档数据库...")
    engine = create_engine('mysql+pymysql://root:123456@192.168.3.145:3306/knowledgebase', echo=False)
    # engine = create_engine('sqlite:///doc_db.db', echo=False)
    # engine = create_engine('mysql+pymysql://root:123456@10.74.15.171:3306/knowledgebase_xx25', echo=False)
    engine = create_engine(f'sqlite:///{project_path}/docs/doc_db.db', echo=False)
    Base.metadata.create_all(engine)
    SessionFactory = sessionmaker(bind=engine)
    Session = scoped_session(SessionFactory)
    session = Session()
    return session

# _xx25