Feeding_control_system/busisness/blls.py

# import os
# import sys

# # 添加项目根目录到Python路径
# sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))

from dataclasses import fields
from typing import List, Optional, Dict, Any
from datetime import datetime
from .models import ArtifactInfoModel, PDRecordModel
from .dals import ArtifactDal, PDRecordDal


class ArtifactBll:
    def __init__(self):
        """初始化数据访问层，创建数据库连接"""
        # 假设数据库文件在db目录下
        self.dal = ArtifactDal()
        pass

    def get_artifact_task(self) -> List[ArtifactInfoModel]:
        """获取官片任务数据"""
        return self.dal.get_top_artifact(5, "ArtifactID asc")


    def get_artifacting_task(self) -> ArtifactInfoModel:
            """获取正在进行的官片任务数据"""
            loc_item= self.dal.get_top_artifact(1,"","Status=2")
            if loc_item:
                return loc_item[0]
            else:
                return None

class PDRecordBll:
    def __init__(self):
        """初始化数据访问层，创建数据库连接"""
        # 假设数据库文件在db目录下
        self.dal = PDRecordDal()
        pass

    def get_PD_record(self) -> List[PDRecordModel]:
        """获取PD官片任务数据"""
        return self.dal.get_top_pd(5, "ID desc")


if __name__ == "__main__":
    artifact_dal = ArtifactBll()

    artifacts = artifact_dal.get_artifact_task()
    
    print("\n打印artifacts数据:")
    for i, artifact in enumerate(artifacts):
        artifact_id = artifact.ArtifactID
        # 如果是数据类对象，转换为字典输出
        print(artifact.MouldCode)
        if hasattr(artifact, "__dataclass_fields__"):
            print(f"第{i+1}条: {artifact.__dict__}")
        else:
            print(f"第{i+1}条: {artifact}")

    pdrecord_dal = PDRecordBll()
    pdrecords = pdrecord_dal.get_PD_record()
    # print(pdrecords[0].MouldCode)
    print("\n打印pdrecords数据:")
    for i, record in enumerate(pdrecords):
        # 如果是数据类对象，转换为字典输出
        # print(record.__dict__["MouldCode"])
        if hasattr(record, "__dataclass_fields__"):
            print(f"第{i+1}条: {record.__dict__}")
        else:
            print(f"第{i+1}条: {record}")

    print("\ntest success")

    # # 1. 提取 artifacts 中第1-3条的 MouldCode 和对应的 BetonVolume
    # print("=== artifacts 提取结果 ===")
    # artifact_extracts = []
    # for i, artifact in enumerate(artifacts[:3], 1):  # 取前3条（正好3条）
    #     mould_code = artifact['MouldCode']
    #     beton_volume = artifact['BetonVolume']
    #     artifact_extracts.append({'序号': i, 'MouldCode': mould_code, 'BetonVolume': beton_volume})
    #     print(f"第{i}条: MouldCode={mould_code}, BetonVolume={beton_volume}")

    # # 2. 提取 pdrecords 中的所有 BetonVolume
    # print("\n=== pdrecords 提取结果 ===")
    # pd_beton_volumes = []
    # for i, record in enumerate(pdrecords, 1):
    #     beton_volume = record['BetonVolume']
    #     pd_beton_volumes.append({'序号': i, 'BetonVolume': beton_volume})
    #     print(f"第{i}条: BetonVolume={beton_volume}")