430 lines
17 KiB
Python
430 lines
17 KiB
Python
import os
|
||
import sys
|
||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
from dataclasses import fields
|
||
from typing import List, Optional, Dict, Any
|
||
from datetime import datetime, timedelta
|
||
from busisness.models import ArtifactInfoModel,PDRecordModel,FreqRecordModel
|
||
from common.sqlite_handler import SQLiteHandler
|
||
|
||
|
||
def filter_dict_for_model(data_dict: Dict[str, Any], model_class) -> Dict[str, Any]:
|
||
"""过滤字典，只保留模型中定义的字段"""
|
||
# 获取模型中定义的所有字段名称
|
||
model_fields = {field.name for field in fields(model_class)}
|
||
# 过滤字典，只保留模型中存在的字段
|
||
return {k: v for k, v in data_dict.items() if k in model_fields}
|
||
|
||
class BaseDal:
|
||
def __init__(self) -> None:
|
||
"""初始化数据访问层，创建数据库连接"""
|
||
# 假设数据库文件在db目录下
|
||
self.db_dao = SQLiteHandler.get_instance("db/three.db", max_readers=50, busy_timeout=4000)
|
||
self.db_dao_record = SQLiteHandler.get_instance("db/three-record.db", max_readers=50, busy_timeout=4000)
|
||
|
||
class ArtifactDal(BaseDal):
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
def get_all(self) -> List[ArtifactInfoModel]:
|
||
"""获取所有ArtifactTask记录"""
|
||
try:
|
||
# 查询所有记录
|
||
sql = "SELECT * FROM ArtifactTask"
|
||
results = self.db_dao.execute_read(sql)
|
||
|
||
# 将查询结果转换为ArtifactInfo对象列表
|
||
artifacts = []
|
||
for row in results:
|
||
# 过滤字典，只保留模型中定义的字段
|
||
# filtered_data = filter_dict_for_model(dict(row), ArtifactInfoModel)
|
||
artifact = ArtifactInfoModel(**row)
|
||
artifacts.append(artifact)
|
||
|
||
return artifacts
|
||
except Exception as e:
|
||
print(f"获取所有构件任务失败: {e}")
|
||
return []
|
||
|
||
def get_top_artifact(self, top: int,desc:str="ID desc",where:str="1=1") -> List[ArtifactInfoModel]:
|
||
"""获取top条数数据，根据ArtifactID升序"""
|
||
try:
|
||
# 确保top为正整数
|
||
if not isinstance(top, int) or top <= 0:
|
||
raise ValueError("top参数必须是正整数")
|
||
|
||
# 查询指定数量的记录，按ArtifactID升序排列
|
||
sql = f"SELECT * FROM ArtifactTask WHERE {where} ORDER BY {desc} LIMIT ?"
|
||
results = self.db_dao.execute_read(sql, (top,))
|
||
|
||
# 将所有查询结果转换为ArtifactInfoModel对象列表
|
||
artifacts = []
|
||
for row in results:
|
||
# 保证row的变量和模板变量一致
|
||
artifact = ArtifactInfoModel(**row)
|
||
artifacts.append(artifact)
|
||
|
||
return artifacts
|
||
except Exception as e:
|
||
print(f"获取top构件任务失败: {e}")
|
||
return []
|
||
|
||
|
||
def exists_by_id(self, artifact_id: int) -> bool:
|
||
"""根据构件ID获取构件任务"""
|
||
try:
|
||
sql = "SELECT count(1) FROM ArtifactTask WHERE ArtifactID = ?"
|
||
results = self.db_dao.execute_read(sql, (artifact_id,))
|
||
|
||
rows = list(results)
|
||
if rows[0][0] == 1:
|
||
return True
|
||
|
||
return False
|
||
except Exception as e:
|
||
print(f"根据ID获取构件任务失败: {e}")
|
||
return False
|
||
|
||
def exists_by_module_code(self, module_code: str) -> bool:
|
||
"""根据模具编号获取构件任务"""
|
||
try:
|
||
sql = "SELECT count(1) FROM ArtifactTask WHERE MouldCode = ? and OptTime>?"
|
||
results = self.db_dao.execute_read(sql, (module_code,datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)))
|
||
rows = list(results)
|
||
if rows[0][0] == 1:
|
||
return True
|
||
|
||
return False
|
||
except Exception as e:
|
||
print(f"根据ID获取构件任务失败: {e}")
|
||
return False
|
||
|
||
def get_artifactid(self, module_code: str) -> Optional[str]:
|
||
"""根据模具编号查找ID，根据ID确认是否扫码"""
|
||
try:
|
||
sql = "SELECT ArtifactID FROM ArtifactTask WHERE MouldCode = ? and OptTime>?"
|
||
_target_time = datetime.now() - timedelta(hours=4)
|
||
results = self.db_dao.fetch_one(sql, (module_code,_target_time))
|
||
if results is None:
|
||
return None
|
||
else:
|
||
return results['ArtifactID']
|
||
except Exception as e:
|
||
print(f"根据ID获取构件任务失败: {e}")
|
||
return None
|
||
|
||
def get_by_id(self, artifact_id: int) -> Optional[ArtifactInfoModel]:
|
||
"""根据构件ID获取构件任务"""
|
||
try:
|
||
sql = "SELECT * FROM ArtifactTask WHERE ArtifactID = ?"
|
||
results = self.db_dao.execute_read(sql, (artifact_id,))
|
||
|
||
rows = list(results)
|
||
if not rows:
|
||
return None
|
||
|
||
return ArtifactInfoModel(**dict(rows[0]))
|
||
except Exception as e:
|
||
print(f"根据ID获取构件任务失败: {e}")
|
||
return None
|
||
|
||
|
||
def insert_artifact(self, model: ArtifactInfoModel) -> int:
|
||
"""插入一条构件任务记录"""
|
||
try:
|
||
# 使用insert方法插入数据
|
||
model.OptTime=datetime.now()
|
||
model.__dict__.pop("ProductionProcessCode", None)
|
||
model.__dict__.pop("ID", None)
|
||
row_id = self.db_dao.insert("ArtifactTask", model.__dict__)
|
||
return row_id
|
||
except Exception as e:
|
||
print(f"插入构件任务失败: {e}")
|
||
return 0
|
||
|
||
def update_artifact(self, artifact_id: int, update_data: dict) -> int:
|
||
"""更新构件任务记录"""
|
||
try:
|
||
# 构建WHERE条件
|
||
where_condition = f"ArtifactID='{artifact_id}'"
|
||
# 使用update方法更新数据
|
||
affected_rows = self.db_dao.update("ArtifactTask", update_data, where_condition)
|
||
return affected_rows
|
||
except Exception as e:
|
||
print(f"更新构件任务失败: {e}")
|
||
return 0
|
||
|
||
def update_by_modulecode(self, module_code: str, update_data: dict) -> int:
|
||
"""更新构件任务记录"""
|
||
try:
|
||
# 构建WHERE条件
|
||
_target_time = datetime.now() - timedelta(hours=4)
|
||
where_condition = f"MouldCode='{module_code}' and OptTime>='{_target_time}'"
|
||
# 使用update方法更新数据
|
||
affected_rows = self.db_dao.update("ArtifactTask", update_data, where_condition)
|
||
return affected_rows
|
||
except Exception as e:
|
||
print(f"更新构件任务失败: {e}")
|
||
return 0
|
||
|
||
def validate_artifact(self, artifact_info: ArtifactInfoModel) -> bool:
|
||
"""验证构件信息是否符合业务规则"""
|
||
try:
|
||
# 检查必要字段
|
||
if not hasattr(artifact_info, 'Name') or not artifact_info.Name:
|
||
return False
|
||
|
||
# 可以添加更多业务规则验证
|
||
# 例如：检查构件编号格式、验证日期等
|
||
|
||
return True
|
||
except Exception as e:
|
||
print(f"验证构件信息失败: {e}")
|
||
return False
|
||
|
||
class PDRecordDal(BaseDal):
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
def get_top_pd(self, top: int,desc:str="ID desc",where:str="1=1") -> List[PDRecordModel]:
|
||
"""获取top条数数据，根据ID降序"""
|
||
try:
|
||
# 确保top为正整数
|
||
if not isinstance(top, int) or top <= 0:
|
||
raise ValueError("top参数必须是正整数")
|
||
|
||
# 查询指定数量的记录，按ID降序排列
|
||
sql = f"SELECT * FROM PDRecord WHERE {where} ORDER BY {desc} LIMIT ?"
|
||
results = self.db_dao.execute_read(sql, (top,))
|
||
|
||
pdrecords = []
|
||
for row in results:
|
||
#
|
||
pdrecord = PDRecordModel()
|
||
pdrecord.ID=row["ID"]
|
||
pdrecord.ArtifactID=row["ArtifactID"]
|
||
pdrecord.ArtifactActionID=row["ArtifactActionID"]
|
||
pdrecord.TaskID=row["TaskID"]
|
||
pdrecord.ProjectName=row["ProjectName"]
|
||
pdrecord.ProduceMixID=row["ProduceMixID"]
|
||
pdrecord.FBetonVolume=row["FBetonVolume"]
|
||
pdrecord.BetonVolume=row["BetonVolume"]
|
||
pdrecord.MouldCode=row["MouldCode"]
|
||
pdrecord.SkeletonID=row["SkeletonID"]
|
||
pdrecord.RingTypeCode=row["RingTypeCode"]
|
||
pdrecord.SizeSpecification=row["SizeSpecification"]
|
||
pdrecord.BuriedDepth=row["BuriedDepth"]
|
||
pdrecord.Mode=row["Mode"]
|
||
pdrecord.Status=row["Status"]
|
||
pdrecord.GStatus=row["GStatus"]
|
||
pdrecord.Source=row["Source"]
|
||
pdrecord.CreateTime=row["CreateTime"]
|
||
pdrecord.OptTime=row["OptTime"]
|
||
pdrecords.append(pdrecord)
|
||
|
||
return pdrecords
|
||
except Exception as e:
|
||
print(f"获取top PD官片任务失败: {e}")
|
||
return []
|
||
|
||
def get_last_pds(self, top: int,mould_code:str) -> List[PDRecordModel]:
|
||
"""获取派单的记录进行派单"""
|
||
try:
|
||
# 确保top为正整数
|
||
if not isinstance(top, int) or top <= 0:
|
||
raise ValueError("top参数必须是正整数")
|
||
_target_time = datetime.now() - timedelta(hours=4)
|
||
# _target_time ='2026-03-31 00:00:00'
|
||
# 查询当前浇筑模具编号的前一条记录
|
||
sql = f"""SELECT * FROM PDRecord WHERE CreateTime>? and ID>(
|
||
select ID FROM PDRecord WHERE MouldCode=? and CreateTime>?
|
||
) order by ID asc LIMIT ?"""
|
||
results = self.db_dao.execute_read(sql, (_target_time,mould_code,_target_time,top))
|
||
|
||
pdrecords = []
|
||
for row in results:
|
||
pdrecord = PDRecordModel(**row)
|
||
pdrecords.append(pdrecord)
|
||
return pdrecords
|
||
except Exception as e:
|
||
print(f"获取top PD官片任务失败: {e}")
|
||
return []
|
||
|
||
def get_artifactid(self, module_code: str) -> Optional[str]:
|
||
"""根据模具编号查找ID，根据ID确认是否扫码"""
|
||
try:
|
||
sql = "SELECT ArtifactID FROM PDRecord WHERE MouldCode = ? and CreateTime>?"
|
||
_target_time = datetime.now() - timedelta(hours=4)
|
||
results = self.db_dao.fetch_one(sql, (module_code,_target_time))
|
||
if results is None:
|
||
return None
|
||
else:
|
||
return results['ArtifactID']
|
||
except Exception as e:
|
||
print(f"根据ID获取构件任务失败: {e}")
|
||
return None
|
||
|
||
def exists_by_module_code(self, module_code: str) -> bool:
|
||
"""根据模具编号获取PD官片任务"""
|
||
try:
|
||
sql = "SELECT count(1) FROM PDRecord WHERE MouldCode = ? and OptTime>?"
|
||
results = self.db_dao.execute_read(sql, (module_code,datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)))
|
||
|
||
rows = list(results)
|
||
if rows[0][0] == 1:
|
||
return True
|
||
|
||
return False
|
||
except Exception as e:
|
||
print(f"PDRecord(exists_by_module_code)失败: {e}")
|
||
return False
|
||
|
||
def exists_artifactid(self, artifact_action_id: str) -> bool:
|
||
"""根据构件动作编号获取PD官片任务"""
|
||
try:
|
||
sql = "SELECT count(1) FROM PDRecord WHERE ArtifactActionID = ?"
|
||
results = self.db_dao.execute_read(sql, (artifact_action_id,))
|
||
|
||
rows = list(results)
|
||
if rows[0][0] == 1:
|
||
return True
|
||
|
||
return False
|
||
except Exception as e:
|
||
print(f"PDRecord(exists_by_module_code)失败: {e}")
|
||
return False
|
||
|
||
def insert_PD_record(self, model: PDRecordModel) -> int:
|
||
"""插入一条PD官片任务记录"""
|
||
try:
|
||
# 使用insert方法插入数据
|
||
model.CreateTime=datetime.now()
|
||
model.__dict__.pop("ID", None)
|
||
row_id = self.db_dao.insert("PDRecord", model.__dict__)
|
||
return row_id
|
||
except Exception as e:
|
||
print(f"插入PD官片任务失败: {e}")
|
||
return 0
|
||
|
||
def update_pd(self, artifact_action_id: int, update_data: dict) -> bool:
|
||
"""更新PD官片任务记录"""
|
||
try:
|
||
# 构建WHERE条件
|
||
where_condition = f"ArtifactActionID='{artifact_action_id}'"
|
||
# 使用update方法更新数据
|
||
affected_rows = self.db_dao.update("PDRecord", update_data, where_condition)
|
||
return affected_rows > 0
|
||
except Exception as e:
|
||
print(f"更新PD管片任务失败: {e}")
|
||
return False
|
||
|
||
def update_pd_byid(self, id: int, update_data: dict) -> bool:
|
||
"""更新PD官片任务记录"""
|
||
try:
|
||
# 构建WHERE条件
|
||
where_condition = f"ID={id}"
|
||
# 使用update方法更新数据
|
||
affected_rows = self.db_dao.update("PDRecord", update_data, where_condition)
|
||
return affected_rows > 0
|
||
except Exception as e:
|
||
print(f"更新PD管片任务失败: {e}")
|
||
return False
|
||
|
||
def update_pd_notify(self, id: int, update_data: dict) -> bool:
|
||
"""更新PD官片任务记录"""
|
||
try:
|
||
# 构建WHERE条件
|
||
where_condition = f"ID={id} and Status<>5"
|
||
# 使用update方法更新数据
|
||
affected_rows = self.db_dao.update("PDRecord", update_data, where_condition)
|
||
return affected_rows > 0
|
||
except Exception as e:
|
||
raise
|
||
|
||
def update_by_modulecode(self, module_code: str, update_data: dict) -> bool:
|
||
"""更新构件派单记录"""
|
||
try:
|
||
# 构建WHERE条件
|
||
_target_time = datetime.now() - timedelta(hours=4)
|
||
# _target_time ='2026-02-06 00:00:00'
|
||
where_condition = f"MouldCode='{module_code}' and CreateTime>='{_target_time}'"
|
||
# 使用update方法更新数据
|
||
affected_rows = self.db_dao.update("PDRecord", update_data, where_condition)
|
||
return affected_rows > 0
|
||
except Exception as e:
|
||
print(f"更新PD官片任务失败: {e}")
|
||
return False
|
||
|
||
class FreqRecordDal(BaseDal):
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
def get_all_sync(self) -> List[FreqRecordModel]:
|
||
"""获取未同步的记录"""
|
||
try:
|
||
# 查询所有记录
|
||
sql = "SELECT * FROM FreqRecord WHERE IsSync=0"
|
||
results = self.db_dao.execute_read(sql)
|
||
|
||
# 将查询结果转换为FreqRecord对象列表
|
||
artifacts = []
|
||
for row in results:
|
||
# 过滤字典，只保留模型中定义的字段
|
||
# filtered_data = filter_dict_for_model(dict(row), ArtifactInfoModel)
|
||
artifact = FreqRecordModel(**row)
|
||
artifacts.append(artifact)
|
||
|
||
return artifacts
|
||
except Exception as e:
|
||
print(f"获取所有频率记录失败: {e}")
|
||
return []
|
||
|
||
|
||
def insert_record(self, model: FreqRecordModel) -> int:
|
||
"""插入一条频率记录"""
|
||
try:
|
||
# 使用insert方法插入数据
|
||
model.OptTime=datetime.now()
|
||
model.__dict__.pop("ID", None)
|
||
row_id = self.db_dao_record.insert("FreqRecord", model.__dict__)
|
||
return row_id
|
||
except Exception as e:
|
||
print(f"插入频率记录失败: {e}")
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
artifact_dal = ArtifactDal()
|
||
|
||
artifacts = artifact_dal.get_top_artifact(5, "ID desc")
|
||
|
||
# 显示获取到的数据
|
||
print(f"获取到 {len(artifacts)} 条构件任务数据：")
|
||
for i, artifact in enumerate(artifacts, 1):
|
||
print(f"\n记录 {i}:")
|
||
print(f"ID: {artifact.ID}")
|
||
print(f"ArtifactID: {artifact.ArtifactID}")
|
||
print(f"ProduceRingNumber: {artifact.ProduceRingNumber}")
|
||
print(f"MouldCode: {artifact.MouldCode}")
|
||
print(f"SkeletonID: {artifact.SkeletonID}")
|
||
print(f"RingTypeCode: {artifact.RingTypeCode}")
|
||
print(f"SizeSpecification: {artifact.SizeSpecification}")
|
||
print(f"BuriedDepth: {artifact.BuriedDepth}")
|
||
print(f"BlockNumber: {artifact.BlockNumber}")
|
||
print(f"HoleRingMarking: {artifact.HoleRingMarking}")
|
||
print(f"GroutingPipeMarking: {artifact.GroutingPipeMarking}")
|
||
print(f"PolypropyleneFiberMarking: {artifact.PolypropyleneFiberMarking}")
|
||
print(f"BetonVolume: {artifact.BetonVolume}")
|
||
print(f"BetonTaskID: {artifact.BetonTaskID}")
|
||
print(f"FK_PDID: {artifact.FK_PDID}")
|
||
print(f"Status: {artifact.Status}")
|
||
print(f"BeginTime: {artifact.BeginTime}")
|
||
print(f"EndTime: {artifact.EndTime}")
|
||
print(f"PStatus: {artifact.PStatus}")
|
||
print(f"Source: {artifact.Source}")
|
||
print(f"FBetonVolume: {artifact.FBetonVolume}")
|
||
print(f"OptTime: {artifact.OptTime}")
|
||
|
||
# 可以打印更多字段...
|
||
|
||
print('\ntest success') |