Spaces:
Running
on
L40S
Running
on
L40S
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): | |
async def get_settings(request): | |
return web.json_response(self.get_settings(request)) | |
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) | |
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) | |
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) |