Merge pull request #1 from warrant-dev/AddListWarrants

Add list warrants

Author: stanleyphu

examples/example.py

@@ -4,15 +4,18 @@ def make_warrant_requests(api_key):
     client = Warrant(api_key)
 
     # Create users and session tokens
-    print("Created user with provided id: " + client.create_user("custom_id_001"))
+    print("Created tenant with provided id: " + client.create_tenant("custom_tenant_001"))
+    print("Created user with provided id: " + client.create_user("custom_user_001"))
     new_user = client.create_user()
     print("Created user with generated id: " + new_user)
+    print(client.create_warrant(object_type="tenant", object_id="custom_tenant_001", relation="member", user=new_user))
     print("Created session token: " + client.create_session(new_user))
 
     # Create and check warrants
     print(client.create_warrant(object_type="store", object_id="store1", relation="owner", user=new_user))
     is_authorized = client.is_authorized(object_type="store", object_id="store1", relation="owner", user_to_check=new_user)
     print(f"New user authorization result: {is_authorized}")
+    print(f"All warrants: {client.list_warrants()}")
 
 if __name__ == '__main__':
     # Replace with your Warrant api key

warrant/__init__.py

@@ -32,6 +32,14 @@ def _make_post_request(self, uri, json={}):
         else:
             raise WarrantException(msg=resp.text, status_code=resp.status_code)
 
+    def _make_get_request(self, uri, params={}):
+        headers = { "Authorization": "ApiKey " + self._apiKey }
+        resp = requests.get(url = API_ENDPOINT+API_VERSION+uri, headers = headers, params = params)
+        if resp.status_code == 200:
+            return resp.json()
+        else:
+            raise WarrantException(msg=resp.text, status_code=resp.status_code)
+
     def create_user(self, user_id=""):
         if user_id == "":
             payload = {}
@@ -40,6 +48,14 @@ def create_user(self, user_id=""):
         json = self._make_post_request(uri="/users", json=payload)
         return json['userId']
 
+    def create_tenant(self, tenant_id=""):
+        if tenant_id == "":
+            payload = {}
+        else:
+            payload = { "tenantId": tenant_id }
+        json = self._make_post_request(uri="/tenants", json=payload)
+        return json['tenantId']
+
     def create_session(self, user_id):
         if user_id == "":
             raise WarrantException(msg="Invalid userId provided")
@@ -63,6 +79,16 @@ def create_warrant(self, object_type, object_id, relation, user):
         resp = self._make_post_request(uri="/warrants", json=payload)
         return resp['id']
 
+    def list_warrants(self, object_type="", object_id="", relation="", user_id=""):
+        filters = {
+            "objectType": object_type,
+            "objectId": object_id,
+            "relation": relation,
+            "userId": user_id,
+        }
+        resp = self._make_get_request(uri="/warrants", params=filters)
+        return resp
+
     def is_authorized(self, object_type, object_id, relation, user_to_check):
         if object_type == "" or object_id == "" or relation == "":
             raise WarrantException(msg="Invalid object_type, object_id and/or relation")