chore: added more type hints

routingpy/routers/google.py

@@ -16,6 +16,7 @@
 #
 
 from operator import itemgetter
+from typing import List, Optional, Tuple, Union
 
 from .. import convert, utils
 from ..client_base import DEFAULT
@@ -71,12 +72,12 @@ class Google:
 
     def __init__(
         self,
-        api_key,
-        user_agent=None,
-        timeout=DEFAULT,
-        retry_timeout=None,
+        api_key: str,
+        user_agent: Optional[str] = None,
+        timeout: Optional[int] = DEFAULT,
+        retry_timeout: Optional[int] = None,
         retry_over_query_limit=True,
-        skip_api_error=None,
+        skip_api_error: Optional[bool] = None,
         client=Client,
         **client_kwargs
     ):
@@ -177,20 +178,20 @@ def make_waypoint(self):
 
     def directions(  # noqa: C901
         self,
-        locations,
-        profile,
-        alternatives=None,
-        avoid=None,
-        optimize=None,
-        language=None,
-        region=None,
-        units=None,
-        arrival_time=None,
-        departure_time=None,
-        traffic_model=None,
-        transit_mode=None,
-        transit_routing_preference=None,
-        dry_run=None,
+        locations: List[List[float]],
+        profile: str,
+        alternatives: Optional[bool] = None,
+        avoid: Optional[List[str]] = None,
+        optimize: Optional[bool] = None,
+        language: Optional[str] = None,
+        region: Optional[str] = None,
+        units: Optional[str] = None,
+        arrival_time: Optional[int] = None,
+        departure_time: Optional[int] = None,
+        traffic_model: Optional[str] = None,
+        transit_mode: Optional[Union[List[str], Tuple[str]]] = None,
+        transit_routing_preference: Optional[str] = None,
+        dry_run: Optional[bool] = None,
     ):
         """Get directions between an origin point and a destination point.
 
@@ -381,20 +382,20 @@ def isochrones(self):  # pragma: no cover
 
     def matrix(  # noqa: C901
         self,
-        locations,
-        profile,
-        sources=None,
-        destinations=None,
-        avoid=None,
-        language=None,
-        region=None,
-        units=None,
-        arrival_time=None,
-        departure_time=None,
-        traffic_model=None,
-        transit_mode=None,
-        transit_routing_preference=None,
-        dry_run=None,
+        locations: List[List[float]],
+        profile: str,
+        sources: Optional[Union[List[int], Tuple[int]]] = None,
+        destinations: Optional[Union[List[int], Tuple[int]]] = None,
+        avoid: Optional[List[str]] = None,
+        language: Optional[str] = None,
+        region: Optional[str] = None,
+        units: Optional[str] = None,
+        arrival_time: Optional[int] = None,
+        departure_time: Optional[int] = None,
+        traffic_model: Optional[str] = None,
+        transit_mode: Optional[Union[List[str], Tuple[str]]] = None,
+        transit_routing_preference: Optional[str] = None,
+        dry_run: Optional[bool] = None,
     ):
         """Gets travel distance and time for a matrix of origins and destinations.
 

routingpy/routers/graphhopper.py

@@ -15,7 +15,7 @@
 # the License.
 #
 
-from typing import List, Tuple  # noqa: F401
+from typing import List, Optional, Tuple, Union  # noqa: F401
 
 from .. import convert, utils
 from ..client_base import DEFAULT
@@ -32,13 +32,13 @@ class Graphhopper:
 
     def __init__(
         self,
-        api_key=None,
-        base_url=_DEFAULT_BASE_URL,
-        user_agent=None,
-        timeout=DEFAULT,
-        retry_timeout=None,
-        retry_over_query_limit=False,
-        skip_api_error=None,
+        api_key: Optional[str] = None,
+        base_url: Optional[str] = _DEFAULT_BASE_URL,
+        user_agent: Optional[str] = None,
+        timeout: Optional[int] = DEFAULT,
+        retry_timeout: Optional[int] = None,
+        retry_over_query_limit: Optional[bool] = False,
+        skip_api_error: Optional[bool] = None,
         client=Client,
         **client_kwargs
     ):
@@ -99,32 +99,32 @@ def __init__(
 
     def directions(  # noqa: C901
         self,
-        locations,
-        profile,
-        format=None,
-        optimize=None,
-        instructions=None,
-        locale=None,
-        elevation=None,
-        points_encoded=True,
-        calc_points=None,
-        debug=None,
-        point_hints=None,
-        details=None,
-        ch_disable=None,
-        custom_model=None,
-        headings=None,
-        heading_penalty=None,
-        pass_through=None,
-        algorithm=None,
-        round_trip_distance=None,
-        round_trip_seed=None,
-        alternative_route_max_paths=None,
-        alternative_route_max_weight_factor=None,
-        alternative_route_max_share_factor=None,
-        dry_run=None,
-        snap_preventions=None,
-        curbsides=None,
+        locations: Union[List[List[float]], Tuple[Tuple[float]]],
+        profile: str,
+        format: Optional[str] = None,
+        optimize: Optional[bool] = None,
+        instructions: Optional[bool] = None,
+        locale: Optional[str] = None,
+        elevation: Optional[bool] = None,
+        points_encoded: Optional[bool] = True,
+        calc_points: Optional[bool] = None,
+        debug: Optional[bool] = None,
+        point_hints: Optional[List[str]] = None,
+        details: Optional[List[str]] = None,
+        ch_disable: Optional[bool] = None,
+        custom_model: Optional[dict] = None,
+        headings: Optional[List[int]] = None,
+        heading_penalty: Optional[int] = None,
+        pass_through: Optional[bool] = None,
+        algorithm: Optional[str] = None,
+        round_trip_distance: Optional[int] = None,
+        round_trip_seed: Optional[int] = None,
+        alternative_route_max_paths: Optional[int] = None,
+        alternative_route_max_weight_factor: Optional[float] = None,
+        alternative_route_max_share_factor: Optional[float] = None,
+        dry_run: Optional[bool] = None,
+        snap_preventions: Optional[List[str]] = None,
+        curbsides: Optional[List[str]] = None,
         **direction_kwargs
     ):
         """Get directions between an origin point and a destination point.
@@ -402,15 +402,15 @@ def parse_directions_json(response, algorithm, elevation, points_encoded):
 
     def isochrones(
         self,
-        locations,
-        profile,
-        intervals,
-        type="json",
-        buckets=1,
-        interval_type="time",
-        reverse_flow=None,
-        debug=None,
-        dry_run=None,
+        locations: Union[Tuple[float], List[float]],
+        profile: str,
+        intervals: Union[List[int], Tuple[int]],
+        type: Optional[str] = "json",
+        buckets: Optional[int] = 1,
+        interval_type: Optional[str] = "time",
+        reverse_flow: Optional[bool] = None,
+        debug: Optional[bool] = None,
+        dry_run: Optional[bool] = None,
         **isochrones_kwargs
     ):
         """Gets isochrones or equidistants for a range of time/distance values around a given set of coordinates.
@@ -516,13 +516,15 @@ def parse_isochrone_json(response, type, max_range, buckets, center, interval_ty
 
     def matrix(
         self,
-        locations,
-        profile,
-        sources=None,
-        destinations=None,
-        out_array=["times", "distances"],
+        locations: Union[
+            List[Union[List[float], Tuple[float]]], Tuple[Union[List[float], Tuple[float]]]
+        ],
+        profile: str,
+        sources: Optional[List[int]] = None,
+        destinations: Optional[List[int]] = None,
+        out_array: Optional[List[str]] = ["times", "distances"],
         debug=None,
-        dry_run=None,
+        dry_run: Optional[bool] = None,
         **matrix_kwargs
     ):
         """Gets travel distance and time for a matrix of origins and destinations.