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

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


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

    @classmethod
    def requestPrologue(cls, token):
        try:
            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)
        except Exception as e:
            LogManager.method_error(f"获取requestPrologue失败,报错的原因:{e}", "获取requestPrologue异常")

    # 翻译
    @classmethod
    def translation(cls, msg, country="英国"):
        try:
            parame = {
                "msg": msg,
                "country": country,
            }
            url = "https://ai.yolozs.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):
        aiConfig = JsonUtils.read_json("aiConfig")
        agentName = aiConfig.get("agentName")
        guildName = aiConfig.get("guildName")
        contactTool = aiConfig.get("contactTool", "")
        contact = aiConfig.get("contact", "")

        inputs = {
            "agentName":agentName,
            "guildName":guildName,
            "contactTool":contactTool,
            "contact":contact
        }

        param["inputs"] = inputs

        try:
            url = "https://ai.yolozs.com/chat"
            result = requests.request(url=url, json=param, method="POST")
            json = result.json()
            data = json.get("data", {})
            return data
        except Exception as e:
            LogManager.method_error(f"ai聊天失败,ai聊天出现异常,报错的原因:{e}", "ai聊天接口异常")