完整功能

main.py

@@ -15,6 +15,12 @@ from config.settings import (
 )
 from utils.helpers import cleanup_old_timestamps
 
+# 假设同事提供的函数
+def save_to_custom_table(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 start_api_service():
     """启动配比重量API服务"""
     api = MixWeightAPI()
@@ -83,26 +89,18 @@ def main():
                                 # 步骤5：连接SQL Server数据库并插入数据
                                 sql_db = SQLServerDB()
                                 try:
-                                    # 准备插入数据
-                                    insert_data = {
-                                        "ErpID": erp_id,
-                                        "Code": task_info["TaskID"],
-                                        "DatTim": datetime.now(),
-                                        "Recipe": task_info["ProduceMixID"],
-                                        "MorRec": "",
-                                        "ProdMete": task["beton_volume"],
-                                        "MorMete": 0.0,
-                                        "TotVehs": 0,
-                                        "TotMete": task_info["PlannedVolume"],
-                                        "Qualitor": "",
-                                        "Acceptor": "",
-                                        "Attamper": "",
-                                        "Flag": "1",
-                                        "Note": ""
-                                    }
+                                    # 检查 block_number 是否为 "补方"
+                                    if task["block_number"] == "补方":
+                                        print(f"任务 {task['artifact_id']} 的 block_number 为 '补方'，跳过派单")
+                                        task_service.insert_into_produce_table(sql_db, task_info, task["beton_volume"], erp_id,
+                                                                              task["artifact_id"], 0)
+                                        continue
 
-                                    sql_db.insert_produce_data(insert_data)
-                                    print(f"数据已成功插入到Produce表中，ERP ID: {erp_id}")
+                                    print(f"处理新任务: {task['artifact_id']}")
+
+                                    # 步骤3：获取任务单信息
+                                    task_service.insert_into_produce_table(sql_db, task_info, task["beton_volume"], erp_id,
+                                                                          task["artifact_id"], 1)
                                 finally:
                                     sql_db.close()