52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
|
import time
|
||
|
|
|
||
|
|
from db.models.log_data_model import LOT_DATA_MODEL, LOT_DATA
|
||
|
|
from db.session import with_session
|
||
|
|
|
||
|
|
|
||
|
|
@with_session
|
||
|
|
def add_kb_to_db(session, data: LOT_DATA):
|
||
|
|
# 创建知识库实例
|
||
|
|
kb = LOT_DATA_MODEL(**data.__dict__)
|
||
|
|
session.add(kb)
|
||
|
|
|
||
|
|
|
||
|
|
@with_session
|
||
|
|
def delete_kb_from_db(session, id):
|
||
|
|
kb = session.query(LOT_DATA_MODEL).filter_by(id=id).first()
|
||
|
|
if kb:
|
||
|
|
session.delete(kb)
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
# 直接查询所有数据
|
||
|
|
@with_session
|
||
|
|
def get_kb_detail(session):
|
||
|
|
all_data = session.query(LOT_DATA_MODEL).all()
|
||
|
|
data = [LOT_DATA(**d.__dict__) for d in all_data]
|
||
|
|
return data
|
||
|
|
|
||
|
|
|
||
|
|
# 根据时间查询数据
|
||
|
|
@with_session
|
||
|
|
def get_kb_detail_by_time(session, start_time, end_time):
|
||
|
|
all_data = session.query(LOT_DATA_MODEL).filter(
|
||
|
|
start_time <= LOT_DATA_MODEL.create_time, LOT_DATA_MODEL.create_time <= end_time).all()
|
||
|
|
data = [LOT_DATA(**d.__dict__) for d in all_data]
|
||
|
|
return data
|
||
|
|
|
||
|
|
|
||
|
|
# 删除数据
|
||
|
|
@with_session
|
||
|
|
def delete_kb_detail_by_time(session, start_time, end_time):
|
||
|
|
d = session.query(LOT_DATA_MODEL).filter(start_time <= LOT_DATA_MODEL.create_time, LOT_DATA_MODEL.create_time <= end_time).delete()
|
||
|
|
return d
|
||
|
|
|
||
|
|
|
||
|
|
# 查询数据
|
||
|
|
@with_session
|
||
|
|
def get_data_by_field(session, field, value):
|
||
|
|
query = session.query(LOT_DATA_MODEL).filter(getattr(LOT_DATA_MODEL, field) == value)
|
||
|
|
result = query.all()
|
||
|
|
return result
|