From c12e5c1d25638e656dd2dfb570018ce5856123df Mon Sep 17 00:00:00 2001
From: Lasse Edfast <8794658+lasseedfast@users.noreply.github.com>
Date: Wed, 26 Jun 2024 12:34:38 +0200
Subject: [PATCH] Create print_color.py

---
 print_color.py | 65 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 print_color.py

diff --git a/print_color.py b/print_color.py
new file mode 100644
index 0000000..2b72f84
--- /dev/null
+++ b/print_color.py
@@ -0,0 +1,65 @@
+
+def print_green(*args):
+    text = ""
+    for arg in args:
+        text += str(arg) + " "
+    print(f"\033[92m{text}\033[0m")
+
+
+def print_red(*args):
+    text = ""
+    for arg in args:
+        text += str(arg) + " "
+    print(f"\033[91m{text}\033[0m")
+
+
+def print_yellow(*args):
+    text = ""
+    for arg in args:
+        text += str(arg) + " "
+    print(f"\033[93m{text}\033[0m")
+
+
+def print_blue(*args):
+    text = ""
+    for arg in args:
+        text += str(arg) + " "
+    print(f"\033[94m{text}\033[0m")
+
+
+def print_purple(*args):
+    text = ""
+    for arg in args:
+        text += str(arg) + " "
+    print(f"\033[95m{text}\033[0m")
+
+
+def choose_color(last_color_index):
+    colors = {
+        "blue": "\033[94m",
+        "green": "\033[92m",
+        "yellow": "\033[93m",
+        "red": "\033[91m",
+        "purple": "\033[95m",
+    }
+    color_keys = list(colors.keys())
+    color_index = (last_color_index + 1) % len(color_keys)
+    color = color_keys[color_index]
+    return colors[color], color_index
+
+def print_rainbow(*args):
+    color_index = -1
+    text = ""
+    for arg in args:
+        if isinstance(arg, list):
+            for i in arg:
+                color_code, color_index = choose_color(color_index)
+                text += f"{color_code}{i}\033[0m "
+        elif isinstance(arg, dict):
+            for k, v in arg.items():
+                color_code, color_index = choose_color(color_index)
+                text += f"{color_code}{k}: {v}\033[0m "
+        else:
+            color_code, color_index = choose_color(color_index)
+            text += f"{color_code}{arg}\033[0m "
+    print(text)