def run(self, requester_location, desired_drivers, max_depth=5, requester_preferences=None):
requester_location = Helpers.validate_geo_point(requester_location)
result = []
last_bounding_box = (None, None)
exp_factor = 1
while len(result) < desired_drivers and exp_factor <= max_depth:
# TODO check if calculation of bounding_box is correct
left = requester_location["lat"] + (pow(DriverFinder._lat_inc, exp_factor) / 69.0)
top = 3960 * 2 * math.pi / 360 * math.cos(left)
top_left = Helpers.validate_geo_point((left, top))
right = requester_location["lat"] - (pow(DriverFinder._lat_inc, exp_factor) / 69.0)
bottom = 3960 * 2 * math.pi / 360 * math.cos(right)
bottom_right = Helpers.validate_geo_point((right, bottom))
drivers_in_area = self.driver_bus.list_in_rectangle(
top_left, bottom_right, True,
last_bounding_box[0], last_bounding_box[1] # area to exclude
)
# TODO parallel processing of calculate drivers scores
for driver in drivers_in_area:
score = self.calculate_driver_score(driver, requester_location, requester_preferences)
if score > DriverFinder._score_cutoff:
result.append((driver, score))
last_bounding_box = (top_left, bottom_right)
print last_bounding_box
exp_factor += 1
def get_score(item):
return item[1]
# return drivers ordered by score
return [driver[0] for driver in sorted(result, key=get_score, reverse=True)]
def list_in_rectangle(
self, top_left, bottom_right, only_active=True, top_left_exclude=None, bottom_right_exclude=None
):
return self.dao.list_in_rectangle(
Helpers.validate_geo_point(top_left),
Helpers.validate_geo_point(bottom_right),
only_active,
top_left_exclude,
bottom_right_exclude,
)
