完整功能

database/sql_server_connection.py

@@ -0,0 +1,63 @@
+# database/sql_server_connection.py
+"""SQL Server数据库连接模块"""
+import pyodbc
+from config.settings import SQL_SERVER_CONFIG
+
+class SQLServerConnection:
+    def __init__(self):
+        self.config = SQL_SERVER_CONFIG
+
+    def connect(self):
+        """建立数据库连接"""
+        connection_string = (
+            f"DRIVER={self.config['driver']};"
+            f"SERVER={self.config['server']};"
+            f"DATABASE={self.config['database']};"
+            f"UID={self.config['username']};"
+            f"PWD={self.config['password']};"
+        )
+        return pyodbc.connect(connection_string)
+
+    def get_mix_weight_info(self):
+        """
+        获取所有配比号及其对应每方重量的信息
+        重量计算方式：U1到U17的和，NULL值不计入
+        """
+        connection = self.connect()
+        cursor = connection.cursor()
+
+        try:
+            # 查询所有配比信息
+            query = """
+            SELECT 
+                Code, 
+                U1, U2, U3, U4, U5, U6, U7, U8, U9, U10, U11, U12, U13, U14, U15, U16, U17
+            FROM MixTable 
+            WHERE Code IS NOT NULL
+            ORDER BY Code
+            """
+
+            cursor.execute(query)
+            results = cursor.fetchall()
+
+            mix_info = []
+            for row in results:
+                code = row[0]
+                weights = [row[i] for i in range(1, 18)]  # U1到U17
+
+                # 计算总重量（忽略NULL值）
+                total_weight = sum(weight for weight in weights if weight is not None and weight != 'NULL')
+
+                mix_info.append({
+                    "Code": code,
+                    "TotalWeight": round(total_weight, 2)  # 保留两位小数
+                })
+
+            return mix_info
+
+        except Exception as e:
+            print(f"查询配比信息时出错: {e}")
+            return []
+
+        finally:
+            connection.close()