|
from server import PromptServer |
|
from aiohttp import web |
|
import os |
|
|
|
dir = os.path.abspath(os.path.join(__file__, "../../user")) |
|
if not os.path.exists(dir): |
|
os.mkdir(dir) |
|
file = os.path.join(dir, "autocomplete.txt") |
|
|
|
|
|
@PromptServer.instance.routes.get("/pysssss/autocomplete") |
|
async def get_autocomplete(request): |
|
if os.path.isfile(file): |
|
return web.FileResponse(file) |
|
return web.Response(status=404) |
|
|
|
|
|
@PromptServer.instance.routes.post("/pysssss/autocomplete") |
|
async def update_autocomplete(request): |
|
with open(file, "w", encoding="utf-8") as f: |
|
f.write(await request.text()) |
|
return web.Response(status=200) |
|
|
