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)
|