Feeding_control_system/services/task_service.py

"""任务处理服务"""
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
import threading
import time


class TaskService:
    def __init__(self, tcp_server=None):
        self.api_client = APIClient()
        self.half_volume = [0, 0]
        self.task_before = {"block_number":None, "beton_volume":None, "artifact_id":None}
        self.artifact_timestamps = {}
        self.tcp_server = tcp_server

    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])

      # 根据F块情况处理任务
        task_result = self._handle_tasks_by_f_blocks(
            f_block_count, f_positions, current_task,
            f_blocks, total_f_volume, artifact_list
        )
        # 更新上一个任务信息
        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"]
        }
        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块的情况"""
        if self.task_before.get("block_number") == "F":
            print("报警:,超出正常补块逻辑")
        else:
            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 = "补方"
        elif f_positions == [1, 2]:
            adjusted_volume = artifact_list[0]["BetonVolume"]
            block_number = current_task["block_number"]
        elif f_positions == [0, 1] and self.task_before.get("block_number") != "F":
            adjusted_volume = total_f_volume - self.half_volume[0]
            block_number = current_task["block_number"]
        else:
            print("报警")

        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()

    def insert_into_produce_table(self, connection, task_info, beton_volume, erp_id, artifact_id, status):
        """插入数据到Produce表"""
        sql_db = SQLServerDB()
        if status == 1:
            # 准备插入数据
            insert_data = {
                "ErpID": erp_id,
                "Code": task_info["TaskID"],
                "DatTim": datetime.now(),
                "Recipe": task_info["ProduceMixID"],
                "MorRec": "",
                "ProdMete": beton_volume,
                "MorMete": 0.0,  # 砂浆方量，根据实际需求填写
                "TotVehs": 0,  # 累计车次，根据实际需求填写
                "TotMete": task_info["PlannedVolume"],  # 累计方量
                "Qualitor": "",  # 质检员，根据实际需求填写
                "Acceptor": "",  # 现场验收，根据实际需求填写
                "Attamper": "",  # 调度员，根据实际需求填写
                "Flag": "1",  # 标识，根据实际需求填写
                "Note": ""  # 备注，根据实际需求填写
            }

            sql_db.insert_produce_data(insert_data)
            print(f"数据已成功插入到Produce表中，ERP ID: {erp_id}")

            # 调用同事提供的保存函数，将数据保存到自定义数据表
            try:
                # 假设同事提供的函数名为 save_to_custom_table
                # 参数包括: MISID(即erp_id), Flag, TaskID, ProduceMixID, ProjectName, BetonGrade, 调整后的方量
                self.save_to_custom_table(
                    misid=erp_id,
                    flag="已插入",  # 初始Flag值
                    task_id=task_info["TaskID"],
                    produce_mix_id=task_info["ProduceMixID"],
                    project_name=task_info["ProjectName"],
                    beton_grade=task_info["BetonGrade"],
                    adjusted_volume=round(beton_volume,2),
                    artifact_id=artifact_id
                    # 已经调整后的方量
                )
                print(f"任务 {erp_id} 的数据已保存到自定义数据表")
            except Exception as e:
                print(f"调用保存函数时出错: {e}")

            # 发送数据给TCP客户端
            if self.tcp_server:
                try:
                    time.sleep(2)
                    task_data = {
                        "erp_id": erp_id,  # 车号，相当于序号
                        "task_id": task_info["TaskID"],  # 任务单号
                        "artifact_id": artifact_id,
                        "produce_mix_id": task_info["ProduceMixID"],  # 配比号
                        "project_name": task_info["ProjectName"],  # 任务名
                        "beton_grade": task_info["BetonGrade"],  # 砼强度
                        "adjusted_volume": round(beton_volume,2),  # 方量
                        "flag": "已插入",  # 状态
                        "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")  # 时间
                    }
                    self.tcp_server.send_data(task_data)
                    print(f"任务 {artifact_id} 的数据已发送给TCP客户端")
                except Exception as e:
                    print(f"发送数据给TCP客户端时出错: {e}")

            return erp_id
        else:
            try:
                # 参数包括: MISID(即erp_id), Flag, TaskID, ProduceMixID, ProjectName, BetonGrade, 调整后的方量
                self.save_to_custom_table(
                    misid=erp_id,
                    flag="已插入",  # 初始Flag值
                    task_id=task_info["TaskID"],
                    produce_mix_id=task_info["ProduceMixID"],
                    project_name=task_info["ProjectName"],
                    beton_grade=task_info["BetonGrade"],
                    adjusted_volume=round(beton_volume,2),
                    artifact_id=artifact_id
                )
                print(f"任务 {erp_id} 的数据已保存到自定义数据表")
            except Exception as e:
                print(f"调用保存函数时出错: {e}")


            return erp_id

    def save_to_custom_table(self,misid, flag, task_id, produce_mix_id, project_name, beton_grade, adjusted_volume, artifact_id):
        print(f"保存到自定义数据表: MISID={misid}, Flag={flag}, TaskID={task_id}, 调整后方量={adjusted_volume}")

    def update_custom_table_status(self, erp_id, status):
        print(f"更新自定义数据表状态: ERP ID={erp_id}, 状态={status}")