#!/usr/bin/env python # -*- coding: utf-8 -*- """ 工单查询模块 - 独立的SQL Server查询功能 """ import json import os from typing import Dict, Optional from pathlib import Path from logger import get_logger logger = get_logger() # 配置文件路径 CONFIG_FILE = Path(__file__).parent / 'work_order_db_config.json' # 默认配置 DEFAULT_CONFIG = { 'debug_mode': False, # Debug模式：True时使用假数据，不连接数据库 'host': 'localhost', # 数据库服务器地址 'port': 1433, # SQL Server端口（通常是1433） 'database': 'MES_DB', # 数据库名 'username': 'sa', # 用户名 'password': 'password', # 密码 'charset': 'utf8mb4' } def load_db_config() -> Dict[str, any]: """ 从配置文件加载数据库配置 如果配置文件不存在，则创建默认配置文件 """ if CONFIG_FILE.exists(): try: with open(CONFIG_FILE, 'r', encoding='utf-8') as f: config = json.load(f) # 只提取需要的配置项 return { 'debug_mode': config.get('debug_mode', DEFAULT_CONFIG['debug_mode']), 'host': config.get('host', DEFAULT_CONFIG['host']), 'port': config.get('port', DEFAULT_CONFIG['port']), 'database': config.get('database', DEFAULT_CONFIG['database']), 'username': config.get('username', DEFAULT_CONFIG['username']), 'password': config.get('password', DEFAULT_CONFIG['password']), 'charset': config.get('charset', DEFAULT_CONFIG['charset']) } except Exception as e: logger.error(f"读取配置文件失败: {e}") return DEFAULT_CONFIG.copy() else: # 创建默认配置文件 try: with open(CONFIG_FILE, 'w', encoding='utf-8') as f: config_with_desc = DEFAULT_CONFIG.copy() config_with_desc['description'] = '工单数据库连接配置' config_with_desc['comment'] = '请根据实际环境修改以上配置' json.dump(config_with_desc, f, ensure_ascii=False, indent=2) logger.info(f"已创建默认配置文件: {CONFIG_FILE}") except Exception as e: logger.error(f"创建配置文件失败: {e}") return DEFAULT_CONFIG.copy() # 工单数据库配置（从文件加载） WORK_ORDER_DB_CONFIG = load_db_config() def reload_config(): """ 重新加载配置文件 """ global WORK_ORDER_DB_CONFIG WORK_ORDER_DB_CONFIG = load_db_config() logger.info("工单数据库配置已重新加载") def query_work_order(work_order_no: str) -> Optional[Dict[str, str]]: """ 查询工单信息 - SQL Server版本 Args: work_order_no: 工单号 Returns: 包含工单信息的字典，如果查询失败则返回None 字典包含： - work_order_no: 工单号 - process_no: 工序号 (CSYMBOL) - part_no: 零件号 (CMATERIALSYMBOL) - executor: 执行人 (CEXCUTOR) """ if not work_order_no: logger.warning("工单号为空") return None # Debug模式：返回假数据 if WORK_ORDER_DB_CONFIG.get('debug_mode', False): logger.info(f"[DEBUG模式] 返回工单号 {work_order_no} 的假数据") return { 'work_order_no': work_order_no, 'process_no': 'DEBUG-001', 'part_no': 'PART-DEBUG-12345', 'executor': '测试人员' } conn = None cursor = None try: import pyodbc # 尝试不同的ODBC驱动 drivers = [ "ODBC Driver 17 for SQL Server", "ODBC Driver 18 for SQL Server", "ODBC Driver 13 for SQL Server", "ODBC Driver 11 for SQL Server", "SQL Server Native Client 11.0", "SQL Server Native Client 10.0", "SQL Server" ] # 查找可用的驱动 available_drivers = pyodbc.drivers() driver_to_use = None for driver in drivers: if driver in available_drivers: driver_to_use = driver logger.info(f"使用ODBC驱动: {driver}") break if not driver_to_use: logger.warning(f"未找到合适的SQL Server ODBC驱动，可用驱动: {available_drivers}") # 尝试使用第一个包含SQL的驱动 for available in available_drivers: if 'SQL' in available.upper(): driver_to_use = available logger.info(f"尝试使用驱动: {driver_to_use}") break if not driver_to_use: raise Exception("未找到SQL Server ODBC驱动，将尝试pymssql") # 构建SQL Server连接字符串 conn_str = ( f"DRIVER={{{driver_to_use}}};" f"SERVER={WORK_ORDER_DB_CONFIG['host']},{WORK_ORDER_DB_CONFIG['port']};" f"DATABASE={WORK_ORDER_DB_CONFIG['database']};" f"UID={WORK_ORDER_DB_CONFIG['username']};" f"PWD={WORK_ORDER_DB_CONFIG['password']}" ) logger.info(f"连接到SQL Server: {WORK_ORDER_DB_CONFIG['host']}:{WORK_ORDER_DB_CONFIG['port']}/{WORK_ORDER_DB_CONFIG['database']}") # 连接数据库 conn = pyodbc.connect(conn_str) cursor = conn.cursor() # 构建SQL查询语句（SQL Server使用?作为参数占位符） # 使用LIKE模糊匹配，支持工单号前缀查询（如 W2001150.001 可以匹配 W2001150.001-01:10） sql_query = """ SELECT mte.CSYMBOL, mpe.CMATERIALSYMBOL, mte.CEXCUTOR FROM MES_TASK_EXTEND mte INNER JOIN MBP_PROJECT_EXTEND mpe ON mte.CPGDID = mpe.CID WHERE mte.CSYMBOL LIKE ? """ # 添加通配符进行模糊匹配 search_pattern = f"{work_order_no}%" logger.debug(f"执行查询: {sql_query} with parameter: {search_pattern}") cursor.execute(sql_query, search_pattern) row = cursor.fetchone() if not row: logger.warning(f"未找到工单号 '{work_order_no}' 的相关信息") return None # 提取查询结果 - pyodbc返回Row对象，可以通过索引或属性访问 work_order_info = { 'work_order_no': work_order_no, 'process_no': row.CSYMBOL if row.CSYMBOL else '', # 工序号 'part_no': row.CMATERIALSYMBOL if row.CMATERIALSYMBOL else '', # 零件号 'executor': row.CEXCUTOR if row.CEXCUTOR else '' # 执行人 } logger.info(f"工单查询成功: {work_order_info}") return work_order_info except ImportError: logger.error("pyodbc未安装，请运行: pip install pyodbc") # 尝试使用pymssql作为备选 try: import pymssql logger.info("尝试使用pymssql连接SQL Server") conn = pymssql.connect( server=WORK_ORDER_DB_CONFIG['host'], port=WORK_ORDER_DB_CONFIG['port'], user=WORK_ORDER_DB_CONFIG['username'], password=WORK_ORDER_DB_CONFIG['password'], database=WORK_ORDER_DB_CONFIG['database'] ) cursor = conn.cursor(as_dict=True) # 使用LIKE模糊匹配，支持工单号前缀查询 sql_query = """ SELECT mte.CSYMBOL, mpe.CMATERIALSYMBOL, mte.CEXCUTOR FROM MES_TASK_EXTEND mte INNER JOIN MBP_PROJECT_EXTEND mpe ON mte.CPGDID = mpe.CID WHERE mte.CSYMBOL LIKE %s """ search_pattern = f"{work_order_no}%" cursor.execute(sql_query, search_pattern) result = cursor.fetchone() if not result: logger.warning(f"未找到工单号 '{work_order_no}' 的相关信息") return None work_order_info = { 'work_order_no': work_order_no, 'process_no': result.get('CSYMBOL', ''), 'part_no': result.get('CMATERIALSYMBOL', ''), 'executor': result.get('CEXCUTOR', '') } logger.info(f"工单查询成功(pymssql): {work_order_info}") return work_order_info except ImportError: error_msg = "SQL Server驱动未安装，请安装以下任一库：\n1. pip install pyodbc\n2. pip install pymssql" logger.error(error_msg) raise Exception(error_msg) except Exception as e: logger.error(f"pymssql查询错误: {e}") raise Exception(f"数据库查询错误: {e}") except Exception as e: logger.error(f"SQL Server查询错误: {e}") raise Exception(f"数据库查询错误: {e}") finally: if cursor: cursor.close() if conn: conn.close() logger.debug("数据库连接已关闭") def update_work_order_db_config(host: str = None, port: int = None, database: str = None, username: str = None, password: str = None, save_to_file: bool = True): """ 更新工单数据库连接配置 Args: host: 数据库服务器地址 port: 数据库端口 database: 数据库名 username: 用户名 password: 密码 save_to_file: 是否保存到配置文件 """ global WORK_ORDER_DB_CONFIG if host is not None: WORK_ORDER_DB_CONFIG['host'] = host if port is not None: WORK_ORDER_DB_CONFIG['port'] = port if database is not None: WORK_ORDER_DB_CONFIG['database'] = database if username is not None: WORK_ORDER_DB_CONFIG['username'] = username if password is not None: WORK_ORDER_DB_CONFIG['password'] = password logger.info(f"工单数据库配置已更新: host={WORK_ORDER_DB_CONFIG['host']}, " f"port={WORK_ORDER_DB_CONFIG['port']}, database={WORK_ORDER_DB_CONFIG['database']}") # 保存到配置文件 if save_to_file: try: config_to_save = WORK_ORDER_DB_CONFIG.copy() config_to_save['description'] = '工单数据库连接配置' config_to_save['comment'] = '请根据实际环境修改以上配置' with open(CONFIG_FILE, 'w', encoding='utf-8') as f: json.dump(config_to_save, f, ensure_ascii=False, indent=2) logger.info(f"配置已保存到文件: {CONFIG_FILE}") except Exception as e: logger.error(f"保存配置文件失败: {e}")