def get_or_add_location(place_list, mig_user=mig_user):
place_list = map(lambda x: x.title(), filter(lambda x: x, place_list))
prev_loc = None
prev_tree = Location.objects.all()
location_created = False
for in_loc_name in reversed(place_list):
found_loc = prev_tree.filter(Q(name__exact=in_loc_name), Q(politically_in=None))
found_loc_count = found_loc.count()
if found_loc_count > 1:
loc = None
if prev_loc:
immediate_subloc_tree = prev_loc.get_geographic_sub_locations(include_self=False, max_distance=1)
immed_found_loc = immediate_subloc_tree.filter(Q(name__exact=in_loc_name), Q(politically_in=None))
if immed_found_loc.count() > 0: # Just taking the first match no matter what for now
loc = immed_found_loc[0]
if not loc:
loc = found_loc[0] # Taking the first match for now
elif found_loc_count == 0:
loc = Location(
name=in_loc_name, geographically_in=prev_loc, active=True, submitted_by=mig_user
) # , log = default_log)
loc.save()
location_created = True
else:
loc = found_loc[0]
if location_created and not prev_loc.is_root():
raise LocationTooComplicated("Found existing location after having created a non-existent one")
else:
loc.in_location = prev_loc
prev_loc = loc
prev_tree = prev_loc.get_geographic_sub_locations(include_self=False)
return prev_loc
