fix: 移除ERP密码和API密钥硬编码，改为环境变量；erp-sync接口加鉴权

2026-06-05 07:51:20 +08:00
parent 2c92168e0e
commit 7b9ef8ac27
9 changed files with 844 additions and 8 deletions
@@ -0,0 +1,130 @@
+"""补充更多KPI-ERP数据同步
+运行: python3 scripts/extend_erp_sync.py
+"""
+import sys, os, logging
+from os import getenv
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from app.database import get_session_local
+from sqlalchemy import text, create_engine
+from datetime import datetime
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s [%(levelname)s] %(message)s')
+logger = logging.getLogger('erp_extend')
+
+erp_user = os.getenv("ERP_DB_USER", "zxbtest")
+erp_pass = os.getenv("ERP_DB_PASS")
+erp_host = os.getenv("ERP_DB_HOST", "211.149.143.215")
+erp_port = os.getenv("ERP_DB_PORT", "1433")
+erp_db = os.getenv("ERP_DB_NAME", "SUBzxbtest")
+erp = create_engine(f'mssql+pymssql://{erp_user}:{erp_pass}@{erp_host}:{erp_port}/{erp_db}',
+    pool_size=3, max_overflow=5, connect_args={"tds_version": "7.0"})
+db = get_session_local()()
+
+kpi_map = {}
+rows = db.execute(text("SELECT id, kpi_code FROM kpi_definitions")).fetchall()
+for r in rows:
+    kpi_map[r[1]] = r[0]
+
+batch = f"erp_sync_{datetime.now().strftime('%Y%m%d_%H%M')}"
+written = 0
+
+def p2ym(p):
+    base_year, base_period = 2021, 3
+    diff = p - base_period
+    year = base_year + diff // 12
+    month = (diff % 12) + 3
+    if month > 12: month -= 12; year += 1
+    return f"{year}-{month:02d}"
+
+def write_kpi(code, period, value):
+    global written
+    kid = kpi_map.get(code)
+    if not kid or value is None: return
+    try:
+        db.execute(text("""
+            INSERT INTO kpi_values (kpi_id, period, actual_value, source_type, source_batch, data_status, calculated_at)
+            VALUES (:kpi_id, :period, :value, 'erp', :batch, 'verified', NOW())
+            ON DUPLICATE KEY UPDATE actual_value=:value2, source_batch=:batch2, data_status='verified', calculated_at=NOW()
+        """), {"kpi_id": kid, "period": period, "value": float(value),
+               "batch": batch, "value2": float(value), "batch2": batch})
+        db.commit()
+        written += 1
+    except Exception as e:
+        db.rollback()
+
+with erp.connect() as c:
+    # 获取全期间销售收入
+    rev_rows = c.execute(text("""
+        SELECT Period, SUM(SumMoney) as rev, COALESCE(SUM(SumCostMoney),0) as cost
+        FROM MasterBill WHERE BillType=1 AND BillState>=3
+        GROUP BY Period ORDER BY Period
+    """)).fetchall()
+    
+    # 库存总额
+    total_inv = float(c.execute(text("SELECT COALESCE(SUM(CostMoney),0) FROM Storage WHERE CostMoney > 0")).fetchone()[0])
+    avg_inv = max(total_inv, 1)
+    
+    # 应收账款总额
+    total_ar = float(c.execute(text("SELECT COALESCE(SUM(AReceive),0) FROM Units WHERE AReceive > 0")).fetchone()[0])
+    avg_ar = max(total_ar, 1)
+
+for row in rev_rows:
+    p = row[0]
+    if p < 41: continue
+    ps = p2ym(p)
+    rev = float(row.rev)
+    cost = float(row.cost)
+    gross = round((rev - cost) / rev * 100, 2) if rev > 0 else 0
+    turnover = round(rev / avg_ar, 2) if avg_ar > 0 else 0
+    inv_turn = round(cost / avg_inv, 2) if avg_inv > 0 else 0
+    inv_days = round(365 / inv_turn, 1) if inv_turn > 0 else 0
+    
+    write_kpi("F_REVENUE_001", ps, rev)
+    write_kpi("F_PROFIT_001", ps, gross)
+    write_kpi("F_AR_001", ps, turnover)
+    write_kpi("P_INV_001", ps, inv_turn)
+    write_kpi("P_INV_004", ps, inv_days)
+
+# 前5客户集中度 和 逾期应收 另查
+with erp.connect() as c:
+    for row in rev_rows:
+        p = row[0]
+        if p < 41: continue
+        ps = p2ym(p)
+        
+        r = c.execute(text("""
+            SELECT TOP 5 SUM(SumMoney) as amt FROM (
+                SELECT SUM(SumMoney) as SumMoney FROM MasterBill 
+                WHERE BillType=1 AND BillState>=3 AND Period=:p GROUP BY Unit_ID
+            ) t ORDER BY amt DESC
+        """), {"p": p}).fetchall()
+        top5 = sum(float(r2[0]) for r2 in r)
+        ratio = round(top5 / float(row.rev) * 100, 2) if float(row.rev) > 0 else 0
+        write_kpi("C_CUST_002", ps, ratio)
+        
+        r2 = c.execute(text("""
+            SELECT SUM(CASE WHEN DATEDIFF(day, BillDate, GETDATE()) > 30 THEN SumMoney ELSE 0 END) as overdue
+            FROM MasterBill WHERE BillType=1 AND BillState>=3 AND Period=:p
+        """), {"p": p}).fetchone()
+        overdue = float(r2[0]) if r2 and r2[0] else 0
+        overdue_ratio = round(overdue / float(row.rev) * 100, 2) if float(row.rev) > 0 else 0
+        write_kpi("F_AR_002", ps, overdue_ratio)
+    
+    # 费用控制率
+    r = c.execute(text("""
+        SELECT Period,
+               SUM(CASE WHEN BillType=1 THEN SumMoney ELSE 0 END) as rev,
+               SUM(CASE WHEN BillType=30 THEN SumMoney ELSE 0 END) as expense
+        FROM MasterBill WHERE BillState>=3 GROUP BY Period ORDER BY Period
+    """)).fetchall()
+    for row in r:
+        p = row[0]
+        if p < 41: continue
+        ps = p2ym(p)
+        exp_rev = float(row.rev)
+        exp_amt = float(row.expense)
+        exp_ratio = round(exp_amt / exp_rev * 100, 2) if exp_rev > 0 else 0
+        write_kpi("F_COST_001", ps, exp_ratio)
+
+db.close()
+logger.info(f"完成! 共写入 {written} 条")