304 lines
13 KiB
Python
304 lines
13 KiB
Python
"""任务处理服务"""
|
|
from datetime import datetime
|
|
from API.client import APIClient
|
|
from database.access_db import AccessDB
|
|
from database.sql_server import SQLServerDB
|
|
from config.settings import ACCESS_DB_PATH, ACCESS_DB_PASSWORD
|
|
from utils.helpers import get_f_block_positions
|
|
|
|
|
|
class TaskService:
|
|
def __init__(self):
|
|
self.api_client = APIClient()
|
|
self.half_volume = [0, 0]
|
|
self.task_before = None
|
|
self.artifact_timestamps = {}
|
|
|
|
def process_not_pour_info(self):
|
|
"""处理未浇筑信息"""
|
|
artifact_list = self.api_client.get_not_pour_info()
|
|
|
|
if not artifact_list:
|
|
return [], [], [], self.half_volume
|
|
|
|
# 处理F块信息
|
|
f_blocks_info = self._process_f_blocks(artifact_list)
|
|
f_blocks = f_blocks_info["f_blocks"]
|
|
f_block_count = f_blocks_info["f_block_count"]
|
|
total_f_volume = f_blocks_info["total_f_volume"]
|
|
f_positions = f_blocks_info["f_positions"]
|
|
|
|
# 处理当前任务
|
|
current_task = self._process_current_task(artifact_list[0])
|
|
|
|
# 更新上一个任务信息
|
|
self.task_before = {
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": current_task["beton_volume"],
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": current_task["block_number"]
|
|
}
|
|
|
|
# 根据F块情况处理任务
|
|
task_result = self._handle_tasks_by_f_blocks(
|
|
f_block_count, f_positions, current_task,
|
|
f_blocks, total_f_volume, artifact_list
|
|
)
|
|
|
|
return task_result
|
|
|
|
def _process_f_blocks(self, artifact_list):
|
|
"""处理F块相关信息"""
|
|
f_blocks = [artifact for artifact in artifact_list if artifact.get("BlockNumber") == "F"]
|
|
f_block_count = len(f_blocks)
|
|
total_f_volume = sum(artifact["BetonVolume"] for artifact in f_blocks)
|
|
f_positions = get_f_block_positions(artifact_list)
|
|
|
|
return {
|
|
"f_blocks": f_blocks,
|
|
"f_block_count": f_block_count,
|
|
"total_f_volume": total_f_volume,
|
|
"f_positions": f_positions
|
|
}
|
|
|
|
def _process_current_task(self, latest_artifact):
|
|
"""处理当前任务信息"""
|
|
task_data = self.api_client.get_task_info(latest_artifact["BetonTaskID"])
|
|
return {
|
|
"beton_task_id": latest_artifact["BetonTaskID"],
|
|
"beton_volume": latest_artifact["BetonVolume"],
|
|
"artifact_id": latest_artifact["ArtifactActionID"],
|
|
"block_number": latest_artifact.get("BlockNumber", ""),
|
|
"task_data": task_data
|
|
}
|
|
|
|
def _handle_tasks_by_f_blocks(self, f_block_count, f_positions, current_task,
|
|
f_blocks, total_f_volume, artifact_list):
|
|
"""根据F块数量和位置处理任务"""
|
|
# 多个F块情况
|
|
if f_block_count > 2:
|
|
return self._handle_multiple_f_blocks(current_task, total_f_volume, artifact_list)
|
|
|
|
# 两个F块情况
|
|
elif f_block_count == 2:
|
|
return self._handle_two_f_blocks(f_positions, current_task, total_f_volume, artifact_list)
|
|
|
|
# 一个F块情况
|
|
elif f_block_count == 1:
|
|
return self._handle_single_f_block(f_positions, current_task, f_blocks,
|
|
total_f_volume, artifact_list)
|
|
|
|
# 无F块情况
|
|
elif f_block_count == 0:
|
|
return self._handle_no_f_blocks(current_task, artifact_list)
|
|
|
|
else:
|
|
print("报警")
|
|
return [], [], [], self.half_volume
|
|
|
|
def _handle_multiple_f_blocks(self, current_task, total_f_volume, artifact_list):
|
|
"""处理多个F块的情况"""
|
|
adjusted_volume = total_f_volume - self.half_volume[0]
|
|
|
|
tasks = [{
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": adjusted_volume,
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": current_task["block_number"],
|
|
}]
|
|
|
|
send_list = [{
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": adjusted_volume,
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": current_task["block_number"],
|
|
"beton_grade": current_task["task_data"]["BetonGrade"],
|
|
"mix_id": current_task["task_data"]["MixID"],
|
|
"time": self.artifact_timestamps.get(current_task["artifact_id"], datetime.now())
|
|
}]
|
|
|
|
# 处理后续任务
|
|
self._append_additional_tasks(send_list, artifact_list, [0, 0])
|
|
|
|
# 更新时间戳
|
|
self._update_artifact_timestamps(send_list)
|
|
|
|
return tasks, artifact_list, send_list, self.half_volume
|
|
|
|
def _handle_two_f_blocks(self, f_positions, current_task, total_f_volume, artifact_list):
|
|
"""处理两个F块的情况"""
|
|
if f_positions == [0, 1] and self.task_before.get("block_number") == "F":
|
|
adjusted_volume = 0
|
|
block_number = "补方"
|
|
else:
|
|
adjusted_volume = total_f_volume - self.half_volume[0]
|
|
block_number = current_task["block_number"]
|
|
|
|
tasks = [{
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": adjusted_volume,
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": block_number,
|
|
}]
|
|
|
|
send_list = [{
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": adjusted_volume,
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": block_number,
|
|
"beton_grade": current_task["task_data"]["BetonGrade"],
|
|
"mix_id": current_task["task_data"]["MixID"],
|
|
"time": self.artifact_timestamps.get(current_task["artifact_id"], datetime.now())
|
|
}]
|
|
|
|
# 处理后续任务
|
|
volumes = [adjusted_volume,
|
|
artifact_list[2]["BetonVolume"] if len(artifact_list) > 2 else 0] if f_positions == [0,
|
|
1] and self.task_before.get(
|
|
"block_number") == "F" else [0, 0]
|
|
self._append_additional_tasks(send_list, artifact_list, volumes)
|
|
|
|
# 更新时间戳
|
|
self._update_artifact_timestamps(send_list)
|
|
|
|
return tasks, artifact_list, send_list, self.half_volume
|
|
|
|
def _handle_single_f_block(self, f_positions, current_task, f_blocks, total_f_volume, artifact_list):
|
|
"""处理单个F块的情况"""
|
|
if f_positions == [2]:
|
|
f_volume = f_blocks[0].get("BetonVolume") if f_blocks else 0
|
|
self.half_volume[0] = round(total_f_volume / 2, 2)
|
|
self.half_volume[1] = f_volume - self.half_volume[0]
|
|
adjusted_volume = current_task["beton_volume"] + self.half_volume[0]
|
|
elif f_positions == [1]:
|
|
adjusted_volume = current_task["beton_volume"] + self.half_volume[1]
|
|
elif f_positions == [0]:
|
|
adjusted_volume = 0
|
|
else:
|
|
adjusted_volume = current_task["beton_volume"]
|
|
|
|
block_number = "补方" if f_positions == [0] else current_task["block_number"]
|
|
|
|
tasks = [{
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": adjusted_volume,
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": block_number,
|
|
}]
|
|
|
|
send_list = [{
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": adjusted_volume,
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": block_number,
|
|
"beton_grade": current_task["task_data"]["BetonGrade"],
|
|
"mix_id": current_task["task_data"]["MixID"],
|
|
"time": self.artifact_timestamps.get(current_task["artifact_id"], datetime.now())
|
|
}]
|
|
|
|
# 处理后续任务
|
|
self._append_additional_tasks_for_single_f(send_list, artifact_list, f_positions)
|
|
|
|
# 更新时间戳
|
|
self._update_artifact_timestamps(send_list)
|
|
|
|
return tasks, artifact_list, send_list, self.half_volume
|
|
|
|
def _handle_no_f_blocks(self, current_task, artifact_list):
|
|
"""处理无F块的情况"""
|
|
tasks = [{
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": current_task["beton_volume"],
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": current_task["block_number"],
|
|
}]
|
|
|
|
send_list = [{
|
|
"beton_task_id": current_task["beton_task_id"],
|
|
"beton_volume": current_task["beton_volume"],
|
|
"artifact_id": current_task["artifact_id"],
|
|
"block_number": current_task["block_number"],
|
|
"beton_grade": current_task["task_data"]["BetonGrade"],
|
|
"mix_id": current_task["task_data"]["MixID"],
|
|
"time": self.artifact_timestamps.get(current_task["artifact_id"], datetime.now())
|
|
}]
|
|
|
|
# 处理后续任务
|
|
self._append_additional_tasks(send_list, artifact_list,
|
|
[artifact_list[2]["BetonVolume"] if len(artifact_list) > 2 else 0,
|
|
artifact_list[2]["BetonVolume"] if len(artifact_list) > 2 else 0])
|
|
|
|
# 更新时间戳
|
|
self._update_artifact_timestamps(send_list)
|
|
|
|
return tasks, artifact_list, send_list, self.half_volume
|
|
|
|
def _append_additional_tasks(self, send_list, artifact_list, volumes):
|
|
"""添加额外的任务到发送列表"""
|
|
# 安全获取后续任务
|
|
second_task = artifact_list[1] if len(artifact_list) > 1 else None
|
|
third_task = artifact_list[2] if len(artifact_list) > 2 else None
|
|
|
|
if second_task:
|
|
task_data_second = self.api_client.get_task_info(second_task["BetonTaskID"])
|
|
send_list.append({
|
|
"beton_task_id": second_task["BetonTaskID"],
|
|
"beton_volume": volumes[0],
|
|
"artifact_id": second_task["ArtifactActionID"],
|
|
"block_number": second_task["BlockNumber"],
|
|
"beton_grade": task_data_second["BetonGrade"],
|
|
"mix_id": task_data_second["MixID"],
|
|
"time": self.artifact_timestamps.get(second_task["ArtifactActionID"], datetime.now())
|
|
})
|
|
|
|
if third_task:
|
|
task_data_third = self.api_client.get_task_info(third_task["BetonTaskID"])
|
|
send_list.append({
|
|
"beton_task_id": third_task["BetonTaskID"],
|
|
"beton_volume": volumes[1],
|
|
"artifact_id": third_task["ArtifactActionID"],
|
|
"block_number": third_task["BlockNumber"],
|
|
"beton_grade": task_data_third["BetonGrade"],
|
|
"mix_id": task_data_third["MixID"],
|
|
"time": self.artifact_timestamps.get(third_task["ArtifactActionID"], datetime.now())
|
|
})
|
|
|
|
def _append_additional_tasks_for_single_f(self, send_list, artifact_list, f_positions):
|
|
"""为单个F块情况添加额外任务"""
|
|
second_task = artifact_list[1] if len(artifact_list) > 1 else None
|
|
third_task = artifact_list[2] if len(artifact_list) > 2 else None
|
|
|
|
if second_task:
|
|
task_data_second = self.api_client.get_task_info(second_task["BetonTaskID"])
|
|
volume = (third_task["BetonVolume"] if third_task else 0) if f_positions != [2] else (
|
|
second_task["BetonVolume"] + self.half_volume[1])
|
|
send_list.append({
|
|
"beton_task_id": second_task["BetonTaskID"],
|
|
"beton_volume": volume,
|
|
"artifact_id": second_task["ArtifactActionID"],
|
|
"block_number": second_task["BlockNumber"],
|
|
"beton_grade": task_data_second["BetonGrade"],
|
|
"mix_id": task_data_second["MixID"],
|
|
"time": self.artifact_timestamps.get(second_task["ArtifactActionID"], datetime.now())
|
|
})
|
|
|
|
if third_task:
|
|
task_data_third = self.api_client.get_task_info(third_task["BetonTaskID"])
|
|
volume = third_task["BetonVolume"] if f_positions in [[1], [0]] else 0
|
|
send_list.append({
|
|
"beton_task_id": third_task["BetonTaskID"],
|
|
"beton_volume": volume,
|
|
"artifact_id": third_task["ArtifactActionID"],
|
|
"block_number": third_task["BlockNumber"],
|
|
"beton_grade": task_data_third["BetonGrade"],
|
|
"mix_id": task_data_third["MixID"],
|
|
"time": self.artifact_timestamps.get(third_task["ArtifactActionID"], datetime.now())
|
|
})
|
|
|
|
def _update_artifact_timestamps(self, send_list):
|
|
"""更新artifact时间戳"""
|
|
current_artifact_ids = {item["artifact_id"] for item in send_list}
|
|
for artifact_id in current_artifact_ids:
|
|
if artifact_id not in self.artifact_timestamps:
|
|
self.artifact_timestamps[artifact_id] = datetime.now()
|