def update_company(company_id, # pylint: disable=too-many-arguments
company_name=None,
company_manager_id=None,
username=None):
"""
Updates an company and returns the newly updated company in dict format.
A value of None means the field should not be updated.
:param company_id: ID of the company to update.
:type company_id: ID
:param company_name: New company name.
:type company_name: string | None
:param company_manager_id: The ID of the company manager user.
:type company_manager_id: string
:return: dict representation of the company object.
:rtype: dict
"""
company = Company()
try:
company.load(str(company_id))
except DoesNotExist as err:
return {'errors': {'company_id': str(err)}}
company_acl_verify(username, company)
if company_name is not None:
company.set_company_name(company_name)
if company_manager_id is not None:
val = hug.types.uuid(company_manager_id)
company.set_company_manager_id(str(val))
company.save()
return company.to_dict()
def create_company(auth_user: AuthUser,
company_name: str = None,
company_manager_id=None,
company_manager_user_name=None,
company_manager_user_email=None,
user_id=None,
response=None):
"""
Creates an company and returns the newly created company in dict format.
:param auth_user: The authenticated user
:type auth_user: object
:param company_name: The company name.
:type company_name: string
:param company_manager_id: The ID of the company manager user.
:type company_manager_id: string
:param company_manager_user_name: The user name of the company manager user.
:type company_manager_user_name: string
:param company_manager_user_email: The user email of the company manager user.
:type company_manager_user_email: string
:return: dict representation of the company object.
:rtype: dict
"""
fn = 'controllers.company.create_company'
manager = cla.controllers.user.get_or_create_user(auth_user)
for company in get_companies():
if company.get("company_name") == company_name:
cla.log.error({"error": "Company already exists"})
response.status = HTTP_409
return {"status_code": HTTP_409,
"data": {"error": "Company already exists.",
"company_id": company.get("company_id")}
}
cla.log.debug(f'{fn} - creating company with name: {company_name}')
company = Company()
company.set_company_id(str(uuid.uuid4()))
company.set_company_name(company_name)
company.set_company_manager_id(manager.get_user_id())
company.set_company_acl(manager.get_lf_username())
company.save()
cla.log.debug(f'{fn} - created company with name: {company_name} with company_id: {company.get_company_id()}')
# Create audit trail for company
event_data = f'User {auth_user.username} created Company {company.get_company_name()} ' \
f'with company_id: {company.get_company_id()}.'
event_summary = f'User {auth_user.username} created Company {company.get_company_name()}.'
Event.create_event(
event_type=EventType.CreateCompany,
event_company_id=company.get_company_id(),
event_data=event_data,
event_summary=event_summary,
event_user_id=user_id,
contains_pii=False,
)
return {"status_code": HTTP_200, "data": company.to_dict()}
def create_company(auth_user,
company_name=None,
company_manager_id=None,
company_manager_user_name=None,
company_manager_user_email=None,
user_id=None,
response=None):
"""
Creates an company and returns the newly created company in dict format.
:param company_name: The company name.
:type company_name: string
:param company_manager_id: The ID of the company manager user.
:type company_manager_id: string
:param company_manager_user_name: The user name of the company manager user.
:type company_manager_user_name: string
:param company_manager_user_email: The user email of the company manager user.
:type company_manager_user_email: string
:return: dict representation of the company object.
:rtype: dict
"""
manager = cla.controllers.user.get_or_create_user(auth_user)
for company in get_companies():
if company.get("company_name") == company_name:
cla.log.error({"error": "Company already exists"})
response.status = HTTP_409
return {
"status_code": HTTP_409,
"data": {
"error": "Company already exists.",
"company_id": company.get("company_id")
}
}
company = Company()
company.set_company_id(str(uuid.uuid4()))
company.set_company_name(company_name)
company.set_company_manager_id(manager.get_user_id())
company.set_company_acl(manager.get_lf_username())
company.save()
# Create audit trail for company
event_data = 'Company-{} created'.format(company.get_company_name())
Event.create_event(
event_type=EventType.CreateCompany,
event_company_id=company.get_company_id(),
event_data=event_data,
event_user_id=user_id,
contains_pii=False,
)
return {"status_code": HTTP_200, "data": company.to_dict()}
def update_company(
company_id: str, # pylint: disable=too-many-arguments
company_name: str = None,
company_manager_id: str = None,
username: str = None):
"""
Updates an company and returns the newly updated company in dict format.
A value of None means the field should not be updated.
:param company_id: ID of the company to update.
:type company_id: str
:param company_name: New company name.
:type company_name: string | None
:param company_manager_id: The ID of the company manager user.
:type company_manager_id: str
:param username: The username of the existing company manager user who performs the company update.
:type username: str
:return: dict representation of the company object.
:rtype: dict
"""
company = Company()
try:
company.load(str(company_id))
except DoesNotExist as err:
return {'errors': {'company_id': str(err)}}
company_acl_verify(username, company)
update_str = ""
if company_name is not None:
company.set_company_name(company_name)
update_str += f"The company name was updated to {company_name}. "
if company_manager_id is not None:
val = hug.types.uuid(company_manager_id)
company.set_company_manager_id(str(val))
update_str += f"The company company manager id was updated to {val}"
company.save()
# Audit update event
event_data = update_str
Event.create_event(
event_data=event_data,
event_summary=event_data,
event_type=EventType.UpdateCompany,
event_company_id=company_id,
contains_pii=False,
)
return company.to_dict()
def create_company(auth_user,
company_name=None,
company_manager_id=None,
company_manager_user_name=None,
company_manager_user_email=None,
user_id=None):
"""
Creates an company and returns the newly created company in dict format.
:param company_name: The company name.
:type company_name: string
:param company_manager_id: The ID of the company manager user.
:type company_manager_id: string
:param company_manager_user_name: The user name of the company manager user.
:type company_manager_user_name: string
:param company_manager_user_email: The user email of the company manager user.
:type company_manager_user_email: string
:return: dict representation of the company object.
:rtype: dict
"""
manager = cla.controllers.user.get_or_create_user(auth_user)
for company in get_companies():
if company.get("company_name") == company_name:
cla.log.error({"error": "Company already exists"})
return {
"status_code": HTTP_409,
"data": {
"error": "Company already exists.",
"company_id": company.get("company_id")
}
}
company = Company()
company.set_company_id(str(uuid.uuid4()))
company.set_company_name(company_name)
company.set_company_manager_id(manager.get_user_id())
company.set_company_acl(manager.get_lf_username())
company.save()
return {"status_code": HTTP_200, "data": company.to_dict()}
