709 lines
29 KiB
Python
709 lines
29 KiB
Python
import sys
|
||
import os
|
||
import time
|
||
import threading
|
||
import re
|
||
import urllib.parse
|
||
import webbrowser
|
||
from datetime import datetime
|
||
import tkinter as tk
|
||
from tkinter import filedialog, messagebox
|
||
|
||
import requests
|
||
import pandas as pd
|
||
from lxml import html
|
||
|
||
# ================= 1. 导入 UI 库 =================
|
||
import ttkbootstrap as ttk
|
||
from ttkbootstrap.constants import *
|
||
from ttkbootstrap.dialogs import Messagebox
|
||
|
||
# 兼容导入
|
||
try:
|
||
from ttkbootstrap.widgets import ScrolledText, Tableview, ToastNotification
|
||
except ImportError:
|
||
from ttkbootstrap.scrolled import ScrolledText
|
||
from ttkbootstrap.tableview import Tableview
|
||
from ttkbootstrap.toast import ToastNotification
|
||
|
||
|
||
# ================= 2. 后端核心逻辑 (保持功能完整) =================
|
||
class CRMCrawler:
|
||
def __init__(self, log_callback, data_callback):
|
||
self.log = log_callback
|
||
self.on_data = data_callback
|
||
self.stop_flag = False
|
||
self.session = requests.Session()
|
||
self.base_url = "http://111.198.24.44:88/index.php"
|
||
self.http_headers = {
|
||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
|
||
"X-Requested-With": "XMLHttpRequest",
|
||
"Accept": "application/json, text/javascript, */*; q=0.01"
|
||
}
|
||
|
||
def login(self, username, password):
|
||
self.log(f"🔑 正在登录... 用户: {username}")
|
||
login_payload = {
|
||
"module": "Users", "action": "Authenticate", "return_module": "Users",
|
||
"return_action": "Login", "user_name": username, "user_password": password, "login_theme": "newskin"
|
||
}
|
||
try:
|
||
self.session.get(self.base_url, headers=self.http_headers)
|
||
self.session.post(self.base_url, data=login_payload, headers=self.http_headers)
|
||
if 'PHPSESSID' in self.session.cookies:
|
||
self.log("✅ 登录成功！")
|
||
return True
|
||
else:
|
||
self.log("❌ 登录失败：请检查账号密码")
|
||
return False
|
||
except Exception as e:
|
||
self.log(f"❌ 网络错误: {str(e)}")
|
||
return False
|
||
|
||
def fetch_detail(self, record_id):
|
||
try:
|
||
url = f"{self.base_url}?module=SalesOrder&action=DetailView&record={record_id}"
|
||
resp = self.session.get(url, headers=self.http_headers, timeout=10)
|
||
if resp.status_code == 200:
|
||
tree = html.fromstring(resp.content)
|
||
target = tree.xpath("/html/body/div[1]/div/div[2]/div[2]/form/div[1]/div[1]/div[2]")
|
||
if target:
|
||
import copy
|
||
el = copy.deepcopy(target[0])
|
||
for bad in el.xpath('.//script | .//style'): bad.drop_tree()
|
||
for br in el.xpath('.//br'): br.tail = "\n" + (br.tail if br.tail else "")
|
||
return "\n".join([line.strip() for line in el.text_content().splitlines() if line.strip()])
|
||
except Exception:
|
||
pass
|
||
return ""
|
||
|
||
def parse_data(self, text, cid):
|
||
if not text: return None
|
||
data = {
|
||
"系统ID": cid,
|
||
"合同编号": "", "内贸合同号": "", "外贸合同号": "",
|
||
"签署公司": "", "收款情况": "", "签订日期": "", "销售员": "",
|
||
"最终用户单位": "", "最终用户信息联系人": "", "最终用户信息电话": "", "最终用户信息邮箱": "",
|
||
"最终用户所在地": "",
|
||
"买方单位": "", "买方信息联系人": "", "买方信息电话": "", "买方信息邮箱": "",
|
||
"厂家": "", "厂家型号": "", "合同标的": "", "数量": "", "单位": "台/套",
|
||
"折扣率（％）": "", "合同额": "", "合同总额": "",
|
||
"外购付款方式": "", "最晚发货期": "", "已收款": "", "未收款": "", "收款日期": "",
|
||
"IS_ASD": False, "_temp_second_code": ""
|
||
}
|
||
|
||
lines = [line.strip() for line in text.split('\n') if line.strip()]
|
||
key_map = {
|
||
"收款账户": "签署公司", "收款状态": "收款情况", "签约日期": "签订日期",
|
||
"负责人": "销售员", "客户名称": "最终用户单位", "联系人姓名": "最终用户信息联系人",
|
||
"合同总额": "合同总额", "最新收款日期": "收款日期", "最晚发货期": "最晚发货期",
|
||
"付款比例及期限": "外购付款方式", "地址": "最终用户所在地", "厂家": "厂家"
|
||
}
|
||
|
||
for i, line in enumerate(lines):
|
||
if line == "合同订单编号" and i + 1 < len(lines):
|
||
parts = lines[i + 1].strip().split()
|
||
if len(parts) >= 1: data["合同编号"] = parts[0]
|
||
if len(parts) >= 2: data["_temp_second_code"] = parts[1]
|
||
elif line in key_map and i + 1 < len(lines):
|
||
target = key_map[line]
|
||
if not data[target]: data[target] = lines[i + 1]
|
||
elif "合同标的" in line and "品名/型号" in line and i + 1 < len(lines):
|
||
parts = lines[i + 1].split('/')
|
||
if len(parts) >= 1: data["合同标的"] = parts[0]
|
||
if len(parts) >= 2: data["厂家型号"] = parts[1]
|
||
if len(parts) >= 3: data["数量"] = parts[2]
|
||
if len(parts) >= 5: data["合同额"] = parts[4]
|
||
|
||
buyer_match = re.search(r"(?:买方|The Buyer)[：:]\s*(.*?)(?:\n|$)", text)
|
||
if buyer_match and len(buyer_match.group(1)) > 1: data["买方单位"] = buyer_match.group(1).strip()
|
||
|
||
buyer_ct = re.search(r"联系人（Contact person）[：:]\s*(.*?)(?:\n|$)", text)
|
||
if buyer_ct: data["买方信息联系人"] = buyer_ct.group(1).strip()
|
||
buyer_tel = re.search(r"电话\(Tel\)[：:]\s*(.*?)(?:\s+|$|传真)", text)
|
||
if buyer_tel: data["买方信息电话"] = buyer_tel.group(1).strip()
|
||
|
||
try:
|
||
total = float(data["合同总额"]) if data["合同总额"] else 0
|
||
if "已收" in data["收款情况"]:
|
||
data["已收款"] = str(total);
|
||
data["未收款"] = "0"
|
||
elif "未" in data["收款情况"]:
|
||
data["已收款"] = "0";
|
||
data["未收款"] = str(total)
|
||
except:
|
||
pass
|
||
|
||
factory_val = data.get("厂家", "")
|
||
if factory_val and "ASD" in factory_val.upper():
|
||
data["IS_ASD"] = True
|
||
else:
|
||
data["IS_ASD"] = False
|
||
|
||
c_no = data.get("合同编号", "").strip().upper()
|
||
sec_code = data.pop("_temp_second_code", "")
|
||
if c_no.startswith('W'):
|
||
data["外贸合同号"] = sec_code
|
||
elif c_no.startswith('N'):
|
||
data["内贸合同号"] = sec_code
|
||
else:
|
||
data["内贸合同号"] = sec_code
|
||
|
||
if not c_no: return None
|
||
return data
|
||
|
||
def extract_time(self, text):
|
||
matches = re.findall(r"(\d{4}-\d{2}-\d{2}\s+\d{2}:\d{2}:\d{2})", text)
|
||
if matches:
|
||
dt_objects = [datetime.strptime(m, "%Y-%m-%d %H:%M:%S") for m in matches]
|
||
return max(dt_objects)
|
||
return None
|
||
|
||
def run_task(self, mode, **kwargs):
|
||
crmids = []
|
||
if mode == 'search':
|
||
query = kwargs.get('query')
|
||
self.log(f"🔍 正在搜索: {query}")
|
||
url = f"{self.base_url}?module=Home&action=UnifiedSearch&selectedmodule=undefined&query_string={urllib.parse.quote(query)}"
|
||
resp = self.session.get(url, headers=self.http_headers)
|
||
tree = html.fromstring(resp.content)
|
||
links = tree.xpath('//a[contains(@onclick, "record=")]')
|
||
for link in links:
|
||
onclick = link.get('onclick', '')
|
||
match = re.search(r"record=(\d+)", onclick)
|
||
if match: crmids.append(match.group(1))
|
||
crmids = list(set(crmids))
|
||
|
||
elif mode == 'full':
|
||
self.log("🚀 开始全量爬取 (演示限制前5页)")
|
||
crmids = self._get_list_ids(limit_pages=5)
|
||
|
||
elif mode == 'date':
|
||
s_date = kwargs.get('start')
|
||
e_date = kwargs.get('end')
|
||
self.log(f"📅 时间筛选: {s_date} ~ {e_date}")
|
||
self._process_date_range(s_date, e_date)
|
||
return
|
||
|
||
self.log(f" 共找到 {len(crmids)} 条记录，开始解析详情...")
|
||
for i, cid in enumerate(crmids):
|
||
if self.stop_flag: break
|
||
self._process_single_id(cid)
|
||
self.log(f" 进度: {i + 1}/{len(crmids)}")
|
||
|
||
def _get_list_ids(self, limit_pages=3):
|
||
ids = []
|
||
for p in range(1, limit_pages + 1):
|
||
if self.stop_flag: break
|
||
try:
|
||
ts = int(time.time() * 1000)
|
||
url = f"{self.base_url}?module=SalesOrder&action=SalesOrderAjax&file=ListViewData&start={p}&actionId={ts}"
|
||
resp = self.session.get(url, headers=self.http_headers)
|
||
entries = resp.json().get('data', [])
|
||
if not entries: break
|
||
for item in entries:
|
||
if isinstance(item, dict):
|
||
ids.append(item.get('crmid') or item.get('id'))
|
||
except:
|
||
break
|
||
return list(set(ids))
|
||
|
||
def _process_date_range(self, s_str, e_str):
|
||
try:
|
||
t_start = datetime.strptime(s_str, "%Y-%m-%d")
|
||
t_end = datetime.strptime(e_str, "%Y-%m-%d").replace(hour=23, minute=59, second=59)
|
||
except:
|
||
self.log("❌ 日期格式错误")
|
||
return
|
||
|
||
page = 1
|
||
while not self.stop_flag:
|
||
ts = int(time.time() * 1000)
|
||
url = f"{self.base_url}?module=SalesOrder&action=SalesOrderAjax&file=ListViewData&sorder=DESC&order_by=modifiedtime&start={page}&pagesize=50&actionId={ts}&isFilter=true&search%5Bviewscope%5D=all_to_me&search%5Bviewname%5D=476"
|
||
try:
|
||
resp = self.session.get(url, headers=self.http_headers)
|
||
data = resp.json()
|
||
entries = data.get('data', []) or data.get('entries', [])
|
||
if not entries: break
|
||
|
||
page_ids = [x.get('crmid') or x.get('id') for x in entries if isinstance(x, dict)]
|
||
self.log(f" 🔎 正在检查第 {page} 页 ({len(page_ids)} 条)...")
|
||
|
||
valid_cnt = 0
|
||
for cid in page_ids:
|
||
if self.stop_flag: break
|
||
text = self.fetch_detail(cid)
|
||
r_time = self.extract_time(text)
|
||
|
||
if r_time:
|
||
if r_time > t_end: continue
|
||
if r_time < t_start:
|
||
self.log(f" 🛑 遇到旧数据 ({r_time})，停止爬取")
|
||
self.stop_flag = True
|
||
break
|
||
parsed = self.parse_data(text, cid)
|
||
if parsed:
|
||
self.on_data(parsed)
|
||
valid_cnt += 1
|
||
|
||
if valid_cnt > 0: self.log(f" ✅ 第 {page} 页入库 {valid_cnt} 条")
|
||
page += 1
|
||
if self.stop_flag: break
|
||
|
||
except Exception as e:
|
||
self.log(f"❌ 错误: {e}")
|
||
break
|
||
|
||
def _process_single_id(self, cid):
|
||
text = self.fetch_detail(cid)
|
||
parsed = self.parse_data(text, cid)
|
||
if parsed:
|
||
self.on_data(parsed)
|
||
|
||
|
||
# ================= 3. 界面显示类 (重大升级) =================
|
||
class CRMGUI(ttk.Window):
|
||
def __init__(self):
|
||
super().__init__(themename="cosmo") # 使用 cosmo 主题
|
||
self.title("CRM 智能数据助手 Pro Max")
|
||
self.geometry("1280x850")
|
||
|
||
self.crawler = CRMCrawler(self.log_msg, self.add_record_to_table)
|
||
self.is_running = False
|
||
|
||
# 内存数据结构：{'ASD': {'Domestic':[], 'Foreign':[], 'Other':[]}, 'NON_ASD': {...}}
|
||
self.stored_data = {
|
||
'ASD': {'Domestic': [], 'Foreign': [], 'Other': []},
|
||
'NON_ASD': {'Domestic': [], 'Foreign': [], 'Other': []}
|
||
}
|
||
|
||
# 引用字典，方便后续操作
|
||
self.treeviews = {}
|
||
|
||
# 基础字段
|
||
self.base_cols = [
|
||
"合同编号", "签署公司", "收款情况", "签订日期", "销售员", "厂家",
|
||
"最终用户单位", "最终用户信息联系人", "最终用户信息电话", "买方单位",
|
||
"厂家型号", "合同标的", "数量", "合同额", "合同总额",
|
||
"最晚发货期", "已收款", "未收款", "收款日期"
|
||
]
|
||
# 定制表头
|
||
self.cols_domestic = ["内贸合同号"] + self.base_cols + ["系统ID"]
|
||
self.cols_foreign = ["外贸合同号"] + self.base_cols + ["系统ID"]
|
||
self.cols_other = self.base_cols + ["系统ID"]
|
||
|
||
self.create_widgets()
|
||
|
||
def create_widgets(self):
|
||
# --- 1. 顶部：控制区 ---
|
||
control_frame = ttk.Frame(self, padding=10, bootstyle="light")
|
||
control_frame.pack(fill=X)
|
||
|
||
# 登录
|
||
login_grp = ttk.Labelframe(control_frame, text="身份验证", padding=10)
|
||
login_grp.pack(side=LEFT, padx=5, fill=Y)
|
||
ttk.Label(login_grp, text="用户:").pack(side=LEFT)
|
||
self.user_ent = ttk.Entry(login_grp, width=10);
|
||
self.user_ent.insert(0, "TEST");
|
||
self.user_ent.pack(side=LEFT, padx=5)
|
||
ttk.Label(login_grp, text="密码:").pack(side=LEFT)
|
||
self.pass_ent = ttk.Entry(login_grp, width=10, show="*");
|
||
self.pass_ent.insert(0, "***");
|
||
self.pass_ent.pack(side=LEFT, padx=5)
|
||
|
||
# 模式
|
||
mode_grp = ttk.Labelframe(control_frame, text="任务类型", padding=10)
|
||
mode_grp.pack(side=LEFT, padx=10, fill=Y, expand=True)
|
||
self.nb_mode = ttk.Notebook(mode_grp, bootstyle="primary")
|
||
self.nb_mode.pack(fill=BOTH, expand=True)
|
||
|
||
f_date = ttk.Frame(self.nb_mode, padding=10)
|
||
self.nb_mode.add(f_date, text="📅 按时间范围")
|
||
self.ent_start = ttk.Entry(f_date, width=12);
|
||
self.ent_start.insert(0, "2026-01-14");
|
||
self.ent_start.pack(side=LEFT, padx=5)
|
||
ttk.Label(f_date, text="至").pack(side=LEFT)
|
||
self.ent_end = ttk.Entry(f_date, width=12);
|
||
self.ent_end.insert(0, "2026-01-15");
|
||
self.ent_end.pack(side=LEFT, padx=5)
|
||
|
||
f_search = ttk.Frame(self.nb_mode, padding=10)
|
||
self.nb_mode.add(f_search, text="🔍 关键词搜索")
|
||
self.ent_query = ttk.Entry(f_search, width=25);
|
||
self.ent_query.pack(fill=X)
|
||
|
||
f_full = ttk.Frame(self.nb_mode, padding=10)
|
||
self.nb_mode.add(f_full, text="🚀 全量")
|
||
ttk.Label(f_full, text="数据量大，慎用").pack()
|
||
self.nb_mode.select(f_date)
|
||
|
||
# 运行按钮
|
||
btn_grp = ttk.Frame(control_frame, padding=10)
|
||
btn_grp.pack(side=RIGHT, fill=Y)
|
||
self.btn_run = ttk.Button(btn_grp, text="▶ 开始", bootstyle="success", command=self.start_thread, width=10)
|
||
self.btn_run.pack(side=TOP, pady=2)
|
||
self.btn_stop = ttk.Button(btn_grp, text="⏹ 停止", bootstyle="danger", command=self.stop_task, state=DISABLED,
|
||
width=10)
|
||
self.btn_stop.pack(side=TOP, pady=2)
|
||
|
||
# --- 2. 核心展示区 (解决颜色问题) ---
|
||
# 使用“切换按钮”代替顶层Tab，实现 [选中=蓝色实心] [未选中=白色空心]
|
||
toggle_frame = ttk.Frame(self, padding=(10, 5))
|
||
toggle_frame.pack(fill=X)
|
||
|
||
self.curr_view = tk.StringVar(value="ASD") # 默认 ASD
|
||
|
||
self.btn_view_asd = ttk.Button(toggle_frame, text="ASD 产品列表", command=lambda: self.switch_view("ASD"),
|
||
width=20)
|
||
self.btn_view_asd.pack(side=LEFT, padx=5)
|
||
|
||
self.btn_view_non = ttk.Button(toggle_frame, text="非 ASD 产品列表",
|
||
command=lambda: self.switch_view("NON_ASD"), width=20)
|
||
self.btn_view_non.pack(side=LEFT, padx=5)
|
||
|
||
# 容器 Frame
|
||
self.container = ttk.Frame(self)
|
||
self.container.pack(fill=BOTH, expand=True, padx=10)
|
||
|
||
# 创建两个大 Frame，分别装 ASD 和 NON_ASD 的内容
|
||
self.frame_asd = ttk.Frame(self.container)
|
||
self.frame_non = ttk.Frame(self.container)
|
||
|
||
# 初始化内部结构 (内贸/外贸/其他 分离)
|
||
self._init_inner_tabs(self.frame_asd, "ASD")
|
||
self._init_inner_tabs(self.frame_non, "NON_ASD")
|
||
|
||
# 默认显示 ASD
|
||
self.switch_view("ASD")
|
||
|
||
# --- 3. 底部区 ---
|
||
bottom_frame = ttk.Frame(self, padding=5)
|
||
bottom_frame.pack(fill=X, padx=10, pady=5)
|
||
|
||
log_frame = ttk.Labelframe(bottom_frame, text="系统日志", padding=5)
|
||
log_frame.pack(side=LEFT, fill=BOTH, expand=True)
|
||
self.txt_log = ScrolledText(log_frame, height=5)
|
||
self.txt_log.text.configure(state=DISABLED)
|
||
self.txt_log.pack(fill=BOTH, expand=True)
|
||
|
||
export_frame = ttk.Frame(bottom_frame, padding=10)
|
||
export_frame.pack(side=RIGHT, fill=Y)
|
||
ttk.Button(export_frame, text="📂 导出 Excel", bootstyle="primary", command=self.export_data).pack(fill=X,
|
||
pady=10)
|
||
|
||
def _init_inner_tabs(self, parent_frame, prefix):
|
||
"""在父Frame中创建 内贸/外贸/其他 的Tab结构"""
|
||
nb = ttk.Notebook(parent_frame, bootstyle="info")
|
||
nb.pack(fill=BOTH, expand=True)
|
||
|
||
# 内贸 Tab
|
||
f_dom = ttk.Frame(nb);
|
||
nb.add(f_dom, text="内贸 (Domestic)")
|
||
self._create_treeview(f_dom, self.cols_domestic, f"{prefix}_Domestic")
|
||
|
||
# 外贸 Tab
|
||
f_for = ttk.Frame(nb);
|
||
nb.add(f_for, text="外贸 (Foreign)")
|
||
self._create_treeview(f_for, self.cols_foreign, f"{prefix}_Foreign")
|
||
|
||
# 其他 Tab
|
||
f_oth = ttk.Frame(nb);
|
||
nb.add(f_oth, text="其他 (Other)")
|
||
self._create_treeview(f_oth, self.cols_other, f"{prefix}_Other")
|
||
|
||
def _create_treeview(self, parent, cols, key):
|
||
"""创建表格并注册到 self.treeviews"""
|
||
# 滚动条
|
||
sy = ttk.Scrollbar(parent, orient=VERTICAL)
|
||
sx = ttk.Scrollbar(parent, orient=HORIZONTAL)
|
||
|
||
tv = ttk.Treeview(parent, columns=cols, show="headings", selectmode="browse",
|
||
yscrollcommand=sy.set, xscrollcommand=sx.set)
|
||
|
||
sy.config(command=tv.yview);
|
||
sy.pack(side=RIGHT, fill=Y)
|
||
sx.config(command=tv.xview);
|
||
sx.pack(side=BOTTOM, fill=X)
|
||
tv.pack(side=LEFT, fill=BOTH, expand=True)
|
||
|
||
for c in cols:
|
||
tv.heading(c, text=c)
|
||
w = 100
|
||
if c in ["合同标的", "最终用户单位", "签署公司", "买方单位"]:
|
||
w = 200
|
||
elif c == "系统ID":
|
||
w = 0
|
||
tv.column(c, width=w, minwidth=50)
|
||
|
||
# 绑定双击
|
||
tv.bind("<Double-1>", lambda e: self.on_double_click(e, tv, key))
|
||
# 绑定右键菜单
|
||
tv.bind("<Button-3>", lambda e: self.on_right_click(e, tv, key))
|
||
|
||
self.treeviews[key] = tv
|
||
return tv
|
||
|
||
def switch_view(self, view_name):
|
||
"""切换 ASD / NON_ASD 视图，并处理按钮颜色反转"""
|
||
self.curr_view.set(view_name)
|
||
|
||
if view_name == "ASD":
|
||
self.frame_non.pack_forget()
|
||
self.frame_asd.pack(fill=BOTH, expand=True)
|
||
# ASD选中：ASD实心(primary)，NON空心(outline)
|
||
self.btn_view_asd.configure(bootstyle="primary")
|
||
self.btn_view_non.configure(bootstyle="secondary-outline")
|
||
else:
|
||
self.frame_asd.pack_forget()
|
||
self.frame_non.pack(fill=BOTH, expand=True)
|
||
# NON选中：ASD空心，NON实心
|
||
self.btn_view_asd.configure(bootstyle="secondary-outline")
|
||
self.btn_view_non.configure(bootstyle="primary")
|
||
|
||
# --- 逻辑控制 ---
|
||
def start_thread(self):
|
||
if self.is_running: return
|
||
# 清空所有数据和表格
|
||
self.stored_data = {
|
||
'ASD': {'Domestic': [], 'Foreign': [], 'Other': []},
|
||
'NON_ASD': {'Domestic': [], 'Foreign': [], 'Other': []}
|
||
}
|
||
for tv in self.treeviews.values():
|
||
for item in tv.get_children(): tv.delete(item)
|
||
|
||
self.is_running = True
|
||
self.crawler.stop_flag = False
|
||
self.btn_run.config(state=DISABLED)
|
||
self.btn_stop.config(state=NORMAL)
|
||
|
||
t = threading.Thread(target=self._worker)
|
||
t.daemon = True
|
||
t.start()
|
||
|
||
def stop_task(self):
|
||
self.crawler.stop_flag = True
|
||
self.log_msg("🛑 正在停止...")
|
||
|
||
def _worker(self):
|
||
user = self.user_ent.get()
|
||
pwd = self.pass_ent.get()
|
||
if not self.crawler.login(user, pwd):
|
||
self._reset_ui();
|
||
return
|
||
|
||
curr_idx = self.nb_mode.index(self.nb_mode.select())
|
||
mode = "date"
|
||
kwargs = {}
|
||
if curr_idx == 0:
|
||
mode = "date";
|
||
kwargs = {'start': self.ent_start.get(), 'end': self.ent_end.get()}
|
||
elif curr_idx == 1:
|
||
mode = "search";
|
||
kwargs = {'query': self.ent_query.get()}
|
||
elif curr_idx == 2:
|
||
mode = "full"
|
||
|
||
try:
|
||
self.crawler.run_task(mode, **kwargs)
|
||
self.log_msg("🎉 完成！")
|
||
except Exception as e:
|
||
self.log_msg(f"❌ 错误: {e}")
|
||
finally:
|
||
self._reset_ui()
|
||
|
||
def _reset_ui(self):
|
||
self.is_running = False
|
||
self.after(0, lambda: self.btn_run.config(state=NORMAL))
|
||
self.after(0, lambda: self.btn_stop.config(state=DISABLED))
|
||
|
||
def log_msg(self, msg):
|
||
self.after(0, lambda: self._append_log(msg))
|
||
|
||
def _append_log(self, msg):
|
||
self.txt_log.text.configure(state=NORMAL)
|
||
self.txt_log.text.insert(END, f"[{datetime.now().strftime('%H:%M:%S')}] {msg}\n")
|
||
self.txt_log.text.see(END)
|
||
self.txt_log.text.configure(state=DISABLED)
|
||
|
||
# --- 数据分发逻辑 (核心) ---
|
||
def add_record_to_table(self, record):
|
||
def _update():
|
||
# 1. 确定大类
|
||
main_key = 'ASD' if record['IS_ASD'] else 'NON_ASD'
|
||
|
||
# 2. 确定子类 (内贸/外贸/其他)
|
||
c_no = str(record.get("合同编号", "")).strip().upper()
|
||
sub_key = "Other"
|
||
if c_no.startswith('N'):
|
||
sub_key = "Domestic"
|
||
elif c_no.startswith('W'):
|
||
sub_key = "Foreign"
|
||
|
||
# 3. 存入内存
|
||
self.stored_data[main_key][sub_key].append(record)
|
||
|
||
# 4. 插入对应表格
|
||
# 组合 Key 找到对应的 Treeview
|
||
tv_key = f"{main_key}_{sub_key}"
|
||
tv = self.treeviews.get(tv_key)
|
||
|
||
if tv:
|
||
# 获取该表格对应的列
|
||
# 注意：columns 是 tuple，需要转 list
|
||
cols = list(tv['columns'])
|
||
vals = [record.get(c, "") for c in cols]
|
||
|
||
# iid 设为列表索引，方便查找
|
||
idx = len(self.stored_data[main_key][sub_key]) - 1
|
||
tv.insert("", END, iid=idx, values=vals)
|
||
|
||
self.after(0, _update)
|
||
|
||
# --- 编辑与跳转逻辑 ---
|
||
def on_right_click(self, event, tv, key):
|
||
"""右键菜单"""
|
||
item_id = tv.identify_row(event.y)
|
||
if not item_id: return
|
||
tv.selection_set(item_id)
|
||
|
||
# 解析 key (例如 "ASD_Domestic")
|
||
parts = key.split('_')
|
||
main_key = parts[0]
|
||
if len(parts) > 2: main_key = f"{parts[0]}_{parts[1]}" # 防止 NON_ASD 这种
|
||
sub_key = parts[-1]
|
||
|
||
record = self.stored_data[main_key][sub_key][int(item_id)]
|
||
crm_id = record.get("系统ID", "")
|
||
|
||
menu = tk.Menu(self, tearoff=0)
|
||
menu.add_command(label="🌐 在浏览器查看", command=lambda: self.open_browser(crm_id))
|
||
menu.add_command(label="📝 编辑详情", command=lambda: self.show_detail_popup(record, tv, item_id))
|
||
menu.post(event.x_root, event.y_root)
|
||
|
||
def on_double_click(self, event, tv, key):
|
||
item_id = tv.selection()
|
||
if not item_id: return
|
||
idx = int(item_id[0])
|
||
|
||
parts = key.split('_')
|
||
main_key = parts[0]
|
||
if len(parts) > 2: main_key = f"{parts[0]}_{parts[1]}"
|
||
sub_key = parts[-1]
|
||
|
||
record = self.stored_data[main_key][sub_key][idx]
|
||
self.show_detail_popup(record, tv, item_id)
|
||
|
||
def open_browser(self, crm_id):
|
||
if crm_id:
|
||
url = f"http://111.198.24.44:88/index.php?module=SalesOrder&action=DetailView&record={crm_id}"
|
||
webbrowser.open(url)
|
||
self.log_msg(f"🌐 跳转: {crm_id}")
|
||
|
||
def show_detail_popup(self, record, tv, item_id):
|
||
top = ttk.Toplevel(self)
|
||
top.title(f"订单详情: {record.get('合同编号')}")
|
||
top.geometry("600x700")
|
||
|
||
# 滚动容器
|
||
canvas = tk.Canvas(top)
|
||
sb = ttk.Scrollbar(top, orient="vertical", command=canvas.yview)
|
||
f_scroll = ttk.Frame(canvas)
|
||
f_scroll.bind("<Configure>", lambda e: canvas.configure(scrollregion=canvas.bbox("all")))
|
||
canvas.create_window((0, 0), window=f_scroll, anchor="nw")
|
||
canvas.configure(yscrollcommand=sb.set)
|
||
canvas.pack(side="left", fill="both", expand=True)
|
||
sb.pack(side="right", fill="y")
|
||
|
||
# 滚轮支持
|
||
canvas.bind_all("<MouseWheel>", lambda e: canvas.yview_scroll(int(-1 * (e.delta / 120)), "units"))
|
||
|
||
# 按钮
|
||
crm_id = record.get("系统ID", "")
|
||
ttk.Button(f_scroll, text="🌐 浏览器查看原始网页", bootstyle="info-outline",
|
||
command=lambda: self.open_browser(crm_id)).grid(row=0, column=0, columnspan=2, pady=10)
|
||
|
||
# 字段编辑
|
||
entries = {}
|
||
row = 1
|
||
# 显示该表格对应的所有列
|
||
cols = list(tv['columns'])
|
||
|
||
for field in cols:
|
||
if field == "系统ID": continue
|
||
ttk.Label(f_scroll, text=field + ":").grid(row=row, column=0, sticky=E, padx=5, pady=5)
|
||
ent = ttk.Entry(f_scroll, width=40)
|
||
ent.insert(0, str(record.get(field, "")))
|
||
ent.grid(row=row, column=1, padx=5, pady=5)
|
||
entries[field] = ent
|
||
row += 1
|
||
|
||
def save():
|
||
for k, e in entries.items(): record[k] = e.get()
|
||
new_vals = [record.get(c, "") for c in cols]
|
||
tv.item(item_id, values=new_vals)
|
||
top.destroy()
|
||
ToastNotification("保存成功", "本地数据已更新", 1500).show_toast()
|
||
|
||
ttk.Button(f_scroll, text="💾 保存修改", bootstyle="success", command=save).grid(row=row, column=0, columnspan=2,
|
||
pady=20)
|
||
|
||
# --- 导出 ---
|
||
def export_data(self):
|
||
folder = filedialog.askdirectory()
|
||
if not folder: return
|
||
self.log_msg(f"💾 正在导出...")
|
||
ts = time.strftime("%Y%m%d_%H%M%S")
|
||
|
||
# 完整的字段映射，用于导出时的列顺序（比界面显示的更多更全）
|
||
export_cols = [
|
||
"合同编号", "签署公司", "收款情况", "签订日期", "销售员", "厂家",
|
||
"最终用户单位", "最终用户信息联系人", "最终用户信息电话", "最终用户信息邮箱", "最终用户所在地",
|
||
"买方单位", "买方信息联系人", "买方信息电话", "买方信息邮箱",
|
||
"厂家型号", "合同标的", "数量", "单位", "折扣率（％）", "合同额", "合同总额",
|
||
"外购付款方式", "最晚发货期", "已收款", "未收款", "收款日期"
|
||
]
|
||
|
||
for main_key, prefix in [('ASD', 'ASD_产品表'), ('NON_ASD', 'Non_ASD_产品表')]:
|
||
data_map = self.stored_data[main_key]
|
||
# data_map 结构: {'Domestic': [records], 'Foreign': [], ...}
|
||
|
||
# 检查是否为空
|
||
total = sum(len(v) for v in data_map.values())
|
||
if total == 0: continue
|
||
|
||
path = os.path.join(folder, f"{prefix}_{ts}.xlsx")
|
||
try:
|
||
with pd.ExcelWriter(path, engine='openpyxl') as writer:
|
||
# 内贸 Sheet
|
||
if data_map['Domestic']:
|
||
df = pd.DataFrame(data_map['Domestic'])
|
||
# 插入内贸号
|
||
cols = export_cols[:2] + ["内贸合同号"] + export_cols[2:]
|
||
df = df.reindex(columns=cols)
|
||
df.to_excel(writer, sheet_name='内贸', index=False)
|
||
|
||
# 外贸 Sheet
|
||
if data_map['Foreign']:
|
||
df = pd.DataFrame(data_map['Foreign'])
|
||
# 插入外贸号
|
||
cols = export_cols[:2] + ["外贸合同号"] + export_cols[2:]
|
||
df = df.reindex(columns=cols)
|
||
df.to_excel(writer, sheet_name='外贸', index=False)
|
||
|
||
# 其他 Sheet
|
||
if data_map['Other']:
|
||
df = pd.DataFrame(data_map['Other'])
|
||
cols = export_cols[:2] + ["内贸合同号"] + export_cols[2:] # 默认用内贸结构
|
||
df = df.reindex(columns=cols)
|
||
df.to_excel(writer, sheet_name='其他', index=False)
|
||
|
||
self.log_msg(f" ✅ 导出成功: {os.path.basename(path)}")
|
||
except Exception as e:
|
||
self.log_msg(f" ❌ 导出失败: {e}")
|
||
|
||
Messagebox.show_info("导出完成", "任务结束")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
app = CRMGUI()
|
||
app.mainloop() |