Add post method and let the get method reuse that code

app/main.py

@@ -1,12 +1,11 @@
 '''
 Simple Geolocation with FastAPI
 '''
-import dataclasses
 from ipaddress import IPv4Address, IPv4Network, IPv6Address, IPv6Network
 from typing import Annotated, Optional, Union
 
 import geoip2.database
-from fastapi import FastAPI, Path
+from fastapi import FastAPI, Path, Body
 from pydantic import BaseModel
 
 app = FastAPI()
@@ -14,6 +13,9 @@ app = FastAPI()
 GEOLITE2_ASN_DB = '/usr/local/share/GeoIP/GeoLite2-ASN.mmdb'
 GEOLITE2_CITY_DB = '/usr/local/share/GeoIP/GeoLite2-City.mmdb'
 
+class IPAddressParam(BaseModel):
+    ip: Union[IPv6Address,IPv4Address]
+
 class Locality(BaseModel):
     '''
     Locality data
@@ -33,27 +35,36 @@ class GeoLocation(BaseModel):
     network: Union[IPv6Network,IPv4Network,None]
     locality: Locality
 
+@app.post("/")
+async def root_post(ipaddresses: Annotated[list[IPAddressParam],
+                                    Body(title="The IPAddresses to geolocate")]
+               ) -> list[GeoLocation]:
+    geolocations = []
+    with (geoip2.database.Reader(GEOLITE2_ASN_DB) as reader_asn,
+            geoip2.database.Reader(GEOLITE2_CITY_DB) as reader_city):
+        for ipaddress in ipaddresses:
+            asn_data = reader_asn.asn(ipaddress.ip)
+            city_data = reader_city.city(ipaddress.ip)
+            geolocations.append(GeoLocation(
+                ip=ipaddress.ip,
+                asn=asn_data.autonomous_system_number,
+                asn_org=asn_data.autonomous_system_organization,
+                network=asn_data.network,
+                locality=Locality(
+                    city=city_data.city.name,
+                    country=city_data.country.iso_code,
+                    continent=city_data.continent.code,
+                    is_eu=city_data.country.is_in_european_union
+                )
+            ))
+    return geolocations
+
+
 @app.get("/{ipaddress}")
-async def root(ipaddress: Annotated[Union[IPv4Address,IPv6Address],
+async def root_get(ipaddress: Annotated[Union[IPv4Address,IPv6Address],
                                     Path(title="The IPAddress to geolocate")]
                ) -> GeoLocation:
     '''
     Look up geolocation for ip in path parameter
     '''
-    with (geoip2.database.Reader(GEOLITE2_ASN_DB) as reader_asn,
+    return (await root_post([IPAddressParam(ip=ipaddress)])).pop()
-            geoip2.database.Reader(GEOLITE2_CITY_DB) as reader_city):
-        asn_data = reader_asn.asn(ipaddress)
-        city_data = reader_city.city(ipaddress)
-
-        return GeoLocation(
-            ip=ipaddress,
-            asn=asn_data.autonomous_system_number,
-            asn_org=asn_data.autonomous_system_organization,
-            network=asn_data.network,
-            locality=Locality(
-                city=city_data.city.name,
-                country=city_data.country.iso_code,
-                continent=city_data.continent.code,
-                is_eu=city_data.country.is_in_european_union
-            )
-        )