def company(company_table):
""" create project """
with patch(PATCH_METHOD) as req:
req.return_value = {}
company_instance = Company()
company_instance.set_company_id("foo_company_id")
company_instance.set_company_name("foo_company_name")
company_instance.set_signing_entity_name("foo_comapny_entity_name")
company_instance.save()
yield company_instance
def company_instance():
"""
Mock Company instance
"""
with patch(PATCH_METHOD) as req:
req.return_value = COMPANY_TABLE_DATA
instance = Company()
instance.set_company_id("uuid")
instance.set_company_name("co")
instance.set_signing_entity_name("co entity name")
instance.set_company_external_id("external id")
instance.save()
yield instance
def create_company(auth_user: AuthUser,
company_name: str = None,
signing_entity_name: str = None,
company_manager_id: str = None,
company_manager_user_name: str = None,
company_manager_user_email: str = None,
user_id: str = 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 signing_entity_name: The company's signing entity name.
:type signing_entity_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} with signing entity name: {signing_entity_name}'
)
company = Company()
company.set_company_id(str(uuid.uuid4()))
company.set_company_name(company_name)
company.set_signing_entity_name(signing_entity_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()}
