102 lines
4.0 KiB
Python
102 lines
4.0 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_input = {}
|
|
nodes_output = {}
|
|
|
|
|
|
# Extraire les nœuds WorkflowInput et WorkflowOutput
|
|
for node in workflow_file["nodes"]:
|
|
if node.get("type") == "WorkflowInput":
|
|
# Convertir au format compatible pour le client
|
|
node_id = str(node.get("id", "unknown"))
|
|
nodes_input[node_id] = {
|
|
"class_type": "WorkflowInput",
|
|
"inputs": node.get("widgets_values", {}),
|
|
"position": node.get("pos", {})[1]
|
|
}
|
|
elif node.get("type") == "WorkflowOutput":
|
|
node_id = str(node.get("id", "unknown"))
|
|
nodes_output[node_id] = {
|
|
"class_type": "WorkflowOutput",
|
|
"inputs": node.get("widgets_values", {}),
|
|
"position": node.get("pos", {})[1]
|
|
}
|
|
# sort by position
|
|
nodes_input = dict(sorted(nodes_input.items(), key=lambda item: item[1]["position"]))
|
|
nodes_output = dict(sorted(nodes_output.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)
|
|
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 {file_path}: {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')
|