import requests
import fileBase

def get_will_convert_file_list():
    url = "https://pm.ljsea.top/file/get_file_will_convert_content?super_id=1"  # 替换为实际的API URL
    #post请求
    headers ={
        "token":"5413vfd3sdgtr56hrtagvbbmla",
    }
    response = requests.post(url, headers=headers)
    if response.status_code == 200:
        return response.json()
    else:
        print(f"请求失败，状态码: {response.status_code}, 错误信息: {response.text}")
        return None

def get_user_file_info(user_id,user_file_id):
    url = "https://pm.ljsea.top/file/file_list?super_id=" + str(user_id)  # 替换为实际的API URL
    #post请求
    headers = {
        "token":"5413vfd3sdgtr56hrtagvbbmla",
    }
    data = {
        "type": "one",
        "file_id": user_file_id,  #用户文件id
    }
    response = requests.post(url, headers=headers,json=data)
    if response.status_code == 200:
        res = response.json()
        if res["code"] == 0:
            return res["data"]
        else:
            print("获取文件内容失败:",res)
            return None
    else:
        print(f"请求失败，状态码: {response.status_code}, 错误信息: {response.text}")
        return None


def create_file_content(fileID,fileContent):
    url = "https://pm.ljsea.top/file/create_file_content?super_id=1"  # 替换为实际的API URL
    #post请求
    headers ={
        "token":"5413vfd3sdgtr56hrtagvbbmla",
    }
    data = {
        "file_id": fileID,  #实际文件id
        "file_content": fileContent
    }
    response = requests.post(url, headers=headers,json=data)
    if response.status_code == 200:
        return response.json()
    else:
        print(f"请求失败，状态码: {response.status_code}, 错误信息: {response.text}")
        return None

def get_file_content(id):
    url = "https://pm.ljsea.top/file/find_file_content?super_id=1"  # 替换为实际的API URL
    #post请求
    headers = {
        "token":"5413vfd3sdgtr56hrtagvbbmla",
    }
    data = {
        "user_file_id": id #实际文件id
    }
    response = requests.post(url, headers=headers,json=data)
    if response.status_code == 200:
        res = response.json()
        if res["code"] == 0:
            return res["data"]
        else:
            print("获取文件内容失败:",res)
            return None
    else:
        print(f"请求失败，状态码: {response.status_code}, 错误信息: {response.text}")
        return None

def get_file():
    file_list_resp = get_will_convert_file_list()
    if file_list_resp["code"] == 0:
        file_list = file_list_resp["data"]
        return file_list
    else:
        print("获取文件列表失败:",file_list_resp)
        return None


def get_file_add_convert():
    file_list = get_file()
    if file_list:
        s = {"epub","txt","docx","pdf","doc", "go", "c", "cpp", "java", "py", "js", "html", "css", "json", "xml", "yaml", "md"}
        for file in file_list:
            #文件类型是否允许
            fileType = file["FileStoreName"].split(".")[-1]
            if fileType in s:
                print("文件类型允许")
            else:
                print("文件类型不允许:", file["FileStoreName"])
                continue
            print("文件ID:", file["ID"])
            print("文件名称:",  file["FileName"],  "\t", file["FileStoreName"])
            text = fileBase.get_file_content(file["FileStoreName"])
            if text is None:
                print("文件内容获取失败")
                continue
            print("文件内容长度:", len(text))
            #将文件内容上传
            print(create_file_content(file["ID"], text))
            #break
    else:
        print("未能获取文件列表")

def get_Kbase_server_id():
    url = "https://pm.ljsea.top/file/find_file_content?super_id=1"  # 替换为实际的API URL
    #post请求
    headers = {
        "token":"5413vfd3sdgtr56hrtagvbbmla",
    }
    data = {
        "user_file_id": id #实际文件id
    }
    response = requests.post(url, headers=headers,json=data)
    if response.status_code == 200:
        res = response.json()
        if res["code"] == 0:
            return res["data"]
        else:
            print("获取文件内容失败:",res)
            return None
    else:
        print(f"请求失败，状态码: {response.status_code}, 错误信息: {response.text}")
        return None

# if __name__ == "__main__":
#     get_file_add_convert()