Exemple #1
0
class Views:

    def __init__(self, route_data=None, file_data=None):
        self.data = file_data or FILE
        self.route = FindRoute(file_data=self.data)
        self.route_data = route_data

    def routes(self):
        from travels.helpers import to_json

        routes = self.route.get_rows_from_file()
        obj = to_json(routes)

        return obj

    def new_route(self):
        source = self.route_data['body']['source'].upper()
        destination = self.route_data['body']['destination'].upper()
        price = self.route_data['body']['price']

        if write_file(self.data, source, destination, price):
            return {
                'source': source,
                'destination': destination,
                'price': price
            }
        raise ErrorWriteFile('Error saving data')

    def get_route(self):
        route = FindRoute(
            self.route_data['query_params']['source'].upper(),
            self.route_data['query_params']['destination'].upper(),
            self.data
        )
        route.dijkstra()
        route.best_route()

        result = {
            'route': f"{route.source} - {' - '.join(route.path)}",
            'price': route.price
        }

        return result
Exemple #2
0
 def test_get_rows_from_file_without_file(self):
     route = FindRoute()
     rows = route.get_rows_from_file()
     self.assertEqual(rows, None)
Exemple #3
0
 def test_get_rows_from_file(self):
     route = FindRoute(file_data=FILE)
     rows = route.get_rows_from_file()
     self.assertEqual(len(rows), 7)