Merge pull request #64 from dynata/dev

User Management SDKs

Author: DynataRidley

dynatademand/api.py

@@ -291,6 +291,44 @@ def get_project_detailed_report(self, project_id):
         )
         return self._api_get('/projects/{}/detailedReport'.format(project_id))
 
+    def get_user_info(self):
+        return self._api_get('/user')
+
+    def get_company_users(self):
+        return self._api_get('/users')
+
+    def get_company_teams(self):
+        return self._api_get('/teams')
+
+    def get_roles(self, **kwargs):
+        self.validator.validate_request(
+            'get_roles',
+            query_params=kwargs
+        )
+        return self._api_get('/roles', kwargs)
+
+    def get_project_permissions(self, project_id):
+        self.validator.validate_request(
+            'get_project_permissions',
+            path_data={'extProjectId': '{}'.format(project_id)},
+        )
+        return self._api_get('/projects/{}/permissions'.format(project_id))
+
+    def upsert_project_permissions(self, project_id, upsert_permissions_data):
+        self.validator.validate_request(
+            'upsert_project_permissions',
+            path_data={'extProjectId': '{}'.format(project_id)},
+            request_body=upsert_permissions_data,
+        )
+        response_data = self._api_post('/projects/{}/permissions'.format(project_id), upsert_permissions_data)
+        if response_data.get('status').get('message') != 'success':
+            raise DemandAPIError(
+                "Could not upsert project permissions. Demand API responded with: {}".format(
+                    response_data
+                )
+            )
+        return response_data
+
     def add_line_item(self, project_id, lineitem_data):
         '''
             A line item is a project entity that exist for a specific market and
@@ -524,7 +562,7 @@ def delete_template(self, id):
         self.validator.validate_request(
             'delete_template',
             path_data={'id': '{}'.format(id)},
-         )
+        )
         return self._api_delete('/templates/quotaplan/{}'.format(id))
 
     def get_templates(self, country, lang, **kwargs):

dynatademand/schemas/request/body/upsert_project_permissions.json

@@ -0,0 +1,30 @@
+{
+    "type": "object",
+    "properties": {
+        "users": {
+            "type": "array",
+            "items": {
+                "type": "object",
+                "properties": {
+                    "id": {
+                        "type": "integer"
+                    },
+                    "role": {
+                        "type": "string"
+                    }
+                }
+            }
+        },
+        "teams": {
+            "type": "array",
+            "items": {
+                "type": "object",
+                "properties": {
+                    "id": {
+                        "type": "integer"
+                    }
+                }
+            }
+        }
+    }
+}

dynatademand/schemas/request/path/get_project_permissions.json

@@ -0,0 +1,12 @@
+{
+    "type": "object",
+    "properties": {
+        "extProjectId": {
+            "type": "string",
+            "required": true
+        }
+    },
+    "required": [
+        "extProjectId"
+    ]
+}

dynatademand/schemas/request/path/upsert_project_permissions.json

@@ -0,0 +1,12 @@
+{
+    "type": "object",
+    "properties": {
+        "extProjectId": {
+            "type": "string",
+            "required": true
+        }
+    },
+    "required": [
+        "extProjectId"
+    ]
+}

dynatademand/schemas/request/query/get_roles.json

@@ -0,0 +1,16 @@
+{
+    "type": "object",
+    "properties": {
+        "name": {
+            "type": "string",
+            "description": "This is the name of the role. eg: manager"
+        },
+        "id": {
+            "type": "array",
+            "items": {
+                "type": "string"
+            }
+        }
+    },
+    "required": []
+}

dynatademand/validator.py

@@ -17,6 +17,11 @@
     'buy_project': ['path', 'body', ],
     'get_project_detailed_report': ['path', ],
     'reconcile_project': ['path', ],
+    'get_project_permissions': ['path', ],
+    'upsert_project_permissions': ['path', 'body', ],
+
+    # Roles
+    'get_roles': ['query', ],
 
     # Invoices
     'get_invoice': ['path', ],

pyproject.toml

@@ -18,6 +18,7 @@ pytest = "4.6.6"
 jsonschema = "3.2.0"
 pytest-runner = "5.2"
 flake8 = "^3.7.9"
+pyrsistent = "0.14.11"
 
 [build-system]
 requires = ["poetry>=0.12"]

requirements.txt

@@ -0,0 +1,2 @@
+responses>=0.10.0
+jsonschema>=2.5.0

tests/test_files/get_company_users.json

@@ -0,0 +1,14 @@
+[
+    {
+        "email": "samplifyweb@dynata.com",
+        "id": 71,
+        "name": "Bears Beets",
+        "userName": "battlestargalactica"
+    },
+    {
+        "email": "samplify@dynata.com",
+        "id": 102,
+        "name": "Michael Scott",
+        "userName": "papercompany"
+    }
+]

tests/test_files/get_project_permissions.json

@@ -0,0 +1,21 @@
+{
+    "currentUser": {
+        "roles": [
+            "PROJECT_MANAGER"
+        ]
+    },
+    "extProjectId": "1910a9fe-59de-4796-aac6-8b7356a7d340",
+    "teams": [
+        {
+            "id": 139,
+            "name": "All Users"
+        }
+    ],
+    "users": [
+        {
+            "id": 71,
+            "role": "PROJECT_MANAGER",
+            "username": "samplifyweb"
+        }
+    ]
+}