Add exception classes

Author: joszx

gear/gear_calc.py

@@ -1,6 +1,8 @@
 import gear.substat as substat
 import math
 
+from myexception import GearLevelException
+
 # Constants
 MIN = 0
 MAX = 1
@@ -205,4 +207,4 @@ def get_gear_level(gear_level):
     elif gear_level == '90':
         return LVL90
     else:
-        print('Error gear level not supported')
+        raise GearLevelException("Error gear level not supported")

main.py

@@ -1,7 +1,7 @@
 import cv2 as cv
 import time
 import pytesseract
-from myexception import MyException
+from myexception import GearParseException, GearLevelException
 import screengrabber
 import parser
 from edgedetector import EdgeDetector
@@ -145,9 +145,12 @@ def __init__(self):
                     print(curr_gear)
                     out = curr_gear
 
-                except MyException as e:
+                except GearParseException as e:
                     details = e.args[0]
                     out = "Error parsing gear\n" + details
+                except GearLevelException as e:
+                    details = e.args[0]
+                    out = details
                 except:
                     out = "Error initialising gear"
 

myexception.py

@@ -1,4 +1,8 @@
 class MyException(Exception):
     pass
 
+class GearParseException(Exception):
+    pass
 
+class GearLevelException(Exception):
+    pass

parser.py

@@ -1,5 +1,5 @@
 import gear.substat as substat
-from myexception import MyException
+from myexception import GearParseException, MyException
 
 # trim front and back spaces, removes empty strings, parses to list format
 def str_to_list(string):
@@ -21,7 +21,7 @@ def parse_substats(substat_name_list, substat_value_list):
     remove_substat_modification(substat_name_list)
 
     if len(substat_name_list) != len(substat_value_list):
-        raise MyException("number of substat names =/= number of substat values")
+        raise GearParseException("number of substat names =/= number of substat values")
 
     for i in range(len(substat_name_list)):
         curr_substat_name = substat_name_list[i]