Ethanfel/ComfyUI-Network-Sync
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
Ethanfel
2026-01-05 16:15:43 +01:00
parent 478b2c6196
commit 2453165e12
1 changed files with 39 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
__init__.py
View File

@@ -3,21 +3,25 @@ import json
import aiohttp import aiohttp
from aiohttp import web from aiohttp import web
import server import server
import folder_paths # <--- USE COMFTYUI PATH MANAGER
# --- Configuration --- # --- Improved Path Configuration ---
# Where to save received workflows on the destination machine # This attempts to save where ComfyUI looks for workflows by default
SAVE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../user/default/workflows")) base_path = folder_paths.base_path
# If the above path doesn't work for your setup, use the output folder: # Try the modern 'user/default/workflows' path first
# SAVE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../output/synced_workflows")) SAVE_DIR = os.path.join(base_path, "user", "default", "workflows")
# If that folder structure doesn't exist (older comfy), fallback to 'output'
if not os.path.exists(os.path.dirname(SAVE_DIR)):
SAVE_DIR = folder_paths.get_output_directory()
if not os.path.exists(SAVE_DIR): if not os.path.exists(SAVE_DIR):
os.makedirs(SAVE_DIR) os.makedirs(SAVE_DIR)
class NetworkSync: class NetworkSync:
def __init__(self): pass
pass
# 1. RECEIVE Endpoint (Runs on the Destination Machine) # 1. RECEIVE Endpoint
@server.PromptServer.instance.routes.post("/network-sync/receive") @server.PromptServer.instance.routes.post("/network-sync/receive")
async def receive_workflow(request): async def receive_workflow(request):
try: try:
@@ -25,54 +29,56 @@ async def receive_workflow(request):
filename = data.get("name", "synced_workflow") filename = data.get("name", "synced_workflow")
workflow = data.get("workflow") workflow = data.get("workflow")
# Basic security: prevent saving outside the folder
filename = os.path.basename(filename)
if not filename.endswith(".json"): if not filename.endswith(".json"):
filename += ".json" filename += ".json"
# Sanitize filename to prevent directory traversal
filename = os.path.basename(filename)
save_path = os.path.join(SAVE_DIR, filename) save_path = os.path.join(SAVE_DIR, filename)
# Write the file
with open(save_path, "w", encoding="utf-8") as f: with open(save_path, "w", encoding="utf-8") as f:
json.dump(workflow, f, indent=2) json.dump(workflow, f, indent=2)
print(f"[NetworkSync] Workflow saved to: {save_path}") print(f"✅ [NetworkSync] Received and updated: {filename}")
return web.json_response({"status": "success", "message": f"Saved {filename}"}) return web.json_response({"status": "success", "message": f"Saved to {save_path}"})
except Exception as e: except Exception as e:
print(f"[NetworkSync] Error receiving: {e}") print(f"❌ [NetworkSync] Error: {e}")
return web.json_response({"status": "error", "message": str(e)}, status=500) return web.json_response({"status": "error", "message": str(e)}, status=500)
# 2. PUSH Endpoint (Runs on the Source Machine) # 2. PUSH Endpoint
@server.PromptServer.instance.routes.post("/network-sync/push") @server.PromptServer.instance.routes.post("/network-sync/push")
async def push_workflow(request): async def push_workflow(request):
try: try:
data = await request.json() data = await request.json()
target_ip = data.get("target_ip") # e.g., "192.168.1.5:8188" target_ip = data.get("target_ip")
filename = data.get("name")
workflow = data.get("workflow") # Auto-correct IP formatting
if "http" not in target_ip:
target_ip = "http://" + target_ip
# Ensure port is present if missing (assume default 8188)
if ":" not in target_ip.replace("http://", "").replace("https://", ""):
target_ip += ":8188"
if not target_ip: target_url = f"{target_ip}/network-sync/receive"
return web.json_response({"status": "error", "message": "No target IP provided"}, status=400)
payload = {
"name": data.get("name"),
"workflow": data.get("workflow")
}
# Ensure protocol is present # Timeout added so ComfyUI doesn't freeze if target is offline
if not target_ip.startswith("http"): timeout = aiohttp.ClientTimeout(total=5)
target_url = f"http://{target_ip}/network-sync/receive" async with aiohttp.ClientSession(timeout=timeout) as session:
else:
target_url = f"{target_ip}/network-sync/receive"
payload = {"name": filename, "workflow": workflow}
# Send from Python (Server A) to Python (Server B)
async with aiohttp.ClientSession() as session:
async with session.post(target_url, json=payload) as resp: async with session.post(target_url, json=payload) as resp:
if resp.status == 200: if resp.status == 200:
return web.json_response({"status": "success", "message": "Synced successfully!"}) return web.json_response({"status": "success"})
else: else:
text = await resp.text() return web.json_response({"status": "error", "message": f"Target error: {resp.status}"})
return web.json_response({"status": "error", "message": f"Remote error: {text}"}, status=resp.status)
except Exception as e: except Exception as e:
print(f"[NetworkSync] Error pushing: {e}")
return web.json_response({"status": "error", "message": str(e)}, status=500) return web.json_response({"status": "error", "message": str(e)}, status=500)
NODE_CLASS_MAPPINGS = {} NODE_CLASS_MAPPINGS = {}