示例#1
0
class DriverFinder(object):

    _lat_inc = 3
    _score_cutoff = 10

    def __init__(self):
        self.cfg = Helpers.load_config()
        self.ds_name = self.cfg["api"]["database"]
        self.environ = self.cfg["env"]
        self.driver_bus = DriverBus(self.ds_name, self.environ)
        self.score_cutoff = 10  # ignore drivers with score lower than cutoff

    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 calculate_driver_score(self, driver, requester_location, requester_preferences):
        # TODO implement real score calculation
        return randint(0, 99)
示例#2
0
 def __init__(self):
     self.cfg = Helpers.load_config()
     self.ds_name = self.cfg["api"]["database"]
     self.environ = self.cfg["env"]
     self.driver_bus = DriverBus(self.ds_name, self.environ)
     self.score_cutoff = 10  # ignore drivers with score lower than cutoff