Beispiel #1
0
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