336 lines
11 KiB
Python
336 lines
11 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
每日交接班日志数据库模块
|
|||
|
|
基于新的数据库基类重构
|
|||
|
|
"""
|
|||
|
|
from typing import List, Dict, Optional, Any
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
from src.database.base import DatabaseBase
|
|||
|
|
from src.logging_config import get_logger
|
|||
|
|
|
|||
|
|
logger = get_logger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DailyLogsDatabase(DatabaseBase):
|
|||
|
|
"""每日交接班日志数据库"""
|
|||
|
|
|
|||
|
|
def __init__(self, db_path: Optional[str] = None):
|
|||
|
|
"""
|
|||
|
|
初始化数据库
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
db_path: 数据库文件路径，如果为None则使用默认配置
|
|||
|
|
"""
|
|||
|
|
super().__init__(db_path)
|
|||
|
|
self._init_schema()
|
|||
|
|
|
|||
|
|
def _init_schema(self):
|
|||
|
|
"""初始化表结构"""
|
|||
|
|
with self.get_connection() as conn:
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 创建每日交接班日志表
|
|||
|
|
cursor.execute('''
|
|||
|
|
CREATE TABLE IF NOT EXISTS daily_handover_logs (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
date TEXT NOT NULL,
|
|||
|
|
shift TEXT NOT NULL,
|
|||
|
|
ship_name TEXT NOT NULL,
|
|||
|
|
teu INTEGER,
|
|||
|
|
efficiency REAL,
|
|||
|
|
vehicles INTEGER,
|
|||
|
|
created_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|||
|
|
UNIQUE(date, shift, ship_name) ON CONFLICT REPLACE
|
|||
|
|
)
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
# 检查是否需要迁移旧表结构
|
|||
|
|
cursor.execute("SELECT sql FROM sqlite_master WHERE type='table' AND name='daily_handover_logs'")
|
|||
|
|
table_sql = cursor.fetchone()[0]
|
|||
|
|
if 'UNIQUE' not in table_sql:
|
|||
|
|
logger.warning("检测到旧表结构，正在迁移...")
|
|||
|
|
|
|||
|
|
# 重命名旧表
|
|||
|
|
cursor.execute('ALTER TABLE daily_handover_logs RENAME TO daily_handover_logs_old')
|
|||
|
|
|
|||
|
|
# 创建新表
|
|||
|
|
cursor.execute('''
|
|||
|
|
CREATE TABLE daily_handover_logs (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
date TEXT NOT NULL,
|
|||
|
|
shift TEXT NOT NULL,
|
|||
|
|
ship_name TEXT NOT NULL,
|
|||
|
|
teu INTEGER,
|
|||
|
|
efficiency REAL,
|
|||
|
|
vehicles INTEGER,
|
|||
|
|
created_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|||
|
|
UNIQUE(date, shift, ship_name) ON CONFLICT REPLACE
|
|||
|
|
)
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
# 复制数据（忽略重复）
|
|||
|
|
cursor.execute('''
|
|||
|
|
INSERT OR IGNORE INTO daily_handover_logs
|
|||
|
|
(date, shift, ship_name, teu, efficiency, vehicles, created_at)
|
|||
|
|
SELECT date, shift, ship_name, teu, efficiency, vehicles, created_at
|
|||
|
|
FROM daily_handover_logs_old
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
# 删除旧表
|
|||
|
|
cursor.execute('DROP TABLE daily_handover_logs_old')
|
|||
|
|
logger.info("迁移完成！")
|
|||
|
|
|
|||
|
|
# 创建索引
|
|||
|
|
cursor.execute('CREATE INDEX IF NOT EXISTS idx_date ON daily_handover_logs(date)')
|
|||
|
|
cursor.execute('CREATE INDEX IF NOT EXISTS idx_ship ON daily_handover_logs(ship_name)')
|
|||
|
|
|
|||
|
|
# 创建未统计月报数据表
|
|||
|
|
cursor.execute('''
|
|||
|
|
CREATE TABLE IF NOT EXISTS monthly_unaccounted (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
year_month TEXT NOT NULL UNIQUE,
|
|||
|
|
teu INTEGER NOT NULL,
|
|||
|
|
note TEXT,
|
|||
|
|
created_at TEXT DEFAULT CURRENT_TIMESTAMP
|
|||
|
|
)
|
|||
|
|
''')
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
logger.debug("数据库表结构初始化完成")
|
|||
|
|
|
|||
|
|
def insert(self, log: Dict[str, Any]) -> bool:
|
|||
|
|
"""
|
|||
|
|
插入记录（存在则替换，不存在则插入）
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
log: 日志记录字典
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
是否成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
query = '''
|
|||
|
|
INSERT OR REPLACE INTO daily_handover_logs
|
|||
|
|
(date, shift, ship_name, teu, efficiency, vehicles, created_at)
|
|||
|
|
VALUES (?, ?, ?, ?, ?, ?, CURRENT_TIMESTAMP)
|
|||
|
|
'''
|
|||
|
|
params = (
|
|||
|
|
log['date'], log['shift'], log['ship_name'],
|
|||
|
|
log.get('teu'), log.get('efficiency'), log.get('vehicles')
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
self.execute_update(query, params)
|
|||
|
|
logger.debug(f"插入记录: {log['date']} {log['shift']} {log['ship_name']}")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"插入记录失败: {e}, 记录: {log}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def insert_many(self, logs: List[Dict[str, Any]]) -> int:
|
|||
|
|
"""
|
|||
|
|
批量插入
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
logs: 日志记录列表
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
成功插入的数量
|
|||
|
|
"""
|
|||
|
|
count = 0
|
|||
|
|
for log in logs:
|
|||
|
|
if self.insert(log):
|
|||
|
|
count += 1
|
|||
|
|
|
|||
|
|
logger.info(f"批量插入完成，成功 {count}/{len(logs)} 条记录")
|
|||
|
|
return count
|
|||
|
|
|
|||
|
|
def query_by_date(self, date: str) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
按日期查询
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
date: 日期字符串
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
日志记录列表
|
|||
|
|
"""
|
|||
|
|
query = '''
|
|||
|
|
SELECT * FROM daily_handover_logs
|
|||
|
|
WHERE date = ? ORDER BY shift, ship_name
|
|||
|
|
'''
|
|||
|
|
return self.execute_query(query, (date,))
|
|||
|
|
|
|||
|
|
def query_by_ship(self, ship_name: str) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
按船名查询
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
ship_name: 船名
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
日志记录列表
|
|||
|
|
"""
|
|||
|
|
query = '''
|
|||
|
|
SELECT * FROM daily_handover_logs
|
|||
|
|
WHERE ship_name LIKE ? ORDER BY date DESC
|
|||
|
|
'''
|
|||
|
|
return self.execute_query(query, (f'%{ship_name}%',))
|
|||
|
|
|
|||
|
|
def query_all(self, limit: int = 1000) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
查询所有记录
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
limit: 限制返回数量
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
日志记录列表
|
|||
|
|
"""
|
|||
|
|
query = '''
|
|||
|
|
SELECT * FROM daily_handover_logs
|
|||
|
|
ORDER BY date DESC, shift LIMIT ?
|
|||
|
|
'''
|
|||
|
|
return self.execute_query(query, (limit,))
|
|||
|
|
|
|||
|
|
def get_stats(self) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
获取统计信息
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
统计信息字典
|
|||
|
|
"""
|
|||
|
|
with self.get_connection() as conn:
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
cursor.execute('SELECT COUNT(*) FROM daily_handover_logs')
|
|||
|
|
total = cursor.fetchone()[0]
|
|||
|
|
|
|||
|
|
cursor.execute('SELECT DISTINCT ship_name FROM daily_handover_logs')
|
|||
|
|
ships = [row[0] for row in cursor.fetchall()]
|
|||
|
|
|
|||
|
|
cursor.execute('SELECT MIN(date), MAX(date) FROM daily_handover_logs')
|
|||
|
|
date_range = cursor.fetchone()
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
'total': total,
|
|||
|
|
'ships': ships,
|
|||
|
|
'date_range': {'start': date_range[0], 'end': date_range[1]}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def get_ships_with_monthly_teu(self, year_month: Optional[str] = None) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
获取所有船只及其当月TEU总量
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
year_month: 年月字符串，格式 "2025-12"，如果为None则统计所有
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
船只统计列表
|
|||
|
|
"""
|
|||
|
|
if year_month:
|
|||
|
|
query = '''
|
|||
|
|
SELECT ship_name, SUM(teu) as monthly_teu
|
|||
|
|
FROM daily_handover_logs
|
|||
|
|
WHERE date LIKE ?
|
|||
|
|
GROUP BY ship_name
|
|||
|
|
ORDER BY monthly_teu DESC
|
|||
|
|
'''
|
|||
|
|
return self.execute_query(query, (f'{year_month}%',))
|
|||
|
|
else:
|
|||
|
|
query = '''
|
|||
|
|
SELECT ship_name, SUM(teu) as monthly_teu
|
|||
|
|
FROM daily_handover_logs
|
|||
|
|
GROUP BY ship_name
|
|||
|
|
ORDER BY monthly_teu DESC
|
|||
|
|
'''
|
|||
|
|
return self.execute_query(query)
|
|||
|
|
|
|||
|
|
def insert_unaccounted(self, year_month: str, teu: int, note: str = '') -> bool:
|
|||
|
|
"""
|
|||
|
|
插入未统计数据
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
year_month: 年月字符串，格式 "2025-12"
|
|||
|
|
teu: 未统计TEU数量
|
|||
|
|
note: 备注
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
是否成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
query = '''
|
|||
|
|
INSERT OR REPLACE INTO monthly_unaccounted
|
|||
|
|
(year_month, teu, note, created_at)
|
|||
|
|
VALUES (?, ?, ?, CURRENT_TIMESTAMP)
|
|||
|
|
'''
|
|||
|
|
self.execute_update(query, (year_month, teu, note))
|
|||
|
|
logger.info(f"插入未统计数据: {year_month} {teu}TEU")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"插入未统计数据失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def get_unaccounted(self, year_month: str) -> int:
|
|||
|
|
"""
|
|||
|
|
获取指定月份的未统计数据
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
year_month: 年月字符串，格式 "2025-12"
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
未统计TEU数量
|
|||
|
|
"""
|
|||
|
|
query = 'SELECT teu FROM monthly_unaccounted WHERE year_month = ?'
|
|||
|
|
result = self.execute_query(query, (year_month,))
|
|||
|
|
return result[0]['teu'] if result else 0
|
|||
|
|
|
|||
|
|
def delete_by_date(self, date: str) -> int:
|
|||
|
|
"""
|
|||
|
|
删除指定日期的记录
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
date: 日期字符串
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
删除的记录数
|
|||
|
|
"""
|
|||
|
|
query = 'DELETE FROM daily_handover_logs WHERE date = ?'
|
|||
|
|
return self.execute_update(query, (date,))
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
# 测试代码
|
|||
|
|
db = DailyLogsDatabase()
|
|||
|
|
|
|||
|
|
# 测试插入
|
|||
|
|
test_log = {
|
|||
|
|
'date': '2025-12-30',
|
|||
|
|
'shift': '白班',
|
|||
|
|
'ship_name': '测试船',
|
|||
|
|
'teu': 100,
|
|||
|
|
'efficiency': 3.5,
|
|||
|
|
'vehicles': 5
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
success = db.insert(test_log)
|
|||
|
|
print(f"插入测试: {'成功' if success else '失败'}")
|
|||
|
|
|
|||
|
|
# 测试查询
|
|||
|
|
logs = db.query_by_date('2025-12-30')
|
|||
|
|
print(f"查询结果: {len(logs)} 条记录")
|
|||
|
|
|
|||
|
|
# 测试统计
|
|||
|
|
stats = db.get_stats()
|
|||
|
|
print(f"统计信息: {stats}")
|
|||
|
|
|
|||
|
|
# 测试未统计数据
|
|||
|
|
db.insert_unaccounted('2025-12', 118, '测试备注')
|
|||
|
|
unaccounted = db.get_unaccounted('2025-12')
|
|||
|
|
print(f"未统计数据: {unaccounted}TEU")
|
|||
|
|
|
|||
|
|
# 清理测试数据
|
|||
|
|
db.delete_by_date('2025-12-30')
|
|||
|
|
print("测试数据已清理")
|