break out threat event

rail/__init__.py

@@ -2,5 +2,6 @@
 
 from .tree import Tree
 from .threat_source import ThreatSource, ThreatSources
-from .rail import ThreatSource, ThreatSources, ThreatEvent, ThreatEvents, Control, Controls, Vulnerability, Vulnerabilities, Likelihood, Impact, Risk, Risks
+from .threat_event import ThreatEvent, ThreatEvents
+from .rail import Control, Controls, Vulnerability, Vulnerabilities, Likelihood, Impact, Risk, Risks
 

rail/rail.py

@@ -8,25 +8,13 @@
 
 from .tree import Tree
 from .threat_source import ThreatSource, ThreatSources
+from .threat_event import ThreatEvent, ThreatEvents
 
 pd.set_option('display.float_format', lambda x: '%.2f' % x)
 pd.set_option('display.max_colwidth', -1)
 plt.style.use('seaborn-poster')
 
 
-class ThreatEvent(UserDict):
-    def __init__(self, name: str, threat_source: ThreatSource) -> None:
-        self.data = {}
-        self.data['name'] = name
-        self.data['threat_source'] = threat_source
-
-
-class ThreatEvents(UserDict):
-    def new(self, name: str, threat_source: ThreatSource) -> ThreatEvent:
-        self.data[name] = ThreatEvent(name, threat_source)
-        return self.data[name]
-
-
 class Control(UserDict):
     def __init__(self, name: str, cost: float, reduction: float, implemented: bool=True) -> None:
         self.data = {}

rail/threat_event.py

@@ -0,0 +1,28 @@
+"""
+A class to represent Threat Events
+"""
+from collections import UserDict
+
+from .threat_source import ThreatSource
+
+class ThreatEvent(UserDict): # pylint: disable=too-many-ancestors
+    """
+    A class to represent Threat Events
+    """
+    def __init__(self, name: str, threat_source: ThreatSource) -> None:
+        UserDict.__init__(self)
+        self.data = {}
+        self.data['name'] = name
+        self.data['threat_source'] = threat_source
+
+
+class ThreatEvents(UserDict): # pylint: disable=too-many-ancestors
+    """
+    A class to hold multiple Threat Events
+    """
+    def new(self, name: str, threat_source: ThreatSource) -> ThreatEvent:
+        """
+        A method to add a new threat event to the Threat Events class
+        """
+        self.data[name] = ThreatEvent(name, threat_source)
+        return self.data[name]

rail/threat_source.py

@@ -1,11 +1,11 @@
 """
-A class to implement Threat Sources
+A class to represent Threat Sources
 """
 from collections import UserDict
 
 class ThreatSource(UserDict): # pylint: disable=too-many-ancestors
     """
-    A class to implement Threat Sources
+    A class to represent Threat Sources
     """
     def __init__(self, name: str) -> None:
         UserDict.__init__(self)