X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osmclient%2Fsol005%2Fuser.py;h=6c10325add5802c036019fcf09731f32b0ea23e1;hb=63ab62675e9db9c956f7a676e2f0230f01d8ce9a;hp=29635f9575de62d72c58a2e89f9fbb209527cce6;hpb=fdc97c8f4c123dcb23865e08d187df025123a7f8;p=osm%2Fosmclient.git diff --git a/osmclient/sol005/user.py b/osmclient/sol005/user.py index 29635f9..6c10325 100644 --- a/osmclient/sol005/user.py +++ b/osmclient/sol005/user.py @@ -23,13 +23,14 @@ from osmclient.common import utils from osmclient.common.exceptions import ClientException from osmclient.common.exceptions import NotFound import json -# import yaml +import logging class User(object): def __init__(self, http=None, client=None): self._http = http self._client = client + self._logger = logging.getLogger('osmclient') self._apiName = '/admin' self._apiVersion = '/v1' self._apiResource = '/users' @@ -39,8 +40,30 @@ class User(object): def create(self, name, user): """Creates a new OSM user """ - if len(user["projects"]) == 1: + self._logger.debug("") + self._client.get_token() + if not user["projects"] or (len(user["projects"]) == 1 and not user["projects"][0]): + del user["projects"] + elif len(user["projects"]) == 1: user["projects"] = user["projects"][0].split(",") + + if user["project_role_mappings"]: + project_role_mappings = [] + + for set_mapping in user["project_role_mappings"]: + set_mapping_clean = [m.strip() for m in set_mapping.split(",")] + project, roles = set_mapping_clean[0], set_mapping_clean[1:] + + for role in roles: + mapping = {"project": project, "role": role} + + if mapping not in project_role_mappings: + project_role_mappings.append(mapping) + + user["project_role_mappings"] = project_role_mappings + else: + del user["project_role_mappings"] + http_code, resp = self._http.post_cmd(endpoint=self._apiBase, postfields_dict=user) #print('HTTP CODE: {}'.format(http_code)) @@ -64,18 +87,76 @@ class User(object): def update(self, name, user): """Updates an existing OSM user identified by name """ - myuser = self.get(name) - http_code, resp = self._http.put_cmd(endpoint='{}/{}'.format(self._apiBase,myuser['_id']), - postfields_dict=user) - #print('HTTP CODE: {}'.format(http_code)) - #print('RESP: {}'.format(resp)) - if http_code in (200, 201, 202, 204): + self._logger.debug("") + self._client.get_token() + # print(user) + myuser = self.get(name) + update_user = { + "add_project_role_mappings": [], + "remove_project_role_mappings": [], + } + + # if password is defined, update the password + if user["password"]: + update_user["password"] = user["password"] + if user["username"]: + update_user["username"] = user["username"] + + if user["set-project"]: + # Remove project and insert project role mapping + for set_project in user["set-project"]: + + set_project_clean = [m.strip() for m in set_project.split(",")] + project, roles = set_project_clean[0], set_project_clean[1:] + + update_user["remove_project_role_mappings"].append({"project": project}) + + for role in roles: + mapping = {"project": project, "role": role} + update_user["add_project_role_mappings"].append(mapping) + + if user["remove-project"]: + for remove_project in user["remove-project"]: + update_user["remove_project_role_mappings"].append({"project": remove_project}) + + if user["add-project-role"]: + for add_project_role in user["add-project-role"]: + add_project_role_clean = [m.strip() for m in add_project_role.split(",")] + project, roles = add_project_role_clean[0], add_project_role_clean[1:] + + for role in roles: + mapping = {"project": project, "role": role} + update_user["add_project_role_mappings"].append(mapping) + + if user["remove-project-role"]: + for remove_project_role in user["remove-project-role"]: + remove_project_role_clean = [m.strip() for m in remove_project_role.split(",")] + project, roles = remove_project_role_clean[0], remove_project_role_clean[1:] + + for role in roles: + mapping = {"project": project, "role": role} + update_user["remove_project_role_mappings"].append(mapping) + + if not update_user["remove_project_role_mappings"]: + del update_user["remove_project_role_mappings"] + if not update_user["add_project_role_mappings"]: + del update_user["add_project_role_mappings"] + if not update_user: + raise ClientException("At least something should be changed.") + + http_code, resp = self._http.patch_cmd(endpoint='{}/{}'.format(self._apiBase, myuser['_id']), + postfields_dict=update_user) + # print('HTTP CODE: {}'.format(http_code)) + # print('RESP: {}'.format(resp)) + if http_code in (200, 201, 202): if resp: resp = json.loads(resp) if not resp or 'id' not in resp: raise ClientException('unexpected response from server - {}'.format( resp)) print(resp['id']) + elif http_code == 204: + print('Updated') else: msg = "" if resp: @@ -88,6 +169,8 @@ class User(object): def delete(self, name, force=False): """Deletes an existing OSM user identified by name """ + self._logger.debug("") + self._client.get_token() user = self.get(name) querystring = '' if force: @@ -114,6 +197,8 @@ class User(object): def list(self, filter=None): """Returns the list of OSM users """ + self._logger.debug("") + self._client.get_token() filter_string = '' if filter: filter_string = '?{}'.format(filter) @@ -126,6 +211,8 @@ class User(object): def get(self, name): """Returns an OSM user based on name or id """ + self._logger.debug("") + self._client.get_token() if utils.validate_uuid4(name): for user in self.list(): if name == user['_id']: @@ -136,4 +223,3 @@ class User(object): return user raise NotFound("User {} not found".format(name)) -