recursive_format feature

CHANGELOG.md

@@ -1,7 +1,12 @@
-# `TODO`
-- Recursive formatting.
+## `TODO`
+
 - Multi-item nodes.
 
+# v1.0.1
+
+- ❗ Recursive formatting.
+- Tiny tweaks to tooltips/descriptions.
+
 # v1.0.0
 
 The minimally-complete version.

funcs_formatter.py

@@ -5,17 +5,55 @@ The actual code (MVC pattern) for `StringConstructorFormatter` node.
 
 import typing as _t
 
-from .enums import T as _T
+import sys as _sys
+
 from .funcs_common import _show_text_on_node, _verify_input_dict
 
 
-def formatter(template: str, format_dict: _t.Dict[str, _T] = None, show: bool = True, unique_id: str = None) -> _t.Tuple[str]:
+_RECURSION_LIMIT = max(int(_sys.getrecursionlimit()), 10)  # You can externally monkey-patch it... but if it blows up, your fault 🤷🏻‍♂️
+
+
+def _recursive_format(template: str, format_dict: _t.Dict[str, _t.Any], show: bool = True, unique_id: str = None) -> str:
+	"""
+	It's not actually recursive - because, you know, any recursion could be turned into iteration,
+	and good boys do that. 😊
+	"""
+	assert isinstance(_RECURSION_LIMIT, int) and _RECURSION_LIMIT > 0
+
+	prev: str = ''
+	new: str = template
+	for i in range(_RECURSION_LIMIT):
+		if prev == new:
+			break
+		prev = new
+		new = new.format_map(format_dict)
+	if prev == new:
+		return new
+
+	msg = (
+		f"Recursion limit ({_RECURSION_LIMIT}) reached on attempt to format a string: {template!r}\n"
+		f"Last two formatting attempts:\n{prev!r}\n{new!r}"
+	)
+	if show and unique_id:
+		_show_text_on_node(msg, unique_id)
+	raise RecursionError(msg)
+	# noinspection PyUnreachableCode
+	return ''  # just to be safe
+
+
+def formatter(
+	template: str, recursive: bool = False, format_dict: _t.Dict[str, _t.Any] = None,
+	show: bool = True, unique_id: str = None
+) -> _t.Tuple[str]:
 	format_dict = _verify_input_dict(format_dict)
 	if not isinstance(template, str):
 		raise TypeError(f"Not a string: {template!r}")
-	out_text = template.format_map(format_dict)
 
+	out_text = (
+		_recursive_format(template, format_dict, show, unique_id)
+		if recursive
+		else template.format_map(format_dict)
+	)
 	if show and unique_id:
 		_show_text_on_node(out_text, unique_id)
-
 	return (out_text, )

node_formatter.py

@@ -25,6 +25,15 @@ _input_types = _deepfreeze({
 			"score_9, score_8_up, score_7_up, {char1_short}, standing next to {char2_short},\n"
 			"{char1_long}\n{char2_long}"
 		)}),
+		'recursive_format': (_IO.BOOLEAN, {'default': False, 'label_on': '❗ yes', 'label_off': 'no', 'tooltip': (
+			"Do recursive format - i.e., allow the chunks from the dictionary to reference other chunks - which in turn "
+			"lets you do really crazy stuff like building entire HIERARCHIES of sub-prompts, all linked to the same "
+			"wording typed in one place.\n\n"
+			"The exclamation mark reminds you that with great power comes great responsibility!\n"
+			"You can end up with chunks cross-referencing each other in a loop. In such case, the node will just error out "
+			"and won't let you crash the entire ComfyUI - but still, it's your responsibility to prevent "
+			"such looping dictionaries."
+		)}),
 		'show_status': (_IO.BOOLEAN, {'default': True, 'label_on': 'formatted string', 'label_off': 'no', 'tooltip': (
 			"Show the final string constructed from the text-template and format-dictionary?"
 		)}),
@@ -61,8 +70,8 @@ class StringConstructorFormatter:
 		return _input_types
 
 	def main(
-		self, template: str, show_status: bool,
+		self, template: str, recursive_format: bool, show_status: bool,
 		dict: _t.Dict[str, _t.Any] = None,
 		unique_id: str = None,
 	):
-		return _formatter(template, dict, show=show_status, unique_id=unique_id)
+		return _formatter(template, recursive_format, dict, show=show_status, unique_id=unique_id)