import requests, re, time, json

rec_ids = Context['argv']['rec_id']  # 改為復(fù)數(shù)，獲取所有記錄ID
src_texts = Context['argv']['src_text']  # 改為復(fù)數(shù)，獲取所有文本
url = 'http://api.niutrans.com/NiuTransServer/translation'
apikey = "c4e10095a8c951a6443789345c498e9a" #不保證能用，可以自己申請免費的

def is_english_strict(text):
    if not text:
        return False
    english_chars = re.findall(r'[a-zA-Z]', text)
    if ' ' in english_chars:
        return False
    else:
        return len(english_chars) / len(text) > 0.7

def niu_trans(sentence, src_lan="auto", tgt_lan="zh"):
    data = {"from": src_lan, "to": tgt_lan, "apikey": apikey, "src_text": sentence}
    res = requests.post(url, data=data)
    res_dict = json.loads(res.text)
    if "tgt_text" in res_dict:
        result = res_dict['tgt_text']
    else:
        result = res.text
    return result

def translate(word):
    headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.99 Safari/537.36"}
    url = r"https://dict.cn/search?q=" + word
    try:
        resp = requests.get(url, headers=headers, timeout=30).text
        phonetic = re.search('<bdo lang="EN-US">(.*?)</bdo>', resp)
        pattern = re.compile(r'<ul class="dict-basic-ul">(.*?)</ul>', re.DOTALL)
        matches = pattern.findall(resp)
        mean = []
        if matches:
            content = matches[0]
            meanings = re.findall(r'<li>.*?<span>(.*?)</span>.*?<strong>(.*?)</strong>.*?</li>', content, re.DOTALL)
            for part_of_speech, meaning in meanings:
                mean.append(f"{part_of_speech}{meaning}")
            word_info = {
                "單詞": word,
                "音標(biāo)": phonetic.group(1) if phonetic else "",
                "詞義": "".join(mean) if mean else ""
            }
            return word_info
    except Exception as e:
        return f"翻譯失敗: {str(e)}"

# 循環(huán)處理每一個記錄
for i in range(len(rec_ids)):
    rec_id = rec_ids[i]
    src_text = src_texts[i]
    
    try:
        df = dbt(sheet_name='數(shù)據(jù)表')
        if is_english_strict(src_text):
            info = translate(src_text)
            # 構(gòu)建數(shù)據(jù)記錄 - 使用字段名作為鍵
            record_data = {
                "_rid": rec_id,
                "音標(biāo)": info["音標(biāo)"] if isinstance(info, dict) else "",
                "詞義": info["詞義"] if isinstance(info, dict) else info
            }
        else:
            # 構(gòu)建數(shù)據(jù)記錄 - 使用字段名作為鍵
            record_data = {
                "_rid": rec_id,
                "詞義": niu_trans(src_text)
            }

        # 更新當(dāng)前記錄
        update_dbt(data=record_data, sheet_name='數(shù)據(jù)表')
        time.sleep(0.6)  # 每個記錄處理完后等待0.6秒
        
    except Exception as e:
        print(f"處理記錄 {rec_id} 失敗: {str(e)}")

print(f"? 批量處理完成，共處理了 {len(rec_ids)} 條記錄")