[WIP] basic device add

kentikapi/v5/README.md

@@ -1,6 +1,14 @@
 Kentik Python API
 =================
 
+Site API
+----------------
+ - fetch
+
+Device API
+----------------
+ - create
+
 HyperTagging API
 ----------------
 

kentikapi/v5/device.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+
+from __future__ import print_function
+from builtins import str
+from builtins import object
+import json
+import requests
+
+
+"""Device API client"""
+
+
+class Device(object):
+    """Creates Kentik Device"""
+
+    def __init__(self, api_email, api_token, base_url="https://api.kentik.com"):
+        self.api_email = api_email
+        self.api_token = api_token
+        self.base_url = base_url
+
+    def create(
+        self,
+        host,
+        host_ip,
+        site_id,
+        plan_id,
+        community="public",
+        url="https://api.kentik.com/api/v5/device",
+    ):
+        headers = {
+            "User-Agent": "kentik-python-api/0.2",
+            "Content-Type": "application/json",
+            "X-CH-Auth-Email": self.api_email,
+            "X-CH-Auth-API-Token": self.api_token,
+        }
+        data = {
+            "device": {
+                "device_name": host.replace(".", "_"),
+                "device_type": "router",
+                "device_description": host,
+                "sending_ips": [host_ip],
+                "device_sample_rate": 4096,
+                "plan_id": plan_id,
+                "site_id": site_id,
+                "minimize_snmp": False,
+                "device_snmp_ip": host_ip,
+                "device_snmp_community": community,
+                "device_bgp_type": "none",
+            }
+        }
+        resp = requests.post(url, headers=headers, json=data)
+
+        # break out at first sign of trouble
+        resp.raise_for_status()
+        device = resp.json()
+
+        return device

kentikapi/v5/plan.py

@@ -0,0 +1,37 @@
+#!/usr/bin/env python
+
+from __future__ import print_function
+from builtins import str
+from builtins import object
+import json
+import requests
+
+
+"""Plan API client"""
+
+
+class Plan(object):
+    """Returns all Kentik Plans"""
+
+    def __init__(self, api_email, api_token, base_url="https://api.kentik.com"):
+        self.api_email = api_email
+        self.api_token = api_token
+        self.base_url = base_url
+
+    def list(self, url="https://api.kentik.com/api/v5/plans"):
+        headers = {
+            "User-Agent": "kentik-python-api/0.2",
+            "Content-Type": "application/json",
+            "X-CH-Auth-Email": self.api_email,
+            "X-CH-Auth-API-Token": self.api_token,
+        }
+        resp = requests.get(url, headers=headers)
+
+        # print the HTTP response to help debug
+        # print(resp.text)
+
+        # break out at first sign of trouble
+        resp.raise_for_status()
+        plans = resp.json()
+
+        return plans

kentikapi/v5/site.py

@@ -0,0 +1,37 @@
+#!/usr/bin/env python
+
+from __future__ import print_function
+from builtins import str
+from builtins import object
+import json
+import requests
+
+
+"""Site API client"""
+
+
+class Site(object):
+    """Returns all Kentik Sites"""
+
+    def __init__(self, api_email, api_token, base_url="https://api.kentik.com"):
+        self.api_email = api_email
+        self.api_token = api_token
+        self.base_url = base_url
+
+    def list(self, url="https://api.kentik.com/api/v5/sites"):
+        headers = {
+            "User-Agent": "kentik-python-api/0.2",
+            "Content-Type": "application/json",
+            "X-CH-Auth-Email": self.api_email,
+            "X-CH-Auth-API-Token": self.api_token,
+        }
+        resp = requests.get(url, headers=headers)
+
+        # print the HTTP response to help debug
+        # print(resp.text)
+
+        # break out at first sign of trouble
+        resp.raise_for_status()
+        sites = resp.json()
+
+        return sites

kentikapi/v5/tagging.py

@@ -404,7 +404,7 @@ def _submit_batch(self, url, batch):
 
         guid = ""
         headers = {
-            'User-Agent': 'kentik-python-api/0.1',
+            'User-Agent': 'kentik-python-api/0.2',
             'Content-Type': 'application/json',
             'X-CH-Auth-Email': self.api_email,
             'X-CH-Auth-API-Token': self.api_token
@@ -454,7 +454,7 @@ def fetch_batch_status(self, guid):
         """Fetch the status of a batch, given the guid"""
         url = '%s/api/v5/batch/%s/status' % (self.base_url, guid)
         headers = {
-            'User-Agent': 'kentik-python-api/0.1',
+            'User-Agent': 'kentik-python-api/0.2',
             'Content-Type': 'application/json',
             'X-CH-Auth-Email': self.api_email,
             'X-CH-Auth-API-Token': self.api_token

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='kentikapi',
-    version='0.1.10',
+    version='0.2.0',
     author='Blake Caldwell',
     packages=find_packages(),
     url='https://github.com/kentik/api-client',