128 lines
3.6 KiB
Python
128 lines
3.6 KiB
Python
import json
|
|
import os
|
|
import folder_paths
|
|
|
|
|
|
def get_allowed_dirs():
|
|
dir = os.path.abspath(os.path.join(__file__, "../../user"))
|
|
file = os.path.join(dir, "text_file_dirs.json")
|
|
with open(file, "r") as f:
|
|
return json.loads(f.read())
|
|
|
|
|
|
def get_valid_dirs():
|
|
return get_allowed_dirs().keys()
|
|
|
|
|
|
def get_dir_from_name(name):
|
|
dirs = get_allowed_dirs()
|
|
if name not in dirs:
|
|
raise KeyError(name + " dir not found")
|
|
|
|
path = dirs[name]
|
|
path = path.replace("$input", folder_paths.get_input_directory())
|
|
path = path.replace("$output", folder_paths.get_output_directory())
|
|
path = path.replace("$temp", folder_paths.get_temp_directory())
|
|
return path
|
|
|
|
|
|
def is_child_dir(parent_path, child_path):
|
|
parent_path = os.path.abspath(parent_path)
|
|
child_path = os.path.abspath(child_path)
|
|
return os.path.commonpath([parent_path]) == os.path.commonpath([parent_path, child_path])
|
|
|
|
|
|
def get_real_path(dir):
|
|
dir = dir.replace("/**/", "/")
|
|
dir = os.path.abspath(dir)
|
|
dir = os.path.split(dir)[0]
|
|
return dir
|
|
|
|
|
|
def get_file(root_dir, file):
|
|
if file == "[none]" or not file or not file.strip():
|
|
raise ValueError("No file")
|
|
|
|
root_dir = get_dir_from_name(root_dir)
|
|
root_dir = get_real_path(root_dir)
|
|
if not os.path.exists(root_dir):
|
|
os.mkdir(root_dir)
|
|
full_path = os.path.join(root_dir, file)
|
|
|
|
if not is_child_dir(root_dir, full_path):
|
|
raise ReferenceError()
|
|
|
|
return full_path
|
|
|
|
|
|
class LoadTextLocal:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"root_dir": (list(get_valid_dirs()), {}),
|
|
"file": (["[none]"], {
|
|
"pysssss.binding": [{
|
|
"source": "root_dir",
|
|
"callback": [{
|
|
"type": "set",
|
|
"target": "$this.disabled",
|
|
"value": True
|
|
}, {
|
|
"type": "fetch",
|
|
"url": "/pysssss/text-file/{$source.value}",
|
|
"then": [{
|
|
"type": "set",
|
|
"target": "$this.options.values",
|
|
"value": "$result"
|
|
}, {
|
|
"type": "validate-combo"
|
|
}, {
|
|
"type": "set",
|
|
"target": "$this.disabled",
|
|
"value": False
|
|
}]
|
|
}],
|
|
}]
|
|
}),
|
|
"encoding": ("STRING", {"default": "utf-8"}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
|
|
FUNCTION = "load"
|
|
|
|
CATEGORY = "不忘科技-自定义节点🚩"
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(self, root_dir, file, **kwargs):
|
|
if file == "[none]" or not file or not file.strip():
|
|
return True
|
|
get_file(root_dir, file)
|
|
return True
|
|
|
|
def load(self, root_dir, file, encoding):
|
|
with open(get_file(root_dir,file), "r", encoding=encoding) as f:
|
|
return (f.read(),)
|
|
|
|
class LoadTextOnline:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"file_path":("STRING", {"default": "input/"}),
|
|
"encoding": ("STRING", {"default": "utf-8"}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
|
|
FUNCTION = "load"
|
|
|
|
CATEGORY = "不忘科技-自定义节点🚩"
|
|
|
|
def load(self, file_path, encoding):
|
|
with open(file_path, "r", encoding=encoding) as f:
|
|
return (f.read(),)
|