added categories for vulnerabilities. From now on, all vulnerabilities should be classified with a category

src/core/events/types/common.py

@@ -42,12 +42,17 @@ class Service(object):
         return self.__doc__
 
 class Vulnerability(object):
-    def __init__(self, component, name):
+    def __init__(self, component, name, category=None):
         self.component = component
+        self.category = category
         self.name = name
         self.evidence = ""
         self.role = "Node"
 
+    def get_category(self):
+        if self.category:
+            return self.category.name
+
     def get_name(self):
         return self.name
 

src/core/types.py

@@ -18,4 +18,21 @@ class Kubelet(KubernetesCluster):
     name = "Kubelet"
 
 
+""" Categories """
+class InformationDisclosure(object):
+    name = "Information Disclosure"
+
+class RemoteCodeExec(object):
+    name = "Remote Code Execution"
+
+class IdentityTheft(object):
+    name = "Identity Theft"
+
+class UnauthenticatedAccess(object):
+    name = "Unauthenticated Access"
+
+class AccessRisk(object):
+    name = "Access Risk"
+
+
 from events import handler # import is in the bottom to break import loops