1 files changed, 51 insertions, 0 deletions
diff --git a/app/utils/api_access_base.py b/app/utils/api_access_base.py
new file mode 100644
index 0000000..31f50b4
--- /dev/null
+++ b/app/utils/api_access_base.py
@@ -0,0 +1,51 @@
+###############################################################################
+# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems)   #
+# and others                                                                  #
+#                                                                             #
+# All rights reserved. This program and the accompanying materials            #
+# are made available under the terms of the Apache License, Version 2.0       #
+# which accompanies this distribution, and is available at                    #
+# http://www.apache.org/licenses/LICENSE-2.0                                  #
+###############################################################################
+import requests
+
+from discover.configuration import Configuration
+from discover.fetcher import Fetcher
+
+
+class ApiAccessBase(Fetcher):
+
+    CONNECT_TIMEOUT = 5
+
+    def __init__(self, api_name=None, config=None):
+        super().__init__()
+        if api_name is None:
+            raise ValueError('ApiAccessBase: api_name must be defined')
+        self.config = {api_name: config} if config else Configuration()
+        self.api_config = self.config.get(api_name)
+        if self.api_config is None:
+            raise ValueError('ApiAccessBase: section "{}" missing in config'
+                             .format(api_name))
+        self.host = self.api_config.get('host', '')
+        self.port = self.api_config.get('port', '')
+        if not (self.host and self.port):
+            raise ValueError('Missing definition of host or port ' +
+                             'for {} API access'
+                             .format(api_name))
+
+    def get_rel_url(self, relative_url, headers):
+        req_url = self.base_url + relative_url
+        return self.get_url(req_url, headers)
+
+    def get_url(self, req_url, headers):
+        response = requests.get(req_url, headers=headers)
+        if response.status_code != requests.codes.ok:
+            # some error happened
+            if 'reason' in response:
+                msg = ', reason: {}'.format(response.reason)
+            else:
+                msg = ', response: {}'.format(response.text)
+            self.log.error('req_url: {} {}'.format(req_url, msg))
+            return None
+        ret = response.json()
+        return ret