first commit

pytorrent/services/geoip.py

@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+from functools import lru_cache
+from pathlib import Path
+from ..config import GEOIP_DB
+
+try:
+    import geoip2.database
+except Exception:  # pragma: no cover
+    geoip2 = None
+
+_reader = None
+
+
+def _get_reader():
+    global _reader
+    if _reader is not None:
+        return _reader
+    if not GEOIP_DB.exists() or geoip2 is None:
+        return None
+    _reader = geoip2.database.Reader(str(GEOIP_DB))
+    return _reader
+
+
+@lru_cache(maxsize=50000)
+def lookup_ip(ip: str) -> dict:
+    reader = _get_reader()
+    if not reader:
+        return {"country_iso": "", "country": "", "city": ""}
+    try:
+        hit = reader.city(ip)
+        return {
+            "country_iso": (hit.country.iso_code or "").lower(),
+            "country": hit.country.name or "",
+            "city": hit.city.name or "",
+        }
+    except Exception:
+        return {"country_iso": "", "country": "", "city": ""}