Skip to content
GitLab
菜单
为什么选择 GitLab
定价
联系销售
探索
为什么选择 GitLab
定价
联系销售
探索
登录
获取免费试用
主导航
搜索或转到…
项目
C
ComfyUI
管理
动态
成员
代码
仓库
分支
提交
标签
仓库图
比较修订版本
锁定的文件
部署
模型注册表
分析
模型实验
帮助
帮助
支持
GitLab 文档
比较 GitLab 各版本
社区论坛
为极狐GitLab 提交贡献
提交反馈
隐私声明
快捷键
?
新增功能
4
代码片段
群组
项目
显示更多面包屑
hanamizuki
ComfyUI
提交
0faee118
未验证
提交
0faee118
编辑于
1 year ago
作者:
Dr.Lt.Data
提交者:
GitHub
1 year ago
浏览文件
操作
下载
补丁
差异文件
support on prompt event handler (#765)
Co-authored-by:
Lt.Dr.Data
<
lt.dr.data@gmail.com
>
上级
b8c7c770
No related branches found
No related tags found
无相关合并请求
变更
1
隐藏空白变更内容
行内
左右并排
显示
1 个更改的文件
server.py
+17
-0
17 个添加, 0 个删除
server.py
有
17 个添加
和
0 个删除
server.py
+
17
−
0
浏览文件 @
0faee118
import
os
import
sys
import
asyncio
import
traceback
import
nodes
import
folder_paths
import
execution
...
...
@@ -88,6 +90,8 @@ class PromptServer():
self
.
last_node_id
=
None
self
.
client_id
=
None
self
.
on_prompt_handlers
=
[]
@routes.get
(
'
/ws
'
)
async
def
websocket_handler
(
request
):
ws
=
web
.
WebSocketResponse
()
...
...
@@ -438,6 +442,7 @@ class PromptServer():
resp_code
=
200
out_string
=
""
json_data
=
await
request
.
json
()
json_data
=
self
.
trigger_on_prompt
(
json_data
)
if
"
number
"
in
json_data
:
number
=
float
(
json_data
[
'
number
'
])
...
...
@@ -606,3 +611,15 @@ class PromptServer():
if
call_on_start
is
not
None
:
call_on_start
(
address
,
port
)
def
add_on_prompt_handler
(
self
,
handler
):
self
.
on_prompt_handlers
.
append
(
handler
)
def
trigger_on_prompt
(
self
,
json_data
):
for
handler
in
self
.
on_prompt_handlers
:
try
:
json_data
=
handler
(
json_data
)
except
Exception
as
e
:
print
(
f
"
[ERROR] An error occurred during the on_prompt_handler processing
"
)
traceback
.
print_exc
()
return
json_data
此差异已折叠。
点击以展开。
预览
0%
加载中
请重试
或
添加新附件
.
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
保存评论
取消
想要评论请
注册
或
登录