cc

busisness/blls.py

@@ -1,8 +1,15 @@
+# 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
+from .models import ArtifactInfoModel, PDRecordModel
+from .dals import ArtifactDal, PDRecordDal
+
 
 class ArtifactBll:
     def __init__(self):
@@ -10,10 +17,11 @@ class ArtifactBll:
         # 假设数据库文件在db目录下
         self.dal = ArtifactDal()
         pass
-     
+
     def get_artifact_task(self) -> List[ArtifactInfoModel]:
-            """获取官片任务数据"""
-            return self.dal.get_top_artifact(5,"ArtifactID asc")
+        """获取官片任务数据"""
+        return self.dal.get_top_artifact(5, "ArtifactID asc")
+
 
     def get_artifacting_task(self) -> ArtifactInfoModel:
             """获取正在进行的官片任务数据"""
@@ -29,35 +37,54 @@ class PDRecordBll:
         # 假设数据库文件在db目录下
         self.dal = PDRecordDal()
         pass
-     
+
     def get_PD_record(self) -> List[PDRecordModel]:
-            """获取PD官片任务数据"""
-            return self.dal.get_top_pd(5,"ID desc")
+        """获取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
         # 如果是数据类对象，转换为字典输出
-        if hasattr(artifact, '__dataclass_fields__'):
+        print(artifact.MouldCode)
+        if hasattr(artifact, "__dataclass_fields__"):
             print(f"第{i+1}条: {artifact.__dict__}")
         else:
             print(f"第{i+1}条: {artifact}")
-    
-    pdrecord_dal=PDRecordBll()
+
+    pdrecord_dal = PDRecordBll()
     pdrecords = pdrecord_dal.get_PD_record()
+    # print(pdrecords[0].MouldCode)
     print("\n打印pdrecords数据:")
     for i, record in enumerate(pdrecords):
         # 如果是数据类对象，转换为字典输出
-        if hasattr(record, '__dataclass_fields__'):
+        # print(record.__dict__["MouldCode"])
+        if hasattr(record, "__dataclass_fields__"):
             print(f"第{i+1}条: {record.__dict__}")
         else:
             print(f"第{i+1}条: {record}")
- 
 
-    print('\ntest success')
+    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}")