fk_server.py

import os
from server import PromptServer
from aiohttp import web
import json

class Cancelled(Exception):
    pass
def read_file_content(file_path):
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            return file.read()
    except FileNotFoundError:
        return '{"appid": "","key": "","help": "在上方填写百度翻译API的appid和key,注意你需要申请好翻译权限，否则翻译会不成功~"}'
    except Exception as e:
         return '{"appid": "","key": "","help": "在上方填写百度翻译API的appid和key,注意你需要申请好翻译权限，否则翻译会不成功~"}'
    return None
def string_to_json(json_string):
    try:
        json_object = json.loads(json_string)
        return json_object
    except json.JSONDecodeError as e:
        print(f"JSON解析错误: {e}")
        return None
    
@PromptServer.instance.routes.get("/fk_server")
async def fksapi(request):
         tget = request.rel_url.query
         gtype =  tget['type']
         config_path = os.path.join(os.path.dirname(__file__), "ini.json")
         if gtype == "getpz":
                return web.json_response(string_to_json(read_file_content(config_path)), content_type='application/json')
         else:
             return web.json_response({"v":gtype}, content_type='application/json')
@PromptServer.instance.routes.get("/fkhome")
async def fkweb(request): 
         return web.Response(
                text="hi",
                content_type="text/html",
            )
@PromptServer.instance.routes.post('/fk_server')
async def fkpostapi(request):
    post = await request.post()
    bdappid = post.get("bdappid")
    bdappkey = post.get("appbdkey") 
    if bdappid and bdappkey:
        config_path = os.path.join(os.path.dirname(__file__), "ini.json")
        config_data = string_to_json(read_file_content(config_path))
        config_data["appid"] = bdappid
        config_data["key"] = bdappkey
        config_data["help"] = "在上方填写百度翻译API的appid和key,注意你需要申请好翻译权限，否则翻译会不成功哦~"
        with open(config_path, 'w', encoding='utf-8') as file:
             json.dump(config_data, file, ensure_ascii=False, indent=4)
    return web.json_response({})