Colored error outputs+Adding restore argument

Used colorama for colored outputs for different platforms (Windows, Linux, OSX).
Resotre arguement added to prevent assigning None to the current object (if necessary).

Author: omdxp

cmpx/Complex.py

@@ -4,14 +4,24 @@
 
 @author: Omar Belghaouti
 """
-from termcolor import colored
+import os
+from colorama import init, Fore, Style
 from math import sqrt
+
+# Check the os to make output colorful in any platform
+if os.name == 'nt':
+    init(convert=True)
+elif os.name == 'linux' or 'linux2' or 'darwin':
+    init(convert=False)
+    
 # Complex class for complex number manipulations
 class Complex():
     # Constructor
-    # re : is the real part of complex number
-    # im : is the imaginary part of complex number
-    # restore : whenever an error occurs on any operation of a complex number, the last instance will be restored (By default it's true)
+    """
+    re : is the real part of complex number
+    im : is the imaginary part of complex number
+    restore : whenever an error occurs on any operation of a complex number, the last instance will be restored (By default it's true)
+    """
     def __init__(self, re=0, im=0, restore=True):
         self.re = re
         self.im = im
@@ -25,7 +35,7 @@ def __add__(self, other):
                 other = Complex(other)
             return Complex(self.re + other.re, self.im + other.im)
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 2 : -
     def __sub__(self, other):
         try:
@@ -35,7 +45,7 @@ def __sub__(self, other):
                 other = Complex(other)
             return Complex(self.re - other.re, self.im - other.im)
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 3 : *
     def __mul__(self, other):
         try:
@@ -45,7 +55,7 @@ def __mul__(self, other):
                 other = Complex(other)
             return Complex(self.re * other.re - self.im * other.im, self.re * other.im + self.im * other.re)
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 4 : /
     def __truediv__(self, other):
         try:
@@ -56,12 +66,13 @@ def __truediv__(self, other):
             den = other * other.con()
             num = self * other.con()
             if den.re == 0 and self.restore:
-                print(colored('Float division by zero', 'red'))
-                print(colored('Restoring last number', 'green'))
+                print(Fore.RED + 'Float division by zero')
+                print(Fore.GREEN + 'Restoring last number')
+                print(Style.RESET_ALL, end='')
                 return Complex(self.re, self.im)
             return Complex(num.re / den.re, num.im / den.re)
         except (ZeroDivisionError, ValueError) as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 5 : //
     def __floordiv__(self, other):
         try:
@@ -72,12 +83,13 @@ def __floordiv__(self, other):
             den = other * other.con()
             num = self * other.con()
             if den.re == 0 and self.restore:
-                print(colored('Float division by zero', 'red'))
-                print(colored('Restoring last number', 'green'))
+                print(Fore.RED + 'Float division by zero')
+                print(Fore.GREEN + 'Restoring last number')
+                print(Style.RESET_ALL, end='')
                 return Complex(self.re, self.im)
             return Complex(num.re // den.re, num.im // den.re)
         except (ZeroDivisionError, ValueError) as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     def __gt__(self, other):
         try:
             if other is None:
@@ -86,7 +98,7 @@ def __gt__(self, other):
                 other = Complex(other)
             return self.mod() > other.mod()
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 7 : >=
     def __ge__(self, other):
         try:
@@ -96,7 +108,7 @@ def __ge__(self, other):
                 other = Complex(other)
             return self.mod() >= other.mod()
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 8: <
     def __lt__(self, other):
         try:
@@ -106,7 +118,7 @@ def __lt__(self, other):
                 other = Complex(other)
             return not self >= other
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 9: <=
     def __le__(self, other):
         try:
@@ -116,7 +128,7 @@ def __le__(self, other):
                 other = Complex(other)
             return not self > other
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 10: ==
     def __eq__(self, other):
         try:
@@ -125,8 +137,8 @@ def __eq__(self, other):
             if not isinstance(other, Complex):
                 other = Complex(other)
             return (self.re == other.re) and (self.im == other.im)
-        except ValueError:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+        except ValueError as err:
+            self.print_err(err)
     # Operator overloading 11: !=
     def __ne__(self, other):
         try:
@@ -136,7 +148,7 @@ def __ne__(self, other):
                 other = Complex(other)
             return not self == other
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 12: +=
     def __iadd__(self, other):
         try:
@@ -148,7 +160,7 @@ def __iadd__(self, other):
             self.im += other.im
             return Complex(self.re, self.im)
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 13: -=
     def __isub__(self, other):
         try:
@@ -160,7 +172,7 @@ def __isub__(self, other):
             self.im -= other.im
             return Complex(self.re, self.im)
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 14: *=
     def __imul__(self, other):
         try:
@@ -172,7 +184,7 @@ def __imul__(self, other):
             self.im = self.re * other.im + self.im * other.re
             return Complex(self.re, self.im)
         except ValueError as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 15: /=
     def __idiv__(self, other):
         try:
@@ -183,14 +195,15 @@ def __idiv__(self, other):
             den = other * other.con()
             num = self * other.con()
             if den.re == 0 and self.restore:
-                print(colored('Float division by zero', 'red'))
-                print(colored('Restoring last number', 'green'))
+                print(Fore.RED + 'Float division by zero')
+                print(Fore.GREEN + 'Restoring last number')
+                print(Style.RESET_ALL, end='')
                 return Complex(self.re, self.im)
             self.re = num.re / den.re
             self.im = num.im / den.re
-            return Complex(re, im)
+            return Complex(self.re, self.im)
         except (ZeroDivisionError, ValueError) as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     # Operator overloading 16: //=
     def __ifloordiv__(self, other):
         try:
@@ -201,23 +214,28 @@ def __ifloordiv__(self, other):
             den = other * other.con()
             num = self * other.con()
             if den.re == 0 and self.restore:
-                print(colored('Float division by zero', 'red'))
-                print(colored('Restoring last number', 'green'))
+                print(Fore.RED + 'Float division by zero')
+                print(Fore.GREEN + 'Restoring last number')
+                print(Style.RESET_ALL, end='')
                 return Complex(self.re, self.im)
             self.re = num.re // den.re
             self.im = num.im // den.re
             return Complex(self.re, self.im)
         except (ZeroDivisionError, ValueError) as err:
-            print(colored('{}: {}'.format(err.__class__.__name__, err), 'red'))
+            self.print_err(err)
     ## Helper functions
     # Module function to calculate the modulus of a complex number
     def mod(self):
         return sqrt(self.re**2 + self.im**2)
     # Conjugated of a complex number
     def con(self):
         return Complex(self.re, - self.im)
-    # String function
-    def __str__(self):
+    # function to print the error message
+    def print_err(self, err):
+        print(Fore.RED + '{}: {}'.format(err.__class__.__name__, err))
+        print(Style.RESET_ALL, end='')
+    # Representation function for representing a complex number
+    def __repr__(self):
         output = str(self.re) + ' + ' + str(self.im) + 'j' if(self.im != 0) else str(self.re)
         output = str(self.re) + ' - ' + str(-self.im) + 'j' if (self.im < 0) else output
         return output

setup.py

@@ -5,7 +5,7 @@
 
 setup_args = dict(
     name='cmpx',
-    version='0.2',
+    version='0.3',
     description='Complex class for different operations on complex numbers',
     long_description_content_type='text/markdown',
     long_description=README,