64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import os
|
|
import sys
|
|
import json
|
|
import urllib.parse
|
|
import subprocess
|
|
import threading
|
|
import asyncio
|
|
from aiohttp import web
|
|
import folder_paths
|
|
import struct
|
|
|
|
async def api_get_config(request):
|
|
script_dir = os.path.dirname(os.path.abspath(__file__))
|
|
config_path = os.path.join(script_dir, "config.json")
|
|
has_key = False
|
|
folder_types = []
|
|
if os.path.exists(config_path):
|
|
try:
|
|
with open(config_path, 'r', encoding='utf-8') as f:
|
|
cfg = json.load(f)
|
|
has_key = bool(cfg.get("CIVITAI_API_KEY", "").strip())
|
|
folder_types = cfg.get("folder_types_config", [])
|
|
except:
|
|
pass
|
|
return web.json_response({"has_api_key": has_key, "folder_types_config": folder_types})
|
|
|
|
async def api_save_config(request):
|
|
try:
|
|
data = await request.json()
|
|
api_key = data.get("api_key")
|
|
folder_types_config = data.get("folder_types_config")
|
|
|
|
script_dir = os.path.dirname(os.path.abspath(__file__))
|
|
config_path = os.path.join(script_dir, "config.json")
|
|
|
|
cfg = {}
|
|
if os.path.exists(config_path):
|
|
try:
|
|
with open(config_path, 'r', encoding='utf-8') as f:
|
|
cfg = json.load(f)
|
|
except:
|
|
pass
|
|
|
|
if api_key is not None:
|
|
cfg["CIVITAI_API_KEY"] = api_key.strip()
|
|
|
|
if folder_types_config is not None:
|
|
cfg["folder_types_config"] = folder_types_config
|
|
|
|
if "physical_folders_config" in data:
|
|
cfg["physical_folders_config"] = data.get("physical_folders_config")
|
|
|
|
if "folder_view_mode" in data:
|
|
cfg["folder_view_mode"] = data.get("folder_view_mode")
|
|
|
|
with open(config_path, 'w', encoding='utf-8') as f:
|
|
json.dump(cfg, f, indent=4)
|
|
|
|
return web.json_response({"status": "ok"})
|
|
except Exception as e:
|
|
return web.json_response({"status": "error", "message": str(e)})
|
|
|
|
|