import requests


class Vao:
    def __init__(self, access_id: str):
        self.access_id = access_id
        self.base_url = 'https://routenplaner.verkehrsauskunft.at/vao/restproxy'
        self.version = '1.8.0'

    def get(self, end_point: str, parameter: dict):
        params = {'accessId': self.access_id, 'format': 'json'}
        params.update(parameter)
        url = f'{self.base_url}/v{self.version}/{end_point}'
        return requests.get(url, params=params)

    def trip(self, parameter: dict):
        return self.get('trip', parameter)