Merge pull request #27 from tecknicaltom/noncrashes

fix exception in json if test cases don't crash
jfoote · May 14, 2015 · a5ec635 · a5ec635
2 parents 0444f07 + bc2f732
commit a5ec635
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/exploitable/lib/classifier.py b/exploitable/lib/classifier.py
@@ -102,6 +102,9 @@ def __add__(self, tag):
 
     # for python3
     def __lt__(self, other):
+        if other is None:
+            return False
+
         if not issubclass(type(other), type(self)):
             raise TypeError("cannot compare type {} to type {}".format(type(other), type(self)))
 
@@ -116,6 +119,9 @@ def __lt__(self, other):
         return False
 
     def __cmp__(self, other):
+        if other is None:
+            return 1
+
         if not issubclass(type(other), type(self)):
             raise TypeError("cannot compare type {} to type {}".format(type(other), type(self)))