Merge pull request #10 from IoT-Inspector/playground-auth-header

Playground auth header

Author: martonilles

examples/README.md

@@ -0,0 +1,8 @@
+# Example scripts
+
+This directory contains examples scripts for simple tasks. Feel free to use and customize these scripts to your needs or
+just to use them as inspiration. Please be aware that there are not much error handling in these scripts.
+
+* upload_firmware.py: example how to manage firmware uploads
+* get_tenant_token.py: The GraphQL Playground accessible at /api/graphql requires authentication. This script prints the
+ required authentication header that can be used in playgroud.

examples/get_tenant_token.py

@@ -0,0 +1,28 @@
+import json
+import sys
+from getpass import getpass
+from iot_inspector_client import Client
+
+API_URL = "https://demo.iot-inspector.com/api"
+EMAIL = sys.argv[1]
+PASSWORD = getpass()
+
+
+print("Login to IoT Inspector", EMAIL, "@", API_URL)
+client = Client(API_URL)
+client.login(EMAIL, PASSWORD)
+tenants = client.get_all_tenants()
+
+print("Tenants:", ", ".join([tenant.name for tenant in tenants]))
+
+if len(sys.argv) > 2:
+    # Filter tenants that matches the provided pattern
+    tenants = filter(lambda tenant: sys.argv[2] in tenant.name, tenants)
+
+# Pick the first one
+tenant = tenants[0]
+
+print("Using tenant:", tenant.name)
+client.use_tenant(tenant)
+
+print(json.dumps(client.get_auth_headers()))

iot_inspector_client/client.py

@@ -136,10 +136,14 @@ def _post(self, path: str, headers: Optional[Dict] = None, **kwargs):
 
     @_tenant_required
     def _post_with_token(self, path: str, **kwargs):
-        headers = {"Authorization": "Bearer " + self._state.raw_tenant_token}
+        headers = self.get_auth_headers()
 
         return self._post(path, headers, **kwargs)
 
+    @_tenant_required
+    def get_auth_headers(self):
+        return {"Authorization": "Bearer " + self._state.raw_tenant_token}
+
     @_login_required
     def get_tenant(self, name: str):
         """Get Tenant by name. Raises KeyError if not found."""