77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
import sys
|
|
import os
|
|
import textwrap
|
|
import code
|
|
|
|
from . import register_node
|
|
|
|
|
|
@register_node
|
|
class RestoreStdStreams(object):
|
|
# ComfyUI-Manager patches sys.stdout and sys.stder
|
|
# which breaks GNU Readline support and makes the
|
|
# REPL annoying to use. This context manager temporarily
|
|
# puts back the originals.
|
|
|
|
def __enter__(self):
|
|
self._stdout, self._stderr = sys.stdout, sys.stderr
|
|
sys.stdout, sys.stderr = sys.__stdout__, sys.__stderr__
|
|
self._stdout.flush(); self._stderr.flush()
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
_stdout, _stderr = sys.stdout, sys.stderr
|
|
sys.stdout, sys.stderr = self._stdout, self._stderr
|
|
_stdout.flush(); _stderr.flush()
|
|
|
|
|
|
EOF = 'Ctrl-Z plus Return' if os.sep == '\\' else 'Ctrl-D (i.e. EOF)'
|
|
MESSAGE = f"Press {EOF} to continue processing."
|
|
|
|
|
|
class Quitter:
|
|
# Replace exit() and quit() to make sure the user exits with EOF
|
|
# otherwise the whole server gets terminated.
|
|
|
|
def __repr__(self):
|
|
return MESSAGE
|
|
|
|
def __call__(self):
|
|
print(MESSAGE)
|
|
|
|
|
|
class Interact:
|
|
"""Opens an interactive REPL whenever the node is evaluated."""
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {},
|
|
"optional": {x: "*" for x in ('a', 'b', 'c', 'd')},
|
|
}
|
|
|
|
RETURN_TYPES = ()
|
|
RETURN_NAMES = ()
|
|
FUNCTION = "interact"
|
|
CATEGORY = "ali1234/debug"
|
|
OUTPUT_NODE = True
|
|
|
|
def interact(self, **kwargs):
|
|
if sys.__stdout__.isatty():
|
|
with RestoreStdStreams():
|
|
code.interact(
|
|
banner=textwrap.dedent(f"""
|
|
Interactive debugging started.
|
|
Try `print(a)`.
|
|
{MESSAGE}
|
|
"""),
|
|
exitmsg="Resuming workflow...",
|
|
local={
|
|
**kwargs,
|
|
'quit': Quitter(),
|
|
'exit': Quitter()
|
|
}
|
|
)
|
|
else:
|
|
# Don't block the server if there is no tty.
|
|
print("Skipping interactive prompt because there is no tty.")
|
|
return ()
|