Add TimeUnit

Author: DUYN

investing_bot_framework/core/utils.py

@@ -1,6 +1,9 @@
 import sys
 from threading import Thread
 from wrapt import synchronized
+from enum import Enum
+
+from investing_bot_framework.core.exceptions import OperationalException
 
 
 class Singleton(type):
@@ -52,4 +55,52 @@ def localtrace(self, frame, event, arg):
         return self.localtrace
 
     def kill(self):
-        self.killed = True
+        self.killed = True
+
+
+class TimeUnit(Enum):
+    """
+    Class TimeUnit: Enum for TimeUnit
+    """
+
+    SECOND = 'SEC',
+    MINUTE = 'MIN',
+    HOUR = 'HR',
+    ALWAYS = 'ALWAYS'
+
+    # Static factory method to convert a string to time_unit
+    @staticmethod
+    def from_string(value: str):
+
+        if isinstance(value, str):
+
+            if value.lower() in ('sec', 'second', 'seconds'):
+                return TimeUnit.SECOND
+
+            elif value.lower() in ('min', 'minute', 'minutes'):
+                return TimeUnit.MINUTE
+
+            elif value.lower() in ('hr', 'hour', 'hours'):
+                return TimeUnit.HOUR
+
+            elif value.lower() in ('always', 'every', 'continuous', 'every_time'):
+                return TimeUnit.ALWAYS
+            else:
+                raise OperationalException('Could not convert value {} to a time_unit'.format(value))
+
+        else:
+            raise OperationalException("Could not convert non string value to a time_unit")
+
+    def equals(self, other):
+
+        if isinstance(other, Enum):
+            return self.value == other.value
+        else:
+
+            try:
+                time_unit = TimeUnit.from_string(other)
+                return time_unit == self
+            except OperationalException:
+                pass
+
+            return other == self.value