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 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.save()
yield company_instance
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 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_company_external_id("external id")
instance.save()
yield instance
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()}
def invite_cla_manager(contributor_id, contributor_name, contributor_email,
cla_manager_name, cla_manager_email, project_name,
company_name):
"""
Sends email to the specified CLA Manager to sign up through the Corporate
console and adds the requested user to the Approved List request queue.
:param contributor_id: The id of the user inviting the CLA Manager
:param contributor_name: The name of the user inviting the CLA Manager
:param contributor_email: The email address that this user wants to be added to the Approved List. Must exist in the user's list of emails.
:param cla_manager_name: The name of the CLA manager
:param cla_manager_email: The email address of the CLA manager
:param project_name: The name of the project
:param company_name: The name of the organization/company
"""
user = User()
try:
user.load(contributor_id)
except DoesNotExist as err:
msg = f'unable to load user by id: {contributor_id} for inviting company admin - error: {err}'
cla.log.warning(msg)
return {
'errors': {
'user_id': contributor_id,
'message': msg,
'error': str(err)
}
}
project = Project()
try:
project.load_project_by_name(project_name)
except DoesNotExist as err:
msg = f'unable to load project by name: {project_name} for inviting company admin - error: {err}'
cla.log.warning(msg)
return {
'errors': {
'project_name': project_name,
'message': msg,
'error': str(err)
}
}
company = Company()
try:
company.load_company_by_name(company_name)
except DoesNotExist as err:
msg = f'unable to load company by name: {company_name} - error: {err}'
cla.log.warning(msg)
company.set_company_id(str(uuid.uuid4()))
company.set_company_name(company_name)
company.save()
# Add user lfusername if exists
username = None
if user.get_lf_username():
username = user.get_lf_username()
elif user.get_user_name():
username = user.get_user_name()
if username:
company.add_company_acl(username)
company.save()
# create company invite
company_invite = CompanyInvite()
company_invite.set_company_invite_id(str(uuid.uuid4()))
company_invite.set_requested_company_id(company.get_company_id())
company_invite.set_user_id(user.get_user_id())
company_invite.save()
# We'll use the user's provided contributor name - if not provided use what we have in the DB
if contributor_name is None:
contributor_name = user.get_user_name()
log_msg = (
f'sent email to CLA Manager: {cla_manager_name} with email {cla_manager_email} '
f'for project {project_name} and company {company_name} '
f'to user {contributor_name} with email {contributor_email}')
# Send email to the admin. set account_exists=False since the admin needs to sign up through the Corporate Console.
cla.log.info(log_msg)
send_email_to_cla_manager(project, contributor_name, contributor_email,
cla_manager_name, cla_manager_email,
company_name, False)
# update ccla_whitelist_request
ccla_whitelist_request = CCLAWhitelistRequest()
ccla_whitelist_request.set_request_id(str(uuid.uuid4()))
ccla_whitelist_request.set_company_name(company_name)
ccla_whitelist_request.set_project_name(project_name)
ccla_whitelist_request.set_user_github_id(contributor_id)
ccla_whitelist_request.set_user_github_username(contributor_name)
ccla_whitelist_request.set_user_emails(set([contributor_email]))
ccla_whitelist_request.set_request_status("pending")
ccla_whitelist_request.save()
Event.create_event(
event_user_id=contributor_id,
event_project_name=project_name,
event_data=log_msg,
event_summary=log_msg,
event_type=EventType.InviteAdmin,
contains_pii=True,
)