Change delta to solve+Change solve to equations

Author: omdxp

cmpx/Complex.py

@@ -4,15 +4,8 @@
 
 @author: Omar Belghaouti
 """
-import os
-from colorama import init, Fore, Style
+from cmpx import print_err
 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():
@@ -233,10 +226,6 @@ def mod(self):
     # Conjugated of a complex number
     def con(self):
         return Complex(self.re, - self.im, self.restore)
-    # 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):
         if(self.re != 0 and self.im > 0):

cmpx/__init__.py

@@ -1,10 +1,11 @@
 from .Complex import Complex
+from .error import print_err
 
 __title__      = 'PythonComplex'
 __author__     = 'Omar Belghaouti'
 __maintainer__ = 'Omar Belghaouti'
 __email__      = 'bel_omar18@yahoo.com'
-__version__    = '0.7.2'
+__version__    = '0.7.5'
 __license__    = 'MIT'
 __all__ = [
     'Complex'

cmpx/equations/__init__.py

@@ -0,0 +1,2 @@
+# -*- coding: utf-8 -*-
+from .solve import solve

cmpx/equations/solve.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+from cmpx import Complex
+from cmpx import print_err
+from math import sqrt
+
+# Function to solve a second degree (or linear) equation with passing arguments seperatly
+def solve(a, b, c=0):
+    try:
+        if a and b == 0:
+            raise ValueError('At least give a number to the first two coefficents')
+        if c != 0:
+            delta = b**2 - 4 * a * c
+            if(delta < 0):
+                solution = (Complex(-b/2*a, -sqrt(-delta)/2*a),
+                            Complex(-b/2*a, sqrt(-delta)/2*a))
+            elif(delta == 0):
+                solution = (Complex(-b/2*a))
+            else:
+                solution = (Complex((-b - sqrt(delta))/2*a),
+                            Complex((-b + sqrt(delta))/2*a))
+            return solution
+        else:
+            return (Complex(-b/a))
+    except (ZeroDivisionError, ValueError) as err:
+        print_err(err)

cmpx/error.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Sun Sep 15 12:04:55 2019
+
+@author: Cds
+"""
+import os
+from colorama import init, Fore, Style
+
+# 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)
+    
+# function to print the error message
+def print_err(err):
+    print(Fore.RED + '{}: {}'.format(err.__class__.__name__, err))
+    print(Style.RESET_ALL, end='')

cmpx/solve/__init__.py

@@ -5,7 +5,7 @@
 
 setup_args = dict(
     name='cmpx',
-    version='0.7.2',
+    version='0.7.5',
     description='A package for different operations on complex numbers',
     long_description_content_type='text/markdown',
     long_description=README,