feat:上传文件,生成记录,保存文件
This commit is contained in:
0
db/__init__.py
Normal file
0
db/__init__.py
Normal file
44
db/sql_db.py
Normal file
44
db/sql_db.py
Normal file
@@ -0,0 +1,44 @@
|
||||
from sqlalchemy import Column, DateTime, Integer, BigInteger, String, create_engine, Boolean, Text
|
||||
from sqlalchemy.orm import declarative_base, sessionmaker
|
||||
|
||||
# 申明基类对象
|
||||
Base = declarative_base()
|
||||
from decouple import config
|
||||
|
||||
DB_PATH = config('DB_PATH', default='E://pyptoject//yj_resume//main.sqlite3')
|
||||
|
||||
|
||||
class DBTASK(Base):
|
||||
__tablename__ = 'db_task'
|
||||
id = Column(String(100), primary_key=True)
|
||||
create_time = Column(DateTime, nullable=False, )
|
||||
# 0 代表待执行,1 成功,2 失败
|
||||
status = Column(Integer, nullable=False, default=0)
|
||||
success_num = Column(Integer, nullable=False, default=0)
|
||||
total_num = Column(Integer, nullable=False, default=0)
|
||||
fail_num = Column(Integer, nullable=False, default=0)
|
||||
message = Column(Text, nullable=True)
|
||||
|
||||
|
||||
class DBRESUME(Base):
|
||||
__tablename__ = 'db_resume'
|
||||
id = Column(String(100), primary_key=True)
|
||||
# 每个任务对应一个文件夹ID
|
||||
task_id = Column(String(100), nullable=False)
|
||||
# 0 代表待执行,1 成功,2 失败
|
||||
status = Column(Integer, nullable=False, default=0)
|
||||
file_name = Column(String(100), nullable=True)
|
||||
# 可以用json表示提取的数据
|
||||
data_info = Column(Text, nullable=True)
|
||||
# 错误信息等
|
||||
message = Column(Text, nullable=True)
|
||||
|
||||
|
||||
class SqliteSqlalchemy(object):
|
||||
def __init__(self):
|
||||
# 创建sqlite连接引擎
|
||||
engine = create_engine(f'sqlite:///{DB_PATH}', echo=True)
|
||||
# 创建表
|
||||
Base.metadata.create_all(engine, checkfirst=True)
|
||||
# 创建sqlite的session连接对象
|
||||
self.session = sessionmaker(bind=engine)()
|
||||
Reference in New Issue
Block a user