add collection managment, and first pass at permissions

ms1752 · ms1752 · commit eccc43fd3503 · 2018-02-07T13:44:50.000-07:00
diff --git a/clearblade/Collections.py b/clearblade/Collections.py
@@ -8,8 +8,12 @@ class Collection():
     def __init__(self, system, authenticatedUser, collectionID="", collectionName=""):
         if collectionID:
             self.url = system.url + '/api/v/1/data/' + collectionID
+            self.collectionID = collectionID
+            self.collectionName = None
         elif collectionName:
             self.url = system.url + '/api/v/1/collection/' + system.systemKey + "/" + collectionName
+            self.collectionName = collectionName
+            self.collectionID = None
         else:
             cbLogs.error("You must supply either a collection name or id.")  # beep
             exit(-1)
@@ -69,3 +73,37 @@ def updateItems(self, query, data):
 
     def deleteItems(self, query):
         return restcall.delete(self.url, headers=self.headers, params={"query": json.dumps(query.filters)}, sslVerify=self.sslVerify)
+
+
+###########################
+#   DEVELOPER ENDPOINTS   #
+###########################
+
+def DEVnewCollection(developer, system, name):
+    url = system.url + "/admin/collectionmanagement"
+    data = {
+        "appID": system.systemKey,
+        "name": name
+    }
+    resp = restcall.post(url, headers=developer.headers, data=data, sslVerify=system.sslVerify)
+    cbLogs.info("Successfully created collection: " + name)
+    newCollection = Collection(system, developer, collectionID=resp["collectionID"])
+    return newCollection
+
+def DEVaddColumnToCollection(developer, system, collection, columnName, columnType):
+    if not collection.collectionID:
+        cbLogs.error("You must supply the collection id when adding a column to a collection.")
+        exit(-1)
+    url = system.url + "/admin/collectionmanagement"
+    data = {
+        "id": collection.collectionID,
+        "addColumn": {
+            "id": collection.collectionID,
+            "name": columnName,
+            "type": columnType
+        }
+    }
+    resp = restcall.put(url, headers=developer.headers, data=data, sslVerify=system.sslVerify)
+    cbLogs.info("Successfully added column: " + columnName)
+    return resp
+
diff --git a/clearblade/Developers.py b/clearblade/Developers.py
@@ -1,8 +1,9 @@
 from __future__ import absolute_import
 from . import restcall
 from . import cbLogs
+from . import Collections
 from . import Devices
-
+from . import Permissions
 
 def registerDev(fname, lname, org, email, password, url="https://platform.clearblade.com"):
     newDevCredentials = {
@@ -65,6 +66,16 @@ def logout(self):
     # ~~~~~~~~~~~~~~~~~~~~~~ #
     ##########################
 
+    #################
+    #  Collections  #
+    #################
+
+    def newCollection(self, system, name):
+        return Collections.DEVnewCollection(self, system, name)
+
+    def addColumnToCollection(self, system, collection, columnName, columnType):
+        return Collections.DEVaddColumnToCollection(self, system, collection, columnName, columnType)
+
     ###############
     #   Devices   #
     ###############
@@ -83,3 +94,11 @@ def updateDevice(self, system, name, updates):
 
     def deleteDevice(self, system, name):
         return Devices.DEVdeleteDevice(self, system, name)
+
+    #################
+    #  Permissions  #
+    #################
+
+    def setPermissionsForCollection(self, system, collection, permissionsLevel, roleName):
+        return Permissions.DEVsetPermissionsForCollection(self, system, collection, permissionsLevel, roleName)
+
diff --git a/clearblade/Permissions.py b/clearblade/Permissions.py
@@ -0,0 +1,33 @@
+from __future__ import absolute_import
+from . import cbLogs
+from . import restcall
+
+class Permissions:
+    READ = 1
+    CREATE = 2
+    UPDATE = 4
+    DELETE = 8
+
+
+###########################
+#   DEVELOPER ENDPOINTS   #
+###########################
+
+def DEVsetPermissionsForCollection(developer, system, collection, permissionsLevel, roleName):
+    url = system.url + "/admin/user/" + system.systemKey + "/roles"
+    data = {
+        "id": roleName,
+        "changes": {
+            "collections": [{
+                "itemInfo": {
+                    "name": collection.collectionName,
+                    "id": collection.collectionID
+                },
+                "permissions": permissionsLevel
+            }]
+        }
+    }
+    resp = restcall.put(url, headers=developer.headers, data=data, sslVerify=system.sslVerify)
+    cbLogs.info("Successfully updated permissions for role: " + roleName)
+    return resp
+    
