Add colors in cmd

RashikaKarki · May 28, 2021 · 3c71e8b · 3c71e8b
1 parent a2469b1
commit 3c71e8b
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 8 deletions.
diff --git a/src/Storm/__init__.py b/src/Storm/__init__.py
@@ -1,4 +1 @@
-from Storm.error import *
-from Storm.lexer import *
-from Storm.type import *
-from Storm.token import *
+from Storm.lexer import Lexer
diff --git a/src/Storm/lexer.py b/src/Storm/lexer.py
@@ -1,4 +1,3 @@
-from typing import Text
 from Storm import token
 
 

diff --git a/src/bgcolors.py b/src/bgcolors.py
@@ -0,0 +1,10 @@
+class Bcolors:
+    HEADER = '\033[95m'
+    OKBLUE = '\033[94m'
+    OKCYAN = '\033[96m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
diff --git a/src/main.py b/src/main.py
@@ -1,15 +1,16 @@
 from Storm import Lexer
+from bgcolors import Bcolors
 
 def run(text):
     lexer = Lexer(text)
     tokens, error = lexer.make_tokens()
     print(tokens)
     if error:
-        print("ERROR")
-        print(error)
+        print(f"{Bcolors.FAIL}ERROR{Bcolors.ENDC}")
+        print(f"{Bcolors.FAIL}{error}{Bcolors.ENDC}")
 
 
 while True:
-    text = input("storm > ")
+    text = input(f"{Bcolors.OKBLUE}strom > {Bcolors.ENDC} ")
     run(text)
Original file line number	Diff line number	Diff line change
		@@ -1,4 +1,3 @@
		from typing import Text
		from Storm import token


Expand Down