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, MultiDistance
+from pyzipcodeapi.dataclass import Distance, Error, MultiDistance, Radius
 from pyzipcodeapi.enums import FormatEnum, UnitEnum, CountryEnum
 from pyzipcodeapi.options import OPTIONS
 
@@ -156,3 +156,19 @@ def multi_distance(
         path = f"{zip_code}/{','.join(zip_codes)}/{units}"
         dc = MultiDistance
         return self._api_call("multi-distance", f, path, dc)
+
+    def radius(
+        self,
+        zip_code: str,
+        distance: int,
+        minimal: bool = False,
+        units: UnitEnum = UnitEnum.KM,
+        f: FormatEnum | None = FormatEnum.JSON,
+        country: CountryEnum = CountryEnum.US,
+    ) -> Radius | DictReader | Element:
+        """radius.<format>/<zip_code>/<distance>/<units>"""
+        path = f"{zip_code}/{distance}/{units}"
+        dc = Radius
+        if minimal:
+            path = f"{path}?minimal"
+        return self._api_call("radius", f, path, dc, country=country)

pyzipcodeapi/dataclass.py

@@ -15,3 +15,22 @@ class Distance:
 @dataclass
 class MultiDistance:
     distances: dict[str, float]
+
+
+@dataclass
+class RadiusInfo:
+    zip_code: str
+    distance: float
+    city: str
+    state: str
+
+
+@dataclass
+class Radius:
+    zip_codes: list[str | dict]
+
+    @property
+    def info(self) -> list[RadiusInfo | str]:
+        if self.zip_codes and type(self.zip_codes[0] == dict):
+            return [RadiusInfo(**zp) for zp in self.zip_codes]
+        return self.zip_codes