| import os |
| import json |
| from aiohttp import web |
| import logging |
|
|
|
|
| class AppSettings(): |
| def __init__(self, user_manager): |
| self.user_manager = user_manager |
|
|
| def get_settings(self, request): |
| try: |
| file = self.user_manager.get_request_user_filepath( |
| request, |
| "comfy.settings.json" |
| ) |
| except KeyError as e: |
| logging.error("User settings not found.") |
| raise web.HTTPUnauthorized() from e |
| if os.path.isfile(file): |
| try: |
| with open(file) as f: |
| return json.load(f) |
| except: |
| logging.error(f"The user settings file is corrupted: {file}") |
| return {} |
| 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) |
|
|
