121 lines
4.9 KiB
Python
121 lines
4.9 KiB
Python
import server
|
|
from aiohttp import web
|
|
import os
|
|
import json
|
|
import folder_paths
|
|
from app.user_manager import UserManager
|
|
|
|
|
|
def get_workflow_data(workflow_file):
|
|
nodes = {"WorkflowInput": {}, "WorkflowOutput": {}}
|
|
# Extraire les nœuds WorkflowInput et WorkflowOutput
|
|
# print(workflow_file["nodes"]);
|
|
for node in workflow_file["nodes"]:
|
|
|
|
node_type = node.get("type")
|
|
if node_type in nodes.keys():
|
|
# Convertir au format compatible pour le client
|
|
node_id = str(node.get("id", "unknown"))
|
|
w_values = node.get("widgets_values", [])
|
|
inp = []
|
|
if node_type == "WorkflowInput":
|
|
if len(w_values) < 3:
|
|
inp = [w_values[0], node.get("inputs",[])[0].get("type", "")]
|
|
if len(w_values) == 2:
|
|
inp.append(w_values[1])
|
|
else:
|
|
inp = w_values
|
|
|
|
elif node_type == "WorkflowOutput":
|
|
#print(node.get("outputs",[]))
|
|
#print(w_values)
|
|
if ('Name' in w_values):
|
|
inp = [w_values['Name']['value'], node.get("outputs",[])[0].get("type", "*")]
|
|
elif len(w_values) < 3:
|
|
inp = [w_values[0], node.get("outputs",[])[0].get("type", "*")]
|
|
if len(w_values) == 2:
|
|
inp.append(w_values[1])
|
|
else:
|
|
inp = w_values
|
|
#print(inp)
|
|
nodes[node_type][node_id] = {
|
|
"class_type": node_type,
|
|
"inputs": inp
|
|
}
|
|
if type(node.get("pos")) is list:
|
|
nodes[node_type][node_id]["position"] = node["pos"][1]
|
|
else:
|
|
nodes[node_type][node_id]["position"] = node["pos"]['1']
|
|
# sort by position
|
|
nodes_input = dict(sorted(nodes["WorkflowInput"].items(), key=lambda item: item[1]["position"]))
|
|
nodes_output = dict(sorted(nodes["WorkflowOutput"].items(), key=lambda item: item[1]["position"]))
|
|
return {"inputs": nodes_input,
|
|
"outputs": nodes_output,
|
|
'workflow': workflow_file}
|
|
|
|
|
|
@server.PromptServer.instance.routes.get("/flowchain/workflows")
|
|
async def workflows(request):
|
|
user = UserManager().get_request_user_id(request)
|
|
json_path = os.path.join(folder_paths.user_directory, user, "workflows")
|
|
result = {}
|
|
|
|
# Vérifier si le répertoire principal existe
|
|
if os.path.exists(json_path):
|
|
# Utiliser os.walk pour parcourir récursivement tous les sous-répertoires
|
|
for root, dirs, files in os.walk(str(json_path)):
|
|
for file in files:
|
|
# Ne traiter que les fichiers JSON
|
|
if file.lower().endswith('.json'):
|
|
file_path = os.path.join(root, file)
|
|
|
|
try:
|
|
with open(file_path, "r", encoding="utf-8") as f:
|
|
json_content = json.load(f)
|
|
# print(file_path);
|
|
relative_path = os.path.relpath(file_path, str(json_path))
|
|
if "nodes" in json_content:
|
|
file_conf = get_workflow_data(json_content)
|
|
if file_conf["inputs"] or file_conf["outputs"]:
|
|
result[relative_path] = file_conf
|
|
|
|
except json.JSONDecodeError:
|
|
# Ignorer les fichiers JSON mal formés
|
|
print(f"Ignoring malformed JSON file: {file_path}")
|
|
except Exception as e:
|
|
print(f"Error processing, probably old format: {str(e)}")
|
|
else:
|
|
# Créer le répertoire s'il n'existe pas
|
|
os.makedirs(json_path)
|
|
result["No file in worflows folder"] = {"inputs": {}, "outputs": {}}
|
|
|
|
# Si aucun fichier valide n'a été trouvé
|
|
if not result:
|
|
result["No compatible workflow files found"] = {"inputs": {}, "outputs": {}}
|
|
|
|
return web.json_response(result, content_type='application/json')
|
|
|
|
|
|
@server.PromptServer.instance.routes.get("/flowchain/workflow")
|
|
async def workflow(request):
|
|
user = UserManager().get_request_user_id(request)
|
|
result = {}
|
|
original_path = request.query.get("workflow_path")
|
|
unversal_path = original_path.replace("\\", "/")
|
|
json_path = unversal_path.split("/")
|
|
if ".json" in json_path[-1]:
|
|
# file_name = json_path[-1]
|
|
json_path = folder_paths.user_directory + "/" + user + "/workflows/" + unversal_path
|
|
|
|
if os.path.exists(json_path):
|
|
with open(json_path, "r", encoding="utf-8") as f:
|
|
json_content = json.load(f)
|
|
if "nodes" in json_content:
|
|
result = get_workflow_data(json_content)
|
|
else:
|
|
result = {"error": "File not found"}
|
|
else:
|
|
result = {"error": "File not found"}
|
|
|
|
return web.json_response(result, content_type='application/json')
|