wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-10 01:02:56 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
qwertzman-png 2023-04-21 00:10:10 +02:00 committed by GitHub
parent c88398f75f
commit ae4df2aa34
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server.py
View File

@ -69,6 +69,7 @@ class PromptServer():
self.last_node_id = None self.last_node_id = None
self.client_id = None self.client_id = None
self.poll_id = 0
self.poll_messages = [] self.poll_messages = []
@routes.get('/ws') @routes.get('/ws')
@ -194,7 +195,7 @@ class PromptServer():
@routes.get("/poll_messages") @routes.get("/poll_messages")
async def get_poll_messages(request): async def get_poll_messages(request):
return web.json_response({k: v for v, k in enumerate(self.poll_messages)}) return web.json_response(self.poll_messages))
@routes.get("/queue") @routes.get("/queue")
async def get_queue(request): async def get_queue(request):
@ -286,7 +287,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({"type": event, "data": data, "sid": sid}) self.poll_messages.append({"type": event, "data": data, "sid": sid, "poll_id": poll_id++})
if isinstance(message, str) == False: if isinstance(message, str) == False:
message = json.dumps(message) message = json.dumps(message)