added asa module. fixes #44. Version Bump 1.0!!!

varken/cisco.py

@@ -0,0 +1,62 @@
+import logging
+from requests import Session, Request
+from datetime import datetime, timezone
+
+from varken.helpers import connection_handler
+
+
+class CiscoAPI(object):
+    def __init__(self, firewall, dbmanager):
+        self.now = datetime.now(timezone.utc).astimezone().isoformat()
+        self.dbmanager = dbmanager
+        self.firewall = firewall
+        # Create session to reduce server web thread load, and globally define pageSize for all requests
+        self.session = Session()
+        self.session.auth = (self.firewall.username, self.firewall.password)
+        self.logger = logging.getLogger()
+
+        self.get_token()
+
+    def __repr__(self):
+        return "<ciscoasa-{}>".format(self.firewall.id)
+
+    def get_token(self):
+        endpoint = '/api/tokenservices'
+
+        req = self.session.prepare_request(Request('POST', self.firewall.url + endpoint))
+        post = connection_handler(self.session, req, self.firewall.verify_ssl)
+
+        if not post:
+            return
+
+        self.session.headers = {'X-Auth-Token': post}
+
+    def get_bandwidth(self):
+        self.now = datetime.now(timezone.utc).astimezone().isoformat()
+        endpoint = '/api/monitoring/device/interfaces/' + self.firewall.outside_interface
+
+        if not self.session.headers:
+            return
+
+        req = self.session.prepare_request(Request('GET', self.firewall.url + endpoint))
+        print(req.headers)
+        get = connection_handler(self.session, req, self.firewall.verify_ssl)
+
+        if not get:
+            return
+
+        influx_payload = [
+            {
+                "measurement": "Cisco ASA",
+                "tags": {
+                    "interface": self.firewall.outside_interface
+                },
+                "time": self.now,
+                "fields": {
+                    "upload_bitrate": get['outputBitRate'],
+                    "download_bitrate": get['inputBitRate']
+                }
+            }
+        ]
+
+        self.dbmanager.write_points(influx_payload)