#!/usr/bin/python3.12
"""翻译 finance-data.json 为韩文，保存到 ko-finance-data.json"""
import json, urllib.request, os, re

SRC = '/data/news/json/finance-data.json'
OUT = '/data/news/json/ko-finance-data.json'
KEY_FILE = '/root/.openclaw/openclaw.json'

key = json.load(open(KEY_FILE))['models']['providers']['deepseek']['apiKey']
data = json.load(open(SRC, encoding='utf-8'))

def translate(text):
    if not text or text.strip() == '' or not re.search(r'[\u4e00-\u9fff]', text):
        return text
    payload = json.dumps({
        "model": "deepseek-chat",
        "messages": [{"role": "user", "content": "将以下中文翻译成韩文：\n" + text}],
        "temperature": 0.1, "max_tokens": 200
    }).encode()
    req = urllib.request.Request('https://api.deepseek.com/v1/chat/completions', data=payload,
        headers={'Authorization': 'Bearer ' + key, 'Content-Type': 'application/json'})
    r = json.loads(urllib.request.urlopen(req, timeout=30).read())
    return r['choices'][0]['message']['content'].strip()

for item in data.get('finance_asia_date', []):
    if 'name' in item:
        item['name'] = translate(item['name'])

with open(OUT, 'w', encoding='utf-8') as f:
    json.dump(data, f, ensure_ascii=False, indent=2)
print('ko-finance-data.json 已保存')