def non_administrative_locations_for_select2(request, domain):
id = request.GET.get('id')
query = request.GET.get('name', '').lower()
if id:
try:
loc = SQLLocation.objects.get(location_id=id)
if loc.domain != domain:
raise SQLLocation.DoesNotExist()
except SQLLocation.DoesNotExist:
return json_response(
{'message': 'no location with id %s found' % id},
status_code=404,
)
else:
return json_response(loc_to_payload(loc))
locs = []
user = request.couch_user
user_loc = user.get_sql_location(domain)
if user_can_edit_any_location(user, request.project):
locs = SQLLocation.objects.filter(domain=domain, location_type__administrative=False)
elif user_loc:
locs = user_loc.get_descendants(include_self=True, location_type__administrative=False)
if locs != [] and query:
locs = locs.filter(name__icontains=query)
return json_response(list(map(loc_to_payload, locs[:10])))
def get_locations_from_ids(location_ids, domain):
"""
Returns the SQLLocations with the given location_ids, ensuring
that they belong to the given domain. Raises SQLLocation.DoesNotExist
if any of the locations do not match the given domain or are not
found.
"""
location_ids = list(set(location_ids))
expected_count = len(location_ids)
locations = SQLLocation.objects.filter(
domain=domain,
is_archived=False,
location_id__in=location_ids,
)
if len(locations) != expected_count:
raise SQLLocation.DoesNotExist(
'One or more of the locations was not found.')
return locations
