Read config from a file

Author: Inbal Tako

requirements.txt

@@ -1,2 +1,2 @@
-requests~=2.22.0
+requests~=2.23.0
 pycrypto~=2.6.1

securenative/config/__init__.py

@@ -0,0 +1,61 @@
+from securenative.config.securenative_options import SecurenativeOptions
+from securenative.enums.failover_strategy import FailOverStrategy
+
+
+class ConfigurationBuilder(object):
+
+    def __init__(self):
+        self.api_key = ""
+        self.api_url = "https://api.securenative.com/collector/api/v1"
+        self.interval = 1000
+        self.max_events = 1000
+        self.timeout = 1500
+        self.auto_send = True
+        self.disable = False
+        self.log_level = "CRITICAL"
+        self.fail_over_strategy = FailOverStrategy.FAIL_OPEN
+
+    @staticmethod
+    def default_config_builder():
+        return ConfigurationBuilder()
+
+    def with_api_key(self, api_key):
+        self.api_key = api_key
+        return self
+
+    def with_api_url(self, api_url):
+        self.api_url = api_url
+        return self
+
+    def with_interval(self, interval):
+        self.interval = interval
+        return self
+
+    def with_max_events(self, max_events):
+        self.max_events = max_events
+        return self
+
+    def with_timeout(self, timeout):
+        self.timeout = timeout
+        return self
+
+    def with_auto_send(self, auto_send):
+        self.auto_send = auto_send
+        return self
+
+    def with_disable(self, disable):
+        self.disable = disable
+        return self
+
+    def with_log_level(self, log_level):
+        self.log_level = log_level
+        return self
+
+    def with_fail_over_strategy(self, fail_over_strategy):
+        self.fail_over_strategy = fail_over_strategy
+        return self
+
+    @staticmethod
+    def get_default_securenative_options():
+        return SecurenativeOptions("", "https://api.securenative.com/collector/api/v1", 1000,
+                                   1000, 1500, True, False, "CRITICAL", FailOverStrategy.FAIL_OPEN)

securenative/config/configuration_builder.py

@@ -0,0 +1,57 @@
+import os
+from configparser import ConfigParser, NoSectionError
+
+from securenative.config.configuration_builder import ConfigurationBuilder
+
+
+class ConfigurationManager(object):
+    DEFAULT_CONFIG_FILE = "securenative.ini"
+    CUSTOM_CONFIG_FILE_ENV_NAME = "SECURENATIVE_COMFIG_FILE"
+    config = ConfigParser.ConfigParser()
+
+    @staticmethod
+    def read_resource_file(cls, resource_path):
+        cls.config.read(resource_path)
+        sections = cls.config.sections()
+
+        properties = {}
+        for section in sections:
+            options = cls.config.options(section)
+            for option in options:
+                try:
+                    properties[option] = cls.config.get(section, option)
+                except NoSectionError:
+                    properties[option] = None
+
+        return properties
+
+    @staticmethod
+    def _get_resource_path(cls, env_name):
+        env_value = os.environ.get(env_name)
+
+        if env_value:
+            return env_value
+
+        return os.environ.get(cls.DEFAULT_CONFIG_FILE)
+
+    @staticmethod
+    def config_builder():
+        return ConfigurationBuilder.default_config_builder()
+
+    @staticmethod
+    def load_config(cls):
+        builder = ConfigurationBuilder()
+        options = builder.get_default_securenative_options()
+        resource_path = cls._get_env_or_default(cls.DEFAULT_CONFIG_FILE, cls.CUSTOM_CONFIG_FILE_ENV_NAME)
+        properties = cls.read_resource_file(resource_path)
+
+        builder.\
+            with_api_key(cls._get_env_or_default(properties, "SECURENATIVE_API_KEY", options.api)).\
+            with_api_url(cls._get_env_or_default(properties, "SECURENATIVE_API_URL", options.api_url)).\
+            with_interval(cls._get_env_or_default(properties, "SECURENATIVE_INTERVAL", options.interval)).\
+            with_max_events(cls._get_env_or_default(properties, "SECURENATIVE_MAX_EVENTS", options.max_events)).\
+            with_timeout(cls._get_env_or_default(properties, "SECURENATIVE_TIMEOUT", options.timeout)).\
+            with_auto_send(cls._get_env_or_default(properties, "SECURENATIVE_AUTO_SEND", options.auto_send)).\
+            with_disable(cls._get_env_or_default(properties, "SECURENATIVE_DISABLE", options.disable)).\
+            with_log_level(cls._get_env_or_default(properties, "SECURENATIVE_LOG_LEVEL", options.log_level)).\
+            with_fail_over_strategy(cls._get_env_or_default(properties, "SECURENATIVE_FAILOVER_STRATEGY", options))

securenative/config/configuration_manager.py

@@ -0,0 +1,13 @@
+class SecurenativeOptions(object):
+
+    def __init__(self, api_key, api_url, interval, max_events, timeout, auto_send,
+                 disable, log_level, fail_over_strategy):
+        self.api_key = api_key
+        self.api_url = api_url
+        self.interval = interval
+        self.max_events = max_events
+        self.timeout = timeout
+        self.auto_send = auto_send
+        self.disable = disable
+        self.log_level = log_level
+        self.fail_over_strategy = fail_over_strategy

securenative/config/securenative_options.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+
+class FailOverStrategy(Enum):
+    FAIL_OPEN = "fail-open"
+    FAIL_CLOSED = "fail-closed"

securenative/context/__init__.py

@@ -1,7 +1,7 @@
 import uuid
 import time
 
-from securenative.utils import _parse_cookie
+from securenative.utils.utils import _parse_cookie
 
 
 class User:
@@ -30,7 +30,7 @@ def __init__(self, event_type, user=User(), ip=u'127.0.0.1', remote_ip=u'127.0.0
             self.cid, self.fp = _parse_cookie(sn_cookie_value)
 
         self.vid = str(uuid.uuid4())
-        self.ts = int(time.time())*1000
+        self.ts = int(time.time()) * 1000
 
     def as_dict(self):
         return {