def search_orgs(search: OrgSearch): # pylint: disable=too-many-locals
"""Search for orgs based on input parameters."""
orgs_result = {
'orgs': [],
'page': search.page,
'limit': search.limit,
'total': 0
}
include_invitations: bool = False
search.access_type, is_staff_admin = Org.refine_access_type(
search.access_type)
if search.statuses and OrgStatus.PENDING_ACTIVATION.value in search.statuses:
# only staff admin can see director search accounts
if not is_staff_admin:
raise BusinessException(Error.INVALID_USER_CREDENTIALS, None)
org_models, orgs_result[
'total'] = OrgModel.search_pending_activation_orgs(
name=search.name)
include_invitations = True
else:
org_models, orgs_result['total'] = OrgModel.search_org(search)
for org in org_models:
orgs_result['orgs'].append({
**Org(org).as_dict(),
'contacts': [
ContactSchema(exclude=('links', )).dump(
org.contacts[0].contact, many=False)
] if org.contacts else [],
'invitations': [
InvitationSchema(exclude=('membership', )).dump(
org.invitations[0].invitation, many=False)
] if include_invitations and org.invitations else [],
})
return orgs_result
def search_orgs(**kwargs): # pylint: disable=too-many-locals
"""Search for orgs based on input parameters."""
orgs = {'orgs': []}
if kwargs.get('business_identifier', None):
affiliation: AffiliationModel = AffiliationModel. \
find_affiliations_by_business_identifier(kwargs.get('business_identifier'))
if affiliation:
orgs['orgs'].append(
Org(OrgModel.find_by_org_id(affiliation.org_id)).as_dict())
else:
include_invitations: bool = False
page: int = int(kwargs.get('page'))
limit: int = int(kwargs.get('limit'))
statuses: str = kwargs.get('statuses', None)
name: str = kwargs.get('name', None)
# https://github.com/bcgov/entity/issues/4786
access_type, is_staff_admin = Org.refine_access_type(
kwargs.get('access_type', None), kwargs.get('token', None))
search_args = (access_type, name, statuses,
kwargs.get('bcol_account_id', None), page, limit)
if statuses and OrgStatus.PENDING_ACTIVATION.value in statuses:
# only staff admin can see director search accounts
# https://github.com/bcgov/entity/issues/4786
if not is_staff_admin:
raise BusinessException(Error.INVALID_USER_CREDENTIALS,
None)
org_models, total = OrgModel.search_pending_activation_orgs(
name)
include_invitations = True
else:
org_models, total = OrgModel.search_org(*search_args)
for org in org_models:
org_dict = Org(org).as_dict()
org_dict['contacts'] = []
org_dict['invitations'] = []
if org.contacts:
org_dict['contacts'].append(
ContactSchema(exclude=('links', )).dump(
org.contacts[0].contact, many=False))
if include_invitations and org.invitations:
org_dict['invitations'].append(
InvitationSchema(exclude=('membership', )).dump(
org.invitations[0].invitation, many=False))
orgs['orgs'].append(org_dict)
orgs['total'] = total
orgs['page'] = page
orgs['limit'] = limit
return orgs
def get(org_id):
"""Retrieve the set of invitations for the given org."""
try:
invitation_status = request.args.get('status').upper() if request.args.get('status') else None
invitations = InvitationService.get_invitations_for_org(org_id=org_id,
status=invitation_status)
response, status = {'invitations': InvitationSchema().dump(invitations, many=True)}, http_status.HTTP_200_OK
except BusinessException as exception:
response, status = {'code': exception.code, 'message': exception.message}, exception.status_code
return response, status
def as_dict(self):
"""Return the internal Invitation model as a dictionary."""
invitation_schema = InvitationSchema()
obj = invitation_schema.dump(self._model, many=False)
return obj
