mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-01-11 10:25:16 +00:00
235727fed7
* wip per user data * Rename, hide menu * better error rework default user * store pretty * Add userdata endpoints Change nodetemplates to userdata * add multi user message * make normal arg * Fix tests * Ignore user dir * user tests * Changed to default to browser storage and add server-storage arg * fix crash on empty templates * fix settings added before load * ignore parse errors
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
import os
|
|
import json
|
|
from aiohttp import web
|
|
|
|
|
|
class AppSettings():
|
|
def __init__(self, user_manager):
|
|
self.user_manager = user_manager
|
|
|
|
def get_settings(self, request):
|
|
file = self.user_manager.get_request_user_filepath(
|
|
request, "comfy.settings.json")
|
|
if os.path.isfile(file):
|
|
with open(file) as f:
|
|
return json.load(f)
|
|
else:
|
|
return {}
|
|
|
|
def save_settings(self, request, settings):
|
|
file = self.user_manager.get_request_user_filepath(
|
|
request, "comfy.settings.json")
|
|
with open(file, "w") as f:
|
|
f.write(json.dumps(settings, indent=4))
|
|
|
|
def add_routes(self, routes):
|
|
@routes.get("/settings")
|
|
async def get_settings(request):
|
|
return web.json_response(self.get_settings(request))
|
|
|
|
@routes.get("/settings/{id}")
|
|
async def get_setting(request):
|
|
value = None
|
|
settings = self.get_settings(request)
|
|
setting_id = request.match_info.get("id", None)
|
|
if setting_id and setting_id in settings:
|
|
value = settings[setting_id]
|
|
return web.json_response(value)
|
|
|
|
@routes.post("/settings")
|
|
async def post_settings(request):
|
|
settings = self.get_settings(request)
|
|
new_settings = await request.json()
|
|
self.save_settings(request, {**settings, **new_settings})
|
|
return web.Response(status=200)
|
|
|
|
@routes.post("/settings/{id}")
|
|
async def post_setting(request):
|
|
setting_id = request.match_info.get("id", None)
|
|
if not setting_id:
|
|
return web.Response(status=400)
|
|
settings = self.get_settings(request)
|
|
settings[setting_id] = await request.json()
|
|
self.save_settings(request, settings)
|
|
return web.Response(status=200) |