python-navitia-client 1.1.0__py3-none-any.whl

navitia_client/client/apis/public_transportation_apis/line_apis.py

@@ -0,0 +1,303 @@
+from typing import Any, Optional, Sequence, Tuple
+from navitia_client.client.apis.api_base_client import ApiBaseClient
+from navitia_client.client.apis.public_transportation_apis.entity_apis import EntityApi
+from navitia_client.entities.line_and_route import Line
+from navitia_client.entities.pagination import Pagination
+
+
+class LineApiClient(ApiBaseClient, EntityApi[Line]):
+    """
+    API client for handling 'Line' entities in the Navitia API.
+
+    See https://doc.navitia.io/#pt-ref
+
+    Attributes
+    ----------
+    entity_name : str
+        Name of the entity ('lines').
+    get_navitia_api : method
+        Method to get the Navitia API.
+
+    Methods
+    -------
+    _get_entity_from_response(raw_entity_response: Any) -> Sequence[Line]:
+        Static method to extract Line instances from the raw API response.
+
+    list_entity_collection_from_region(
+        region_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None
+    ) -> Tuple[Sequence[Line], Pagination]:
+        List lines for a given region.
+
+    get_entity_by_id(
+        region_id: str,
+        object_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None
+    ) -> Tuple[Sequence[Line], Pagination]:
+        Get a line by its ID in a given region.
+
+    list_entity_collection_from_coordinates(
+        lon: float,
+        lat: float,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None
+    ) -> Tuple[Sequence[Line], Pagination]:
+        List lines for given geographic coordinates.
+
+    get_entity_by_id_and_coordinates(
+        lon: float,
+        lat: float,
+        object_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None
+    ) -> Tuple[Sequence[Line], Pagination]:
+        Get a line by its ID for given geographic coordinates.
+    """
+
+    entity_name: str = "lines"
+    get_navitia_api = ApiBaseClient.get_navitia_api
+
+    @staticmethod
+    def _get_entity_from_response(raw_entity_response: Any) -> Sequence[Line]:
+        """
+        Static method to extract Line instances from the raw API response.
+
+        Parameters
+        ----------
+        raw_entity_response : Any
+            Raw API response containing Line data.
+
+        Returns
+        -------
+        Sequence[Line]
+            List of Line instances.
+        """
+        entities = []
+        for entity in raw_entity_response:
+            entities.append(Line.from_payload(entity))
+        return entities
+
+    def list_entity_collection_from_region(
+        self,
+        region_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None,
+    ) -> Tuple[Sequence[Line], Pagination]:
+        """
+        List lines for a given region.
+
+        Parameters
+        ----------
+        region_id : str
+            ID of the region.
+        start_page : int, optional
+            Starting page number (default is 0).
+        count : int, optional
+            Number of items per page (default is 25).
+        depth : int, optional
+            Search depth (default is 1).
+        odt : str, optional
+            ODT type filter (default is "all").
+        distance : int, optional
+            Maximum search distance (default is 200).
+        headsign : Optional[str], optional
+            Line headsign.
+
+        Returns
+        -------
+        Tuple[Sequence[Line], Pagination]
+            List of Line instances and pagination information.
+        """
+        filters = {
+            "start_page": start_page,
+            "count": count,
+            "depth": depth,
+            "odt": odt,
+            "distance": distance,
+        }
+
+        if headsign is not None:
+            filters["headsign"] = headsign
+        url = f"{self.base_navitia_url}/coverage/{region_id}/{self.entity_name}"
+        return self._get_entity_results(url, self.entity_name, filters)
+
+    def get_entity_by_id(
+        self,
+        region_id: str,
+        object_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None,
+    ) -> Tuple[Sequence[Line], Pagination]:
+        """
+        Get a line by its ID in a given region.
+
+        Parameters
+        ----------
+        region_id : str
+            ID of the region.
+        object_id : str
+            ID of the line.
+        start_page : int, optional
+            Starting page number (default is 0).
+        count : int, optional
+            Number of items per page (default is 25).
+        depth : int, optional
+            Search depth (default is 1).
+        odt : str, optional
+            ODT type filter (default is "all").
+        distance : int, optional
+            Maximum search distance (default is 200).
+        headsign : Optional[str], optional
+            Line headsign.
+
+        Returns
+        -------
+        Tuple[Sequence[Line], Pagination]
+            List of Line instances and pagination information.
+        """
+        filters = {
+            "start_page": start_page,
+            "count": count,
+            "depth": depth,
+            "odt": odt,
+            "distance": distance,
+        }
+
+        if headsign is not None:
+            filters["headsign"] = headsign
+
+        url = f"{self.base_navitia_url}/coverage/{region_id}/{self.entity_name}/{object_id}"
+        return self._get_entity_results(url, self.entity_name, filters)
+
+    def list_entity_collection_from_coordinates(
+        self,
+        lon: float,
+        lat: float,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None,
+    ) -> Tuple[Sequence[Line], Pagination]:
+        """
+        List lines for given geographic coordinates.
+
+        Parameters
+        ----------
+        lon : float
+            Longitude.
+        lat : float
+            Latitude.
+        start_page : int, optional
+            Starting page number (default is 0).
+        count : int, optional
+            Number of items per page (default is 25).
+        depth : int, optional
+            Search depth (default is 1).
+        odt : str, optional
+            ODT type filter (default is "all").
+        distance : int, optional
+            Maximum search distance (default is 200).
+        headsign : Optional[str], optional
+            Line headsign.
+
+        Returns
+        -------
+        Tuple[Sequence[Line], Pagination]
+            List of Line instances and pagination information.
+        """
+        filters = {
+            "start_page": start_page,
+            "count": count,
+            "depth": depth,
+            "odt": odt,
+            "distance": distance,
+        }
+
+        if headsign is not None:
+            filters["headsign"] = headsign
+
+        url = f"{self.base_navitia_url}/coverage/{lon};{lat}/{self.entity_name}"
+        return self._get_entity_results(url, self.entity_name, filters)
+
+    def get_entity_by_id_and_coordinates(
+        self,
+        lon: float,
+        lat: float,
+        object_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None,
+    ) -> Tuple[Sequence[Line], Pagination]:
+        """
+        Get a line by its ID for given geographic coordinates.
+
+        Parameters
+        ----------
+        lon : float
+            Longitude.
+        lat : float
+            Latitude.
+        object_id : str
+            ID of the line.
+        start_page : int, optional
+            Starting page number (default is 0).
+        count : int, optional
+            Number of items per page (default is 25).
+        depth : int, optional
+            Search depth (default is 1).
+        odt : str, optional
+            ODT type filter (default is "all").
+        distance : int, optional
+            Maximum search distance (default is 200).
+        headsign : Optional[str], optional
+            Line headsign.
+
+        Returns
+        -------
+        Tuple[Sequence[Line], Pagination]
+            List of Line instances and pagination information.
+        """
+        filters = {
+            "start_page": start_page,
+            "count": count,
+            "depth": depth,
+            "odt": odt,
+            "distance": distance,
+        }
+
+        if headsign is not None:
+            filters["headsign"] = headsign
+
+        url = f"{self.base_navitia_url}/coverage/{lon};{lat}/{self.entity_name}/{object_id}"
+        return self._get_entity_results(url, self.entity_name, filters)

navitia_client/client/apis/public_transportation_apis/network_apis.py

@@ -0,0 +1,303 @@
+from typing import Any, Optional, Sequence, Tuple
+from navitia_client.client.apis.api_base_client import ApiBaseClient
+from navitia_client.client.apis.public_transportation_apis.entity_apis import EntityApi
+from navitia_client.entities.network import Network
+from navitia_client.entities.pagination import Pagination
+
+
+class NetworkApiClient(ApiBaseClient, EntityApi[Network]):
+    """
+    API client for handling 'Network' entities in the Navitia API.
+
+    See https://doc.navitia.io/#pt-ref
+
+    Attributes
+    ----------
+    entity_name : str
+        Name of the entity ('networks').
+    get_navitia_api : method
+        Method to get the Navitia API.
+
+    Methods
+    -------
+    _get_entity_from_response(raw_entity_response: Any) -> Sequence[Network]:
+        Static method to extract Network instances from the raw API response.
+
+    list_entity_collection_from_region(
+        region_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None
+    ) -> Tuple[Sequence[Network], Pagination]:
+        List networks for a given region.
+
+    get_entity_by_id(
+        region_id: str,
+        object_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None
+    ) -> Tuple[Sequence[Network], Pagination]:
+        Get a network by its ID in a given region.
+
+    list_entity_collection_from_coordinates(
+        lon: float,
+        lat: float,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None
+    ) -> Tuple[Sequence[Network], Pagination]:
+        List networks for given geographic coordinates.
+
+    get_entity_by_id_and_coordinates(
+        lon: float,
+        lat: float,
+        object_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None
+    ) -> Tuple[Sequence[Network], Pagination]:
+        Get a network by its ID for given geographic coordinates.
+    """
+
+    entity_name: str = "networks"
+    get_navitia_api = ApiBaseClient.get_navitia_api
+
+    @staticmethod
+    def _get_entity_from_response(raw_entity_response: Any) -> Sequence[Network]:
+        """
+        Static method to extract Network instances from the raw API response.
+
+        Parameters
+        ----------
+        raw_entity_response : Any
+            Raw API response containing Network data.
+
+        Returns
+        -------
+        Sequence[Network]
+            List of Network instances.
+        """
+        entities = []
+        for entity in raw_entity_response:
+            entities.append(Network.from_payload(entity))
+        return entities
+
+    def list_entity_collection_from_region(
+        self,
+        region_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None,
+    ) -> Tuple[Sequence[Network], Pagination]:
+        """
+        List networks for a given region.
+
+        Parameters
+        ----------
+        region_id : str
+            ID of the region.
+        start_page : int, optional
+            Starting page number (default is 0).
+        count : int, optional
+            Number of items per page (default is 25).
+        depth : int, optional
+            Search depth (default is 1).
+        odt : str, optional
+            ODT type filter (default is "all").
+        distance : int, optional
+            Maximum search distance (default is 200).
+        headsign : Optional[str], optional
+            Network headsign.
+
+        Returns
+        -------
+        Tuple[Sequence[Network], Pagination]
+            List of Network instances and pagination information.
+        """
+        filters = {
+            "start_page": start_page,
+            "count": count,
+            "depth": depth,
+            "odt": odt,
+            "distance": distance,
+        }
+
+        if headsign is not None:
+            filters["headsign"] = headsign
+        url = f"{self.base_navitia_url}/coverage/{region_id}/{self.entity_name}"
+        return self._get_entity_results(url, self.entity_name, filters)
+
+    def get_entity_by_id(
+        self,
+        region_id: str,
+        object_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None,
+    ) -> Tuple[Sequence[Network], Pagination]:
+        """
+        Get a network by its ID in a given region.
+
+        Parameters
+        ----------
+        region_id : str
+            ID of the region.
+        object_id : str
+            ID of the network.
+        start_page : int, optional
+            Starting page number (default is 0).
+        count : int, optional
+            Number of items per page (default is 25).
+        depth : int, optional
+            Search depth (default is 1).
+        odt : str, optional
+            ODT type filter (default is "all").
+        distance : int, optional
+            Maximum search distance (default is 200).
+        headsign : Optional[str], optional
+            Network headsign.
+
+        Returns
+        -------
+        Tuple[Sequence[Network], Pagination]
+            List of Network instances and pagination information.
+        """
+        filters = {
+            "start_page": start_page,
+            "count": count,
+            "depth": depth,
+            "odt": odt,
+            "distance": distance,
+        }
+
+        if headsign is not None:
+            filters["headsign"] = headsign
+
+        url = f"{self.base_navitia_url}/coverage/{region_id}/{self.entity_name}/{object_id}"
+        return self._get_entity_results(url, self.entity_name, filters)
+
+    def list_entity_collection_from_coordinates(
+        self,
+        lon: float,
+        lat: float,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None,
+    ) -> Tuple[Sequence[Network], Pagination]:
+        """
+        List networks for given geographic coordinates.
+
+        Parameters
+        ----------
+        lon : float
+            Longitude.
+        lat : float
+            Latitude.
+        start_page : int, optional
+            Starting page number (default is 0).
+        count : int, optional
+            Number of items per page (default is 25).
+        depth : int, optional
+            Search depth (default is 1).
+        odt : str, optional
+            ODT type filter (default is "all").
+        distance : int, optional
+            Maximum search distance (default is 200).
+        headsign : Optional[str], optional
+            Network headsign.
+
+        Returns
+        -------
+        Tuple[Sequence[Network], Pagination]
+            List of Network instances and pagination information.
+        """
+        filters = {
+            "start_page": start_page,
+            "count": count,
+            "depth": depth,
+            "odt": odt,
+            "distance": distance,
+        }
+
+        if headsign is not None:
+            filters["headsign"] = headsign
+
+        url = f"{self.base_navitia_url}/coverage/{lon};{lat}/{self.entity_name}"
+        return self._get_entity_results(url, self.entity_name, filters)
+
+    def get_entity_by_id_and_coordinates(
+        self,
+        lon: float,
+        lat: float,
+        object_id: str,
+        start_page: int = 0,
+        count: int = 25,
+        depth: int = 1,
+        odt: str = "all",
+        distance: int = 200,
+        headsign: Optional[str] = None,
+    ) -> Tuple[Sequence[Network], Pagination]:
+        """
+        Get a network by its ID for given geographic coordinates.
+
+        Parameters
+        ----------
+        lon : float
+            Longitude.
+        lat : float
+            Latitude.
+        object_id : str
+            ID of the network.
+        start_page : int, optional
+            Starting page number (default is 0).
+        count : int, optional
+            Number of items per page (default is 25).
+        depth : int, optional
+            Search depth (default is 1).
+        odt : str, optional
+            ODT type filter (default is "all").
+        distance : int, optional
+            Maximum search distance (default is 200).
+        headsign : Optional[str], optional
+            Network headsign.
+
+        Returns
+        -------
+        Tuple[Sequence[Network], Pagination]
+            List of Network instances and pagination information.
+        """
+        filters = {
+            "start_page": start_page,
+            "count": count,
+            "depth": depth,
+            "odt": odt,
+            "distance": distance,
+        }
+
+        if headsign is not None:
+            filters["headsign"] = headsign
+
+        url = f"{self.base_navitia_url}/coverage/{lon};{lat}/{self.entity_name}/{object_id}"
+        return self._get_entity_results(url, self.entity_name, filters)