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'
def create(self, name, user):
"""Creates a new OSM user
"""
- if 'projects' in user and user['projects'] is not None:
- user['projects'] = yaml.safe_load(user['projects'])
+ 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))
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:
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:
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)
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']:
return user
else:
for user in self.list():
- if name == user['name']:
+ if name == user['username']:
return user
raise NotFound("User {} not found".format(name))
-