53a1e33e45
Add optional database persistence feature that automatically saves extracted discrete material plan data to SQL Server. Users can enable this feature in the settings tab. Changes: - Add enable_db_persistence flag to ExtractionConfig (default: disabled) - Create DiscreteMaterialPlanDAO for database operations with REPLACE pattern - Update progress tracking to include database persistence stage (90-100%) - Add database persistence checkbox in settings UI - Remove verbose logging checkbox from data extraction UI (config-only now) - Update extraction workflow to save merged DataFrame to database Progress weights adjusted: - download: 65% -> 60% - database: 10% (new stage) - Other stages adjusted accordingly Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
102 lines
2.7 KiB
Python
102 lines
2.7 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
进度信息模块
|
||
|
||
定义用于进度回调的数据结构。
|
||
"""
|
||
|
||
from dataclasses import dataclass, field
|
||
from typing import Dict, Any, Optional
|
||
|
||
|
||
@dataclass
|
||
class ProgressInfo:
|
||
"""
|
||
进度信息
|
||
|
||
用于在后台任务和 GUI 之间传递进度信息。
|
||
"""
|
||
|
||
stage: (
|
||
str # 阶段标识: 'login', 'query', 'download', 'logout', 'convert', 'database', 'complete'
|
||
)
|
||
current: int # 当前进度值
|
||
total: int # 总量
|
||
message: str # 显示给用户的消息
|
||
detail: Dict[str, Any] = field(default_factory=dict) # 额外详细信息
|
||
|
||
@property
|
||
def percent(self) -> int:
|
||
"""计算进度百分比（0-100）"""
|
||
if self.total <= 0:
|
||
return 0
|
||
return int(self.current * 100 / self.total)
|
||
|
||
def __repr__(self) -> str:
|
||
return f"ProgressInfo(stage={self.stage}, {self.current}/{self.total}, {self.message})"
|
||
|
||
|
||
class ProgressCalculator:
|
||
"""
|
||
进度计算器
|
||
|
||
将各阶段的进度映射到总体进度百分比。
|
||
"""
|
||
|
||
# 各阶段在总进度中的占比
|
||
STAGE_WEIGHTS = {
|
||
"login": 5, # 登录: 0-5%
|
||
"query": 5, # 查询: 5-10%
|
||
"download": 60, # 下载: 10-70%
|
||
"logout": 5, # 注销: 70-75%
|
||
"convert": 15, # 转换: 75-90%
|
||
"database": 10, # 数据库持久化: 90-100%
|
||
"complete": 5, # 完成: 95-100%
|
||
}
|
||
|
||
def __init__(self):
|
||
"""初始化进度计算器"""
|
||
self._stage_offsets = self._calculate_stage_offsets()
|
||
|
||
def _calculate_stage_offsets(self) -> Dict[str, int]:
|
||
"""计算各阶段的起始偏移量（百分比）"""
|
||
offsets = {}
|
||
offset = 0
|
||
for stage, weight in self.STAGE_WEIGHTS.items():
|
||
offsets[stage] = offset
|
||
offset += weight
|
||
return offsets
|
||
|
||
def calculate_overall_percent(self, progress: ProgressInfo) -> int:
|
||
"""
|
||
计算总体进度百分比
|
||
|
||
Args:
|
||
progress: 进度信息
|
||
|
||
Returns:
|
||
总体进度百分比 (0-100)
|
||
"""
|
||
stage = progress.stage
|
||
|
||
if stage == "complete":
|
||
return 100
|
||
|
||
if stage not in self._stage_offsets:
|
||
return 0
|
||
|
||
# 计算阶段起始百分比
|
||
stage_offset = self._stage_offsets[stage]
|
||
|
||
# 计算阶段内的进度百分比
|
||
stage_percent = progress.percent
|
||
|
||
# 计算该阶段的权重
|
||
stage_weight = self.STAGE_WEIGHTS[stage]
|
||
|
||
# 总进度 = 阶段偏移 + (阶段内进度 * 阶段权重 / 100)
|
||
overall = stage_offset + int(stage_percent * stage_weight / 100)
|
||
|
||
return min(overall, 100)
|