feat: add IP geolocation tracking and include location in prompts

- Add GeoLite2-City.mmdb database for IP lookup - Create geoip.py module for IP location services - Extract client IP from requests and log location info - Pass location context to LLM prompts for enhanced responses
2026-02-18 08:59:28 +08:00
parent 2b79f20e19
commit d2b64ad5d6
7 changed files with 181 additions and 13 deletions
@@ -0,0 +1,80 @@
+"""
+GeoIP2 IP归属地查询测试脚本
+
+使用方法：
+1. 安装依赖：pip install geoip2
+2. 下载数据库：https://dev.maxmind.com/geoip/geoip2/geolite2/
+3. 运行测试：python test_geoip.py
+"""
+
+import os
+import sys
+
+try:
+    import geoip2.database
+except ImportError:
+    print("请先安装 geoip2: pip install geoip2")
+    sys.exit(1)
+
+DB_PATH = os.path.join(os.path.dirname(__file__), "GeoLite2-City.mmdb")
+
+TEST_IPS = [
+    "8.8.8.8",           # Google DNS (美国)
+    "114.114.114.114",   # 114 DNS (中国南京)
+    "223.5.5.5",         # 阿里DNS (中国杭州)
+    "1.1.1.1",           # Cloudflare DNS (澳大利亚)
+    "119.29.29.29",      # 腾讯DNS (中国)
+]
+
+
+def get_location(reader, ip: str) -> dict:
+    try:
+        response = reader.city(ip)
+        return {
+            "ip": ip,
+            "country": response.country.name,
+            "country_code": response.country.iso_code,
+            "region": response.subdivisions.most_specific.name if response.subdivisions else None,
+            "city": response.city.name,
+            "latitude": response.location.latitude,
+            "longitude": response.location.longitude,
+            "timezone": response.location.time_zone,
+        }
+    except geoip2.errors.AddressNotFoundError:
+        return {"ip": ip, "error": "IP未在数据库中找到"}
+    except Exception as e:
+        return {"ip": ip, "error": str(e)}
+
+
+def main():
+    if not os.path.exists(DB_PATH):
+        print(f"数据库文件不存在: {DB_PATH}")
+        print("请从 https://dev.maxmind.com/geoip/geoip2/geolite2/ 下载 GeoLite2-City.mmdb")
+        return
+    
+    print(f"加载数据库: {DB_PATH}")
+    reader = geoip2.database.Reader(DB_PATH)
+    
+    print("\n" + "=" * 60)
+    print("IP归属地查询测试")
+    print("=" * 60)
+    
+    for ip in TEST_IPS:
+        result = get_location(reader, ip)
+        if "error" in result:
+            print(f"\n{ip}: {result['error']}")
+        else:
+            print(f"\n{ip}:")
+            print(f"  国家: {result['country']} ({result['country_code']})")
+            print(f"  地区: {result['region'] or '未知'}")
+            print(f"  城市: {result['city'] or '未知'}")
+            print(f"  坐标: {result['latitude']}, {result['longitude']}")
+            print(f"  时区: {result['timezone']}")
+    
+    reader.close()
+    print("\n" + "=" * 60)
+    print("测试完成")
+
+
+if __name__ == "__main__":
+    main()