wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-10 09:12:31 +08:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

poll messages

Browse Source
This commit is contained in:
qwertzman-png 2023-04-20 23:53:04 +02:00 committed by GitHub
parent 96b57a9ad6
commit e485357de4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server.py
View File

@ -69,6 +69,8 @@ class PromptServer():
self.last_node_id = None self.last_node_id = None
self.client_id = None self.client_id = None
self.poll_messages = []
@routes.get('/ws') @routes.get('/ws')
async def websocket_handler(request): async def websocket_handler(request):
ws = web.WebSocketResponse() ws = web.WebSocketResponse()
@ -190,6 +192,10 @@ class PromptServer():
async def get_history(request): async def get_history(request):
return web.json_response(self.prompt_queue.get_history()) return web.json_response(self.prompt_queue.get_history())
@routes.get("/poll_messages")
async def get_poll_messages(request):
return web.json_response({k: v for v, k in enumerate(self.poll_messages)})
@routes.get("/queue") @routes.get("/queue")
async def get_queue(request): async def get_queue(request):
queue_info = {} queue_info = {}
@ -280,6 +286,7 @@ class PromptServer():
async def send(self, event, data, sid=None): async def send(self, event, data, sid=None):
message = {"type": event, "data": data} message = {"type": event, "data": data}
self.poll_messages.append(message)
if isinstance(message, str) == False: if isinstance(message, str) == False:
message = json.dumps(message) message = json.dumps(message)