# encoding:utf-8
__author__ = 'ObsessedCE'

from abc import ABC, abstractmethod


class Translation(ABC):

    @abstractmethod
    def translation(self, text,  src_language = "auto", des_language = "zh"):
        pass

#coding=utf-8
__author__ = 'ObsessedCE'
import requests
import file_util
import sys
import hashlib
import random
import string
import translation

class BAIDUTranslation(translation.ABC):

    def __init__(self, profile):
        self.appid = profile.get("baidu_app_id")
        self.secret = profile.get("baidu_secret")
        self.url = profile.get("baidu_translation_url")
        self.session = requests.Session()
        self.session.trust_env = False

    def translation(self, text,  src_language = "auto", des_language = "zh"):
        header = {
            "content-type":"application/x-www-form-urlencoded"
        }
        data = {
            "q" : text,
            "from" : src_language,
            "to" : des_language,
            "appid":self.appid,
            "salt":"",
            "sign":""
        }

        if not text:
            print("Unspecified content")
            return None
        salt = BAIDUTranslation.generate_random_string(6)
        data["salt"] = salt
        data["sign"] = self.generate_sign(data)

        try:
            response = self.session.post(url= self.url, data = data, headers = header)
            if response.status_code != 200:
                print(f"調(diào)用百度翻譯出錯，狀態(tài)碼為: {str(response.status_code)}")
                return None
            response_data = response.json()
            if response_data.get("code") and response_data.get("code" )!= 52000:
                print(f"調(diào)用百度翻譯出錯，返回錯誤代碼為: {response_data.get('code')}")
                return None
            translation_result = list()
            trans_result = response_data.get("trans_result")
            for result in trans_result:
                src = result.get("src")
                des = result.get("dst")
                translation_result.append(des)
            return translation_result
        except Exception as e:
            print(f"調(diào)用翻譯請求時出現(xiàn)錯誤: {e}")



    def generate_sign(self, data):
        """
        生成簽名
        :param data:
        :return:
        """
        str = data.get("appid")
        str += data.get("q")
        str += data.get("salt")
        str += self.secret
        return  self.generate_md5(str)

    def generate_md5(self, content):
        """
        進行內(nèi)容md5加密，發(fā)揮全小寫的編碼
        :param content:
        :return:
        """
        if content:
            md5_hash = hashlib.md5()
            md5_hash.update(content.encode("utf-8"))
            return md5_hash.hexdigest().lower()

    @staticmethod
    def generate_random_string(length=6):
        # 可用字符：大寫字母、小寫字母和數(shù)字
        chars = string.ascii_letters + string.digits
        # 隨機選擇字符并生成指定長度的字符串
        return ''.join(random.choices(chars, k=length))


if __name__ == "__main__":
    profile = file_util.read_file("./profile.yml")
    if not profile:
        print("no profile")
        sys.exit(0)
    baidu_transltion = BAIDUTranslation(profile)
    text = "You look so handsome today\nI think so, too"
    print(baidu_transltion.translation(text, ))

# encoding:utf-8
__author__ = 'ObsessedCE'
import yaml
import sys

def read_file(file_path):
    try:
        with  open(file_path, "r", encoding="utf-8") as file:
            data = yaml.safe_load(file)
            return data
    except Exception as e:
        print(e)
        sys.exit(0)

baidu_app_id: "個人信息查看"
baidu_secret: "個人信息中查看"
baidu_translation_url: "https://fanyi-api.baidu.com/api/trans/vip/translate"