import requests
from Entity.Variables import prologueList
from Utils.LogManager import LogManager

BaseUrl = "https://crawlclient.api.yolozs.com/api/common/"


class Requester():
    comment = "comment"
    prologue = "prologue"

    @classmethod
    def requestPrologue(cls, token):
        headers = {
            "vvtoken": token,
        }
        url = BaseUrl + cls.prologue
        result = requests.get(headers=headers, url=url)
        json = result.json()
        data = json.get("data")
        for i in data:
            prologueList.append(i)

    # 翻译
    @classmethod
    def translation(cls, msg, country="英国"):
        try:
            parame = {
                "msg": msg,
                "country": country,
            }
            url = "http://ai.zhukeping.com/translation"
            result = requests.request(url=url, json=parame, method="POST")
            json = result.json()
            data = json.get("data")
            print(data)
            return data
        except Exception as e:
            LogManager.method_error(f"翻译失败,报错的原因:{e}","翻译失败")

    # ai聊天
    @classmethod
    def chatToAi(cls, param):
        url = "http://ai.zhukeping.com/chat"
        result = requests.request(url=url, json=param, method="POST")
        json = result.json()
        data = json.get("data", {})
        return data