750 lines
26 KiB
Python
750 lines
26 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
数据库操作模块
|
|||
|
|
使用SQLite存储工单数据
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import sqlite3
|
|||
|
|
import json
|
|||
|
|
from datetime import datetime
|
|||
|
|
from typing import Dict, List, Optional
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Database:
|
|||
|
|
"""数据库操作类"""
|
|||
|
|
|
|||
|
|
def __init__(self, db_path: str = "wrench.db"):
|
|||
|
|
"""
|
|||
|
|
初始化数据库
|
|||
|
|
:param db_path: 数据库文件路径
|
|||
|
|
"""
|
|||
|
|
self.db_path = db_path
|
|||
|
|
self.init_database()
|
|||
|
|
|
|||
|
|
def get_connection(self):
|
|||
|
|
"""获取数据库连接"""
|
|||
|
|
conn = sqlite3.connect(self.db_path)
|
|||
|
|
conn.row_factory = sqlite3.Row # 使查询结果可以按列名访问
|
|||
|
|
return conn
|
|||
|
|
|
|||
|
|
def init_database(self):
|
|||
|
|
"""初始化数据库表"""
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 创建工单表
|
|||
|
|
cursor.execute('''
|
|||
|
|
CREATE TABLE IF NOT EXISTS work_orders (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
trace_id TEXT NOT NULL,
|
|||
|
|
process_id TEXT NOT NULL,
|
|||
|
|
process_name TEXT,
|
|||
|
|
product_name TEXT,
|
|||
|
|
operator TEXT,
|
|||
|
|
status TEXT DEFAULT 'pending',
|
|||
|
|
bolts TEXT NOT NULL,
|
|||
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|||
|
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|||
|
|
UNIQUE(trace_id, process_id)
|
|||
|
|
)
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
# 创建扳手设备表
|
|||
|
|
cursor.execute('''
|
|||
|
|
CREATE TABLE IF NOT EXISTS wrench_devices (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
device_name TEXT NOT NULL,
|
|||
|
|
device_sn TEXT UNIQUE,
|
|||
|
|
ip_address TEXT NOT NULL,
|
|||
|
|
port INTEGER DEFAULT 7888,
|
|||
|
|
address_code INTEGER DEFAULT 1,
|
|||
|
|
status TEXT DEFAULT 'offline',
|
|||
|
|
last_heartbeat TIMESTAMP,
|
|||
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|||
|
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|||
|
|
)
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
# 创建认领表
|
|||
|
|
cursor.execute('''
|
|||
|
|
CREATE TABLE IF NOT EXISTS claimed_orders (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
trace_id TEXT NOT NULL,
|
|||
|
|
process_id TEXT NOT NULL,
|
|||
|
|
process_name TEXT,
|
|||
|
|
operator TEXT,
|
|||
|
|
status TEXT DEFAULT 'claimed',
|
|||
|
|
claimed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|||
|
|
completed_at TIMESTAMP,
|
|||
|
|
UNIQUE(trace_id, process_id)
|
|||
|
|
)
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
# 创建结果表
|
|||
|
|
cursor.execute('''
|
|||
|
|
CREATE TABLE IF NOT EXISTS work_results (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
trace_id TEXT NOT NULL,
|
|||
|
|
process_id TEXT NOT NULL,
|
|||
|
|
process_name TEXT,
|
|||
|
|
bolts TEXT NOT NULL,
|
|||
|
|
device_sn TEXT,
|
|||
|
|
device_name TEXT,
|
|||
|
|
submitted_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|||
|
|
)
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
def create_work_order(self, order_data: Dict) -> bool:
|
|||
|
|
"""
|
|||
|
|
创建工单
|
|||
|
|
:param order_data: 工单数据字典
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 将bolts列表转换为JSON字符串
|
|||
|
|
bolts_json = json.dumps(order_data.get('bolts', []), ensure_ascii=False)
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
INSERT OR REPLACE INTO work_orders
|
|||
|
|
(trace_id, process_id, process_name, product_name, operator, status, bolts, updated_at)
|
|||
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
|||
|
|
''', (
|
|||
|
|
order_data.get('trace_id'),
|
|||
|
|
order_data.get('process_id'),
|
|||
|
|
order_data.get('process_name', ''),
|
|||
|
|
order_data.get('product_name', ''),
|
|||
|
|
order_data.get('operator', ''),
|
|||
|
|
order_data.get('status', 'pending'),
|
|||
|
|
bolts_json,
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"创建工单失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def get_work_order(self, trace_id: str, process_id: str) -> Optional[Dict]:
|
|||
|
|
"""
|
|||
|
|
获取工单详情
|
|||
|
|
:param trace_id: 追溯号
|
|||
|
|
:param process_id: 工序号
|
|||
|
|
:return: 工单数据字典或None
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT * FROM work_orders
|
|||
|
|
WHERE trace_id = ? AND process_id = ?
|
|||
|
|
''', (trace_id, process_id))
|
|||
|
|
|
|||
|
|
row = cursor.fetchone()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
if row:
|
|||
|
|
result = {
|
|||
|
|
"trace_id": row['trace_id'],
|
|||
|
|
"process_id": row['process_id'],
|
|||
|
|
"process_name": row['process_name'],
|
|||
|
|
"product_name": row['product_name'],
|
|||
|
|
"operator": row['operator'],
|
|||
|
|
"status": row['status'],
|
|||
|
|
"bolts": json.loads(row['bolts'])
|
|||
|
|
}
|
|||
|
|
# 兼容旧数据（如果有station字段）
|
|||
|
|
if 'station' in row.keys():
|
|||
|
|
result['station'] = row['station']
|
|||
|
|
return result
|
|||
|
|
return None
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"查询工单失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def get_available_work_orders(self, trace_id: str, process_id: str) -> List[Dict]:
|
|||
|
|
"""
|
|||
|
|
获取可用工单列表（未认领的）
|
|||
|
|
:param trace_id: 追溯号
|
|||
|
|
:param process_id: 工序号
|
|||
|
|
:return: 工单列表
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 先检查工单是否存在
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT COUNT(*) FROM work_orders
|
|||
|
|
WHERE trace_id = ? AND process_id = ?
|
|||
|
|
''', (trace_id, process_id))
|
|||
|
|
total_count = cursor.fetchone()[0]
|
|||
|
|
print(f"[DEBUG] 工单总数: {total_count} (trace_id={trace_id}, process_id={process_id})")
|
|||
|
|
|
|||
|
|
# 查询匹配的工单，且未被认领
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT wo.* FROM work_orders wo
|
|||
|
|
LEFT JOIN claimed_orders co ON wo.trace_id = co.trace_id AND wo.process_id = co.process_id
|
|||
|
|
WHERE wo.trace_id = ? AND wo.process_id = ? AND co.id IS NULL
|
|||
|
|
''', (trace_id, process_id))
|
|||
|
|
|
|||
|
|
rows = cursor.fetchall()
|
|||
|
|
print(f"[DEBUG] 查询到 {len(rows)} 行数据")
|
|||
|
|
|
|||
|
|
# 检查认领状态
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT COUNT(*) FROM claimed_orders
|
|||
|
|
WHERE trace_id = ? AND process_id = ?
|
|||
|
|
''', (trace_id, process_id))
|
|||
|
|
claimed_count = cursor.fetchone()[0]
|
|||
|
|
print(f"[DEBUG] 已认领数量: {claimed_count}")
|
|||
|
|
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
orders = []
|
|||
|
|
for row in rows:
|
|||
|
|
try:
|
|||
|
|
bolts = json.loads(row['bolts'])
|
|||
|
|
order = {
|
|||
|
|
"trace_id": row['trace_id'],
|
|||
|
|
"process_id": row['process_id'],
|
|||
|
|
"process_name": row['process_name'],
|
|||
|
|
"product_name": row['product_name'],
|
|||
|
|
"operator": row['operator'],
|
|||
|
|
"bolt_count": len(bolts),
|
|||
|
|
"status": row['status']
|
|||
|
|
}
|
|||
|
|
# 兼容旧数据（如果有station字段）
|
|||
|
|
if 'station' in row.keys():
|
|||
|
|
order['station'] = row['station']
|
|||
|
|
orders.append(order)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 解析螺栓数据失败: {e}, row={row}")
|
|||
|
|
|
|||
|
|
print(f"[DEBUG] 返回 {len(orders)} 个可用工单")
|
|||
|
|
return orders
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 查询可用工单失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
def get_work_orders(self) -> List[Dict]:
|
|||
|
|
"""
|
|||
|
|
获取所有工单
|
|||
|
|
:return: 工单列表
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('SELECT * FROM work_orders')
|
|||
|
|
rows = cursor.fetchall()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
orders = []
|
|||
|
|
for row in rows:
|
|||
|
|
order = {
|
|||
|
|
"trace_id": row['trace_id'],
|
|||
|
|
"process_id": row['process_id'],
|
|||
|
|
"process_name": row['process_name'],
|
|||
|
|
"product_name": row['product_name'],
|
|||
|
|
"operator": row['operator'],
|
|||
|
|
"status": row['status'],
|
|||
|
|
"bolts": json.loads(row['bolts'])
|
|||
|
|
}
|
|||
|
|
# 兼容旧数据
|
|||
|
|
if 'station' in row.keys():
|
|||
|
|
order['station'] = row['station']
|
|||
|
|
orders.append(order)
|
|||
|
|
|
|||
|
|
return orders
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"查询所有工单失败: {e}")
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
def get_all_available_work_orders(self) -> List[Dict]:
|
|||
|
|
"""
|
|||
|
|
获取所有可用工单列表（未认领的）
|
|||
|
|
:return: 工单列表
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 查询所有未认领的工单
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT wo.* FROM work_orders wo
|
|||
|
|
LEFT JOIN claimed_orders co ON wo.trace_id = co.trace_id AND wo.process_id = co.process_id
|
|||
|
|
WHERE co.id IS NULL
|
|||
|
|
ORDER BY wo.created_at DESC
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
rows = cursor.fetchall()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
orders = []
|
|||
|
|
for row in rows:
|
|||
|
|
try:
|
|||
|
|
bolts = json.loads(row['bolts'])
|
|||
|
|
order = {
|
|||
|
|
"trace_id": row['trace_id'],
|
|||
|
|
"process_id": row['process_id'],
|
|||
|
|
"process_name": row['process_name'],
|
|||
|
|
"product_name": row['product_name'],
|
|||
|
|
"operator": row['operator'],
|
|||
|
|
"bolt_count": len(bolts),
|
|||
|
|
"status": row['status']
|
|||
|
|
}
|
|||
|
|
# 兼容旧数据（如果有station字段）
|
|||
|
|
if 'station' in row.keys():
|
|||
|
|
order['station'] = row['station']
|
|||
|
|
orders.append(order)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 解析螺栓数据失败: {e}")
|
|||
|
|
|
|||
|
|
return orders
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 查询所有可用工单失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
def get_claimed_order(self, trace_id: str, process_id: str) -> Optional[Dict]:
|
|||
|
|
"""
|
|||
|
|
获取已认领的工单
|
|||
|
|
:param trace_id: 追溯号
|
|||
|
|
:param process_id: 工序号
|
|||
|
|
:return: 认领信息或None
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT * FROM claimed_orders
|
|||
|
|
WHERE trace_id = ? AND process_id = ?
|
|||
|
|
''', (trace_id, process_id))
|
|||
|
|
|
|||
|
|
row = cursor.fetchone()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
if row:
|
|||
|
|
return {
|
|||
|
|
"trace_id": row['trace_id'],
|
|||
|
|
"process_id": row['process_id'],
|
|||
|
|
"process_name": row['process_name'],
|
|||
|
|
"operator": row['operator'],
|
|||
|
|
"status": row['status'],
|
|||
|
|
"claimed_at": row['claimed_at']
|
|||
|
|
}
|
|||
|
|
return None
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"查询认领工单失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def claim_work_order(self, trace_id: str, process_id: str, operator: str) -> bool:
|
|||
|
|
"""
|
|||
|
|
认领工单
|
|||
|
|
:param trace_id: 追溯号
|
|||
|
|
:param process_id: 工序号
|
|||
|
|
:param operator: 操作员
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 获取工单信息以获取process_name
|
|||
|
|
order = self.get_work_order(trace_id, process_id)
|
|||
|
|
process_name = order.get('process_name', '') if order else ''
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
INSERT OR REPLACE INTO claimed_orders
|
|||
|
|
(trace_id, process_id, process_name, operator, status, claimed_at)
|
|||
|
|
VALUES (?, ?, ?, ?, 'claimed', ?)
|
|||
|
|
''', (
|
|||
|
|
trace_id,
|
|||
|
|
process_id,
|
|||
|
|
process_name,
|
|||
|
|
operator,
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"认领工单失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def release_work_order(self, trace_id: str, process_id: str) -> bool:
|
|||
|
|
"""
|
|||
|
|
释放工单（取消认领）
|
|||
|
|
:param trace_id: 追溯号
|
|||
|
|
:param process_id: 工序号
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
DELETE FROM claimed_orders
|
|||
|
|
WHERE trace_id = ? AND process_id = ?
|
|||
|
|
''', (trace_id, process_id))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
deleted = cursor.rowcount > 0
|
|||
|
|
conn.close()
|
|||
|
|
return deleted
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"释放工单失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def submit_work_order(self, trace_id: str, process_id: str, bolts: List[Dict],
|
|||
|
|
device_sn: str = None, device_name: str = None) -> bool:
|
|||
|
|
"""
|
|||
|
|
提交工单结果
|
|||
|
|
:param trace_id: 追溯号
|
|||
|
|
:param process_id: 工序号
|
|||
|
|
:param bolts: 螺栓数据列表
|
|||
|
|
:param device_sn: 设备SN号
|
|||
|
|
:param device_name: 设备名称
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 获取工单信息
|
|||
|
|
order = self.get_work_order(trace_id, process_id)
|
|||
|
|
process_name = order.get('process_name', '') if order else ''
|
|||
|
|
|
|||
|
|
# 更新认领状态为已完成
|
|||
|
|
cursor.execute('''
|
|||
|
|
UPDATE claimed_orders
|
|||
|
|
SET status = 'completed', completed_at = ?
|
|||
|
|
WHERE trace_id = ? AND process_id = ?
|
|||
|
|
''', (
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|||
|
|
trace_id,
|
|||
|
|
process_id
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
# 保存结果
|
|||
|
|
bolts_json = json.dumps(bolts, ensure_ascii=False)
|
|||
|
|
cursor.execute('''
|
|||
|
|
INSERT INTO work_results
|
|||
|
|
(trace_id, process_id, process_name, bolts, device_sn, device_name, submitted_at)
|
|||
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|||
|
|
''', (
|
|||
|
|
trace_id,
|
|||
|
|
process_id,
|
|||
|
|
process_name,
|
|||
|
|
bolts_json,
|
|||
|
|
device_sn,
|
|||
|
|
device_name,
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"提交工单失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def get_work_result(self, trace_id: str, process_id: str) -> Optional[Dict]:
|
|||
|
|
"""
|
|||
|
|
查询工单执行结果
|
|||
|
|
:param trace_id: 追溯号
|
|||
|
|
:param process_id: 工序号
|
|||
|
|
:return: 执行结果字典，如果不存在则返回None
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
conn.row_factory = sqlite3.Row
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT * FROM work_results
|
|||
|
|
WHERE trace_id = ? AND process_id = ?
|
|||
|
|
ORDER BY submitted_at DESC
|
|||
|
|
LIMIT 1
|
|||
|
|
''', (trace_id, process_id))
|
|||
|
|
|
|||
|
|
row = cursor.fetchone()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
if row:
|
|||
|
|
# 解析螺栓数据
|
|||
|
|
try:
|
|||
|
|
bolts = json.loads(row['bolts'])
|
|||
|
|
except:
|
|||
|
|
bolts = []
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"id": row['id'],
|
|||
|
|
"trace_id": row['trace_id'],
|
|||
|
|
"process_id": row['process_id'],
|
|||
|
|
"process_name": row['process_name'],
|
|||
|
|
"bolts": bolts,
|
|||
|
|
"device_sn": row['device_sn'],
|
|||
|
|
"device_name": row['device_name'],
|
|||
|
|
"submitted_at": row['submitted_at']
|
|||
|
|
}
|
|||
|
|
return None
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"查询工单结果失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def get_work_results(self, trace_id: str = None, process_id: str = None) -> List[Dict]:
|
|||
|
|
"""
|
|||
|
|
查询工单执行结果列表
|
|||
|
|
:param trace_id: 追溯号（可选）
|
|||
|
|
:param process_id: 工序号（可选）
|
|||
|
|
:return: 执行结果列表
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
conn.row_factory = sqlite3.Row
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
if trace_id and process_id:
|
|||
|
|
# 查询指定追溯号和工序号的结果
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT * FROM work_results
|
|||
|
|
WHERE trace_id = ? AND process_id = ?
|
|||
|
|
ORDER BY submitted_at DESC
|
|||
|
|
''', (trace_id, process_id))
|
|||
|
|
elif trace_id:
|
|||
|
|
# 只按追溯号查询
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT * FROM work_results
|
|||
|
|
WHERE trace_id = ?
|
|||
|
|
ORDER BY submitted_at DESC
|
|||
|
|
''', (trace_id,))
|
|||
|
|
elif process_id:
|
|||
|
|
# 只按工序号查询
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT * FROM work_results
|
|||
|
|
WHERE process_id = ?
|
|||
|
|
ORDER BY submitted_at DESC
|
|||
|
|
''', (process_id,))
|
|||
|
|
else:
|
|||
|
|
# 查询所有结果
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT * FROM work_results
|
|||
|
|
ORDER BY submitted_at DESC
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
rows = cursor.fetchall()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
results = []
|
|||
|
|
for row in rows:
|
|||
|
|
try:
|
|||
|
|
bolts = json.loads(row['bolts'])
|
|||
|
|
except:
|
|||
|
|
bolts = []
|
|||
|
|
|
|||
|
|
results.append({
|
|||
|
|
"id": row['id'],
|
|||
|
|
"trace_id": row['trace_id'],
|
|||
|
|
"process_id": row['process_id'],
|
|||
|
|
"process_name": row['process_name'],
|
|||
|
|
"bolts": bolts,
|
|||
|
|
"device_sn": row['device_sn'],
|
|||
|
|
"device_name": row['device_name'],
|
|||
|
|
"submitted_at": row['submitted_at']
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
return results
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"查询工单结果列表失败: {e}")
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
# ========== 扳手设备管理方法 ==========
|
|||
|
|
|
|||
|
|
def create_wrench_device(self, device_data: Dict) -> bool:
|
|||
|
|
"""创建扳手设备"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
INSERT OR REPLACE INTO wrench_devices
|
|||
|
|
(device_name, device_sn, ip_address, port, address_code, status, updated_at)
|
|||
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|||
|
|
''', (
|
|||
|
|
device_data.get('device_name'),
|
|||
|
|
device_data.get('device_sn'),
|
|||
|
|
device_data.get('ip_address'),
|
|||
|
|
device_data.get('port', 7888),
|
|||
|
|
device_data.get('address_code', 1),
|
|||
|
|
device_data.get('status', 'offline'),
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"创建扳手设备失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def get_wrench_devices(self) -> List[Dict]:
|
|||
|
|
"""获取所有扳手设备"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('SELECT * FROM wrench_devices ORDER BY device_name')
|
|||
|
|
rows = cursor.fetchall()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
devices = []
|
|||
|
|
for row in rows:
|
|||
|
|
devices.append({
|
|||
|
|
"id": row['id'],
|
|||
|
|
"device_name": row['device_name'],
|
|||
|
|
"device_sn": row['device_sn'],
|
|||
|
|
"ip_address": row['ip_address'],
|
|||
|
|
"port": row['port'],
|
|||
|
|
"address_code": row['address_code'],
|
|||
|
|
"status": row['status'],
|
|||
|
|
"last_heartbeat": row['last_heartbeat']
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
return devices
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"查询扳手设备失败: {e}")
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
def get_wrench_device(self, device_id: int) -> Optional[Dict]:
|
|||
|
|
"""获取扳手设备详情"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('SELECT * FROM wrench_devices WHERE id = ?', (device_id,))
|
|||
|
|
row = cursor.fetchone()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
if row:
|
|||
|
|
return {
|
|||
|
|
"id": row['id'],
|
|||
|
|
"device_name": row['device_name'],
|
|||
|
|
"device_sn": row['device_sn'],
|
|||
|
|
"ip_address": row['ip_address'],
|
|||
|
|
"port": row['port'],
|
|||
|
|
"address_code": row['address_code'],
|
|||
|
|
"status": row['status'],
|
|||
|
|
"last_heartbeat": row['last_heartbeat']
|
|||
|
|
}
|
|||
|
|
return None
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"查询扳手设备失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def update_wrench_device_status(self, device_id: int, status: str) -> bool:
|
|||
|
|
"""更新扳手设备状态"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
UPDATE wrench_devices
|
|||
|
|
SET status = ?, last_heartbeat = ?, updated_at = ?
|
|||
|
|
WHERE id = ?
|
|||
|
|
''', (
|
|||
|
|
status,
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|||
|
|
device_id
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"更新扳手设备状态失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def update_wrench_device_sn(self, device_id: int, device_sn: str) -> bool:
|
|||
|
|
"""更新扳手设备SN码"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
UPDATE wrench_devices
|
|||
|
|
SET device_sn = ?, updated_at = ?
|
|||
|
|
WHERE id = ?
|
|||
|
|
''', (
|
|||
|
|
device_sn,
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|||
|
|
device_id
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"更新扳手设备SN码失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def update_wrench_device(self, device_id: int, device_data: Dict) -> bool:
|
|||
|
|
"""更新扳手设备信息"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('''
|
|||
|
|
UPDATE wrench_devices
|
|||
|
|
SET device_name = ?, device_sn = ?, ip_address = ?, port = ?,
|
|||
|
|
address_code = ?, updated_at = ?
|
|||
|
|
WHERE id = ?
|
|||
|
|
''', (
|
|||
|
|
device_data.get('device_name'),
|
|||
|
|
device_data.get('device_sn'),
|
|||
|
|
device_data.get('ip_address'),
|
|||
|
|
device_data.get('port', 7888),
|
|||
|
|
device_data.get('address_code', 1),
|
|||
|
|
datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|||
|
|
device_id
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"更新扳手设备失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def delete_wrench_device(self, device_id: int) -> bool:
|
|||
|
|
"""删除扳手设备"""
|
|||
|
|
try:
|
|||
|
|
conn = self.get_connection()
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('DELETE FROM wrench_devices WHERE id = ?', (device_id,))
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
deleted = cursor.rowcount > 0
|
|||
|
|
conn.close()
|
|||
|
|
return deleted
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"删除扳手设备失败: {e}")
|
|||
|
|
return False
|