28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
# ComfyUI_DropSendNode/__init__.py
|
|
# Register DropSend Setup + AutoUploader nodes
|
|
from .dropsend_uploader_node import NODE_CLASS_MAPPINGS as RUN_CLASS, NODE_DISPLAY_NAME_MAPPINGS as RUN_DISPLAY
|
|
from .dropsend_setup_node import NODE_CLASS_MAPPINGS as SETUP_CLASS, NODE_DISPLAY_NAME_MAPPINGS as SETUP_DISPLAY
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_CLASS_MAPPINGS.update(RUN_CLASS)
|
|
NODE_CLASS_MAPPINGS.update(SETUP_CLASS)
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(RUN_DISPLAY)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(SETUP_DISPLAY)
|
|
# Define web directory for JavaScript extensions
|
|
WEB_DIRECTORY = "./web"
|
|
# Register OAuth callback route with ComfyUI server
|
|
def register_oauth_routes():
|
|
"""Register OAuth callback routes with ComfyUI's PromptServer"""
|
|
try:
|
|
from server import PromptServer
|
|
from .oauth_handler import handle_oauth_callback
|
|
|
|
# Add the OAuth callback route
|
|
PromptServer.instance.app.router.add_get('/oauth/dropbox/callback', handle_oauth_callback)
|
|
print("[DropSend] Registered OAuth callback route: /oauth/dropbox/callback")
|
|
|
|
except Exception as e:
|
|
print(f"[DropSend] Warning: Could not register OAuth routes: {e}")
|
|
# Register routes when the node is loaded
|
|
register_oauth_routes()
|