Create create_account_api_key_enterprise.py

Author: p-fernandes

probely_api_examples/create_account_api_key_enterprise.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+"""
+Get an API key for your account
+"""
+import getpass
+import requests
+from urllib.parse import urljoin
+
+username = input("Username: ")
+password = getpass.getpass()
+
+api_base_url = "http://localhost:8060"#"https://api.probely.com"
+auth_endpoint = urljoin(api_base_url, "enterprise/auth/obtain/")
+keys_endpoint = urljoin(api_base_url, "keys/")
+
+# Get login token
+response = requests.post(auth_endpoint,
+                         data={'username': username, 'password': password})
+print(response.json())
+token = response.json()['token']
+
+headers = {
+    'Authorization': "JWT {}".format(token),
+    'Content-Type': "application/json",
+}
+
+api_name = input("API key's name: ")
+
+response = requests.post(
+    keys_endpoint,
+    headers=headers,
+    json={'name': api_name})
+
+if response.status_code == 201:
+    print("\nAccount API key:", response.json()['key'])
+
+else:
+    print('\n[%s]\n%s' %(response.status_code, response.text))