#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""企旺陪跑 · 每日自动数据更新"""
import json, urllib.request, os
from datetime import datetime

TOKEN_FILE = '/home/ubuntu/.openclaw/workspace/data/paiduoduo-token.txt'
DATA_FILE = '/var/www/xwznt/tracesource/data.json'

def api(url):
    with open(TOKEN_FILE) as f:
        tok = json.load(f)['token']
    req = urllib.request.Request(url, headers={'token': tok})
    return json.loads(urllib.request.urlopen(req, timeout=15).read())

def update():
    tm = api('http://java.qancloud.com/oa/trademark/search?page=1&limit=1').get('count', 0)
    crm = api('http://java.qancloud.com/crm/base/search?page=1&limit=1').get('count', 0)
    pt = 14997
    
    data = {
        "updated_at": datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
        "total_trademarks": tm,
        "total_customers": crm,
        "total_patents": pt,
        "total_all": tm + crm + pt
    }
    
    os.makedirs(os.path.dirname(DATA_FILE), exist_ok=True)
    with open(DATA_FILE, 'w') as f:
        json.dump(data, f, indent=2)
    
    print(f"✅ {datetime.now()}: 商标{tm:,} 客户{crm:,} 总计{data['total_all']:,}")

if __name__ == '__main__':
    update()