Allow custom words

2023-08-27 21:24:36 +01:00
parent e35fd2c7fe
commit 4ecef08774
4 changed files with 37 additions and 5 deletions
@@ -0,0 +1,19 @@
+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") as f:
+        f.write(request.text())