Peyton DeNiro
5d70efdc88
Declare Python 3.12 support, replace the hidden pip bootstrap with explicit uv sync guidance, and remove prompt() from the Web UI save flow. Add focused contract coverage for the metadata and frontend defaults.
56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
import sys
|
|
import os
|
|
|
|
from io import StringIO
|
|
|
|
import traceback
|
|
|
|
from aiohttp import web
|
|
|
|
ext_dir = os.path.dirname(__file__)
|
|
sys.path.append(ext_dir)
|
|
|
|
try:
|
|
import black
|
|
except ImportError:
|
|
raise ImportError(
|
|
"ComfyUI-to-Python-Extension requires the project dependencies to be installed. "
|
|
f"Run 'uv sync' in {ext_dir} with Python 3.12+ before loading this extension."
|
|
) from None
|
|
|
|
# Prevent reimporting of custom nodes
|
|
os.environ["RUNNING_IN_COMFYUI"] = "TRUE"
|
|
|
|
from comfyui_to_python import ComfyUItoPython
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname(ext_dir)))
|
|
|
|
import server
|
|
|
|
WEB_DIRECTORY = "js"
|
|
NODE_CLASS_MAPPINGS = {}
|
|
|
|
|
|
@server.PromptServer.instance.routes.post("/saveasscript")
|
|
async def save_as_script(request):
|
|
try:
|
|
data = await request.json()
|
|
name = data["name"]
|
|
workflow = data["workflow"]
|
|
frontend_workflow = data.get("frontend_workflow")
|
|
|
|
sio = StringIO()
|
|
ComfyUItoPython(
|
|
workflow=workflow,
|
|
frontend_workflow=frontend_workflow,
|
|
output_file=sio,
|
|
)
|
|
|
|
sio.seek(0)
|
|
data = sio.read()
|
|
|
|
return web.Response(text=data, status=200)
|
|
except Exception as e:
|
|
traceback.print_exc()
|
|
return web.Response(text=str(e), status=500)
|