-
Notifications
You must be signed in to change notification settings - Fork 0
/
__init__.py
151 lines (134 loc) · 4.04 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
# coding=utf-8
import os
import json
import platform
import sys
import numpy as np
import builtins
import torch
import shutil
import hashlib
import atexit
import server
import gc
import execution
import folder_paths
from functools import lru_cache
from aiohttp import web
from pathlib import Path
from PIL import Image
from PIL.PngImagePlugin import PngInfo
VERSION = "0.0.1"
ADDON_NAME = "COMFYUI-DOWNLOAD-ANY" # 存放在web/extensions下的前端html和js的文件夹名
COMFY_PATH = Path(folder_paths.__file__).parent
CUR_PATH = Path(__file__).parent
# @server.PromptServer.instance.routes.post("/yxcda/helloworld")
@server.PromptServer.instance.routes.get("/yxcda/helloworld")
async def get_hellowworld(request):
"""
@des: 自定义http接口
如果是post, js这么请求
static syncTranslation(OnFinished = () => { }) {
var locale = localStorage.getItem(TUtils.LOCALE_ID) || "en-US";
var url = "./agl/get_translation";
var request = new XMLHttpRequest();
request.open("post", url);
request.setRequestHeader("Content-Type", "application/x-www-form-urlencoded");
request.send(`locale=${locale}`);
request.onload = function () {
/* XHR对象获取到返回信息后执行 */
if (request.status != 200)
return;
var resp = JSON.parse(request.responseText);
for (var key in TUtils.T) {
if (key in resp)
TUtils.T[key] = resp[key];
else
TUtils.T[key] = {};
}
TUtils.T.Locales = LOCALES;
// 合并NodeCategory 到 Menu
TUtils.Menu = Object.assign(TUtils.T.Menu, TUtils.T.NodeCategory);
// 提取 Node 中 key 到 Menu
for (let key in TUtils.T.Nodes) {
let node = TUtils.T.Nodes[key];
TUtils.Menu[key] = node["title"] || key;
}
OnFinished();
};
}
如果是get
api
.fetchApi("/yxcda/helloworld")
.then((response) => response.text())
.then((data) => {
alert(data);
});
"""
# post = await request.post()
# locale = post.get("locale", "en_US")
json_data = "{}"
try:
json_data = json.dumps({'hello':'world'}, ensure_ascii=False)
except Exception as e:
sys.stderr.write(f"[yxcda/helloworld error]: {e}\n")
sys.stderr.flush()
return web.Response(status=200, body=json_data)
def rmtree(path: Path):
# unlink symbolic link
if Path(path.resolve()).as_posix() != path.as_posix():
path.unlink()
return
if not path.exists():
return
if path.is_file():
path.unlink()
elif path.is_dir():
# 移除 .git
if path.name == ".git":
if platform.system() == "darwin":
from subprocess import call
call(['rm', '-rf', path.as_posix()])
elif platform.system() == "Windows":
os.system(f'rd/s/q "{path.as_posix()}"')
return
for child in path.iterdir():
rmtree(child)
try:
path.rmdir() # nas 的共享盘可能会有残留
except BaseException:
...
def register():
"""
@des: 将整个项目包含js都注册到comfyui的web中
"""
cda_ext_path = COMFY_PATH.joinpath("web", "extensions", ADDON_NAME)
rmtree(cda_ext_path)
try:
if os.name == "nt":
import _winapi
_winapi.CreateJunction(CUR_PATH.as_posix(), cda_ext_path.as_posix())
else:
# 复制时过滤 .git
shutil.copytree(CUR_PATH.as_posix(), cda_ext_path.as_posix(), ignore=shutil.ignore_patterns(".git"))
except Exception as e:
sys.stderr.write(f"[yxcda/register error]: {e}\n")
sys.stderr.flush()
return
def unregister():
"""
@des: 移除整个项目包含js都注册到comfyui的web中
"""
# 移除缓存json
# for data in CUR_PATH.glob("*.json"):
# if not data.name.startswith("translations_"):
# continue
# data.unlink()
cda_ext_path = COMFY_PATH.joinpath("web", "extensions", ADDON_NAME)
try:
rmtree(cda_ext_path)
except BaseException:
...
register()
atexit.register(unregister)
NODE_CLASS_MAPPINGS = {}