From fcb74782eaeb02d3de88d9b7c8ca7309f2004188 Mon Sep 17 00:00:00 2001
From: Jordan Thompson <jordanslost@gmail.com>
Date: Thu, 8 Jun 2023 12:36:59 -0700
Subject: [PATCH] Add optional token log output

---
 WAS_Node_Suite.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/WAS_Node_Suite.py b/WAS_Node_Suite.py
index ac34fb8..6e24504 100644
--- a/WAS_Node_Suite.py
+++ b/WAS_Node_Suite.py
@@ -8732,6 +8732,7 @@ class WAS_Text_Add_Tokens:
         return {
             "required": {
                 "tokens": ("STRING", {"default": "[hello]: world", "multiline": True}),
+                "print_current_tokens": (["false", "true"],),
             }
         }
 
@@ -8740,7 +8741,7 @@ class WAS_Text_Add_Tokens:
     OUTPUT_NODE = True
     CATEGORY = "WAS Suite/Text/Tokens"
 
-    def text_add_tokens(self, tokens):
+    def text_add_tokens(self, tokens, print_current_tokens="false"):
     
         import io
     
@@ -8755,8 +8756,9 @@ class WAS_Text_Add_Tokens:
             tk.addToken(token, token_value)
         
         # Current Tokens
-        cstr(f'Current Custom Tokens:').msg.print()
-        print(json.dumps(tk.custom_tokens, indent=4))
+        if print_current_tokens == "true":
+            cstr(f'Current Custom Tokens:').msg.print()
+            print(json.dumps(tk.custom_tokens, indent=4))
         
         return tokens
         
@@ -8778,6 +8780,7 @@ class WAS_Text_Add_Token_Input:
             "required": {
                 "token_name": (TEXT_TYPE, {"forceInput": (True if TEXT_TYPE == 'STRING' else False)}),
                 "token_value": (TEXT_TYPE, {"forceInput": (True if TEXT_TYPE == 'STRING' else False)}),
+                "print_current_tokens": (["false", "true"],),
             }
         }
 
@@ -8786,7 +8789,7 @@ class WAS_Text_Add_Token_Input:
     OUTPUT_NODE = True
     CATEGORY = "WAS Suite/Text/Tokens"
 
-    def text_add_token(self, token_name, token_value):
+    def text_add_token(self, token_name, token_value, print_current_tokens="false"):
 
         if token_name.strip() == '':
             cstr(f'A `token_name` is required for a token; token name provided is empty.').error.print()
@@ -8799,8 +8802,9 @@ class WAS_Text_Add_Token_Input:
         tk.addToken(token_name, token_value)
         
         # Current Tokens
-        print(f'\033[34mWAS Node Suite\033[0m Current Custom Tokens:')
-        print(json.dumps(tk.custom_tokens, indent=4))
+        if print_current_tokens == "true":
+            print(f'\033[34mWAS Node Suite\033[0m Current Custom Tokens:')
+            print(json.dumps(tk.custom_tokens, indent=4))
         
         return (token_name, token_value)