new zip code class version 2

Author: Mohamed Ben Makhlouf

pyzipcodeapi/api.py

@@ -6,7 +6,7 @@
 
 import requests
 
-from pyzipcodeapi.dataclass import Distance, Error
+from pyzipcodeapi.dataclass import Distance, Error, MultiDistance
 from pyzipcodeapi.enums import FormatEnum, UnitEnum, CountryEnum
 from pyzipcodeapi.options import OPTIONS
 
@@ -104,13 +104,13 @@ def __init__(self, api_key: str):
         self.api_key = api_key
         self.con = HTTPSConnection(host=self.host)
 
-    def _make_api_call(
+    def _api_call(
         self,
         option: str,
         f: FormatEnum,
         path: str,
-        country: CountryEnum = CountryEnum.US,
         data_class: type | None = None,
+        country: CountryEnum = CountryEnum.US,
     ) -> DictReader | bytes | type | Element | Error:
         base_url = f"rest/v2/CA" if country == CountryEnum.CA else f"rest"
         self.con.request(
@@ -140,10 +140,19 @@ def distance(
         country: CountryEnum = CountryEnum.US,
     ) -> Distance | DictReader | Element:
         """distance.<format>/<zip_code1>/<zip_code2>/<units>"""
-        return self._make_api_call(
-            "distance",
-            f,
-            path=f"{zip_code1}/{zip_code2}/{units}",
-            country=country,
-            data_class=Distance,
-        )
+        path = f"{zip_code1}/{zip_code2}/{units}"
+        dc = Distance
+        return self._api_call("distance", f, path, dc, country=country)
+
+    def multi_distance(
+        self,
+        zip_code: str,
+        zip_codes: list[str],
+        units: UnitEnum = UnitEnum.KM,
+        f: FormatEnum | None = FormatEnum.JSON,
+    ) -> MultiDistance | DictReader | Element:
+        """multi-distance.<format>/<zip_code>/<other_zip_codes>/<units>"""
+        assert len(zip_codes) <= 100
+        path = f"{zip_code}/{','.join(zip_codes)}/{units}"
+        dc = MultiDistance
+        return self._api_call("multi-distance", f, path, dc)

pyzipcodeapi/dataclass.py

@@ -10,3 +10,8 @@ class Error:
 @dataclass
 class Distance:
     distance: float
+
+
+@dataclass
+class MultiDistance:
+    distances: dict[str, float]