HttpTransport first draft

3 files changed, 81 insertions, 0 deletions

diff --git a/automato/command_line.py b/automato/command_line.py
index b8793a8..1b1dc5e 100755
--- a/automato/command_line.py
+++ b/automato/command_line.py
@@ -17,6 +17,7 @@ def setup():
                         datefmt='%c')
 
     logging.getLogger('paramiko').setLevel(logging.WARNING)
+    logging.getLogger('urllib3').setLevel(logging.WARNING)
 
     endpoint_config = load_yaml('endpoints.yml')
     trigger_config = load_yaml('triggers.yml')
diff --git a/automato/transport/http.py b/automato/transport/http.py
new file mode 100644
index 0000000..ca27e82
--- /dev/null
+++ b/automato/transport/http.py
@@ -0,0 +1,79 @@
+import logging
+import requests
+
+from automato.transport import Transport, THROWAWAY
+
+logger = logging.getLogger(__name__)
+
+'''
+HttpTransport
+
+REQUIRED ARGUMENTS
+    address: Base Address of endpoint
+OPTIONAL ARGUMENTS
+    user: Username for HTTP basic auth.
+    password: Password for HTTP basic auth
+    headers: Dict for headers to add in every connection
+    user_agent: Change the user agent form the default requests
+'''
+class HttpTransport(Transport):
+    CONNECTION = THROWAWAY
+
+    def _init(self, address:str,
+              user:[None,str] = None, password:[None,str] = None,
+              headers:[None,dict] = None, user_agent:[None,str] = None,
+              validation_path: str = '/', timeout:int = 30):
+        self._address = address
+        self._user = user
+        self._password = password
+        self._headers = headers
+        self._user_agent = user_agent
+        self._validation_path = '/'
+        self._timeout = timeout
+
+    def check(self):
+        # TODO Here we could maybe also perform a more complex login
+        # with Cookies?
+
+        ret = self._request('HEAD', self._validation_path)
+        logger.debug(f'{self._validation_path} checked {ret.ok} with code {ret.status_code}')
+
+        return ret.ok
+
+
+    def _request(self, method:str, path:str, headers:[None,dict] = None,
+                 data = None, params:[None,str] = None):
+        full_path = self._address.rstrip('/') + '/' + path.lstrip('/')
+        logger.debug(f'requested {method} for {full_path}')
+
+        all_headers = headers
+
+        if self._headers is not None:
+            if headers is not None:
+                all_headers = self._headers | headers
+            else:
+                all_headers = self._headers
+
+        # TODO maybe pass **kwargs here?
+        try:
+            req = requests.request(method, full_path, headers=all_headers,
+                                   data=data, params=params, timeout=self._timeout)
+            return req
+        except requests.RequestException as e:
+            logger.error(f'An exception occured in {method} for {full_path}: {e}')
+            #FIXME We need better error handling
+            return None
+
+
+    def get(self, **kwargs):
+        return self._request('GET', **kwargs)
+    def head(self, **kwargs):
+        return self._request('HEAD', **kwargs)
+    def put(self, **kwargs):
+        return self._request('PUT', **kwargs)
+    def post(self, **kwargs):
+        return self._request('POST', **kwargs)
+    def patch(self, **kwargs):
+        return self._request('PATCH', **kwargs)
+    def delete(self, **kwargs):
+        return self._request('DELETE', **kwargs)
diff --git a/setup.py b/setup.py
index 2607263..b0d60c5 100644
--- a/setup.py
+++ b/setup.py
@@ -15,6 +15,7 @@ setup(
         "paramiko",
         "pyparsing",
         "PyYAML",
+        "requests",
     ],
     long_description=open('README.md').read(),
     long_description_content_type="text/markdown",