Update app.py

app.py

@@ -4,6 +4,11 @@ import geoip2.database
 import os
 from pathlib import Path
 
+GOOGLE_SHEET_WEBHOOK = os.getenv("GOOGLE_SHEET_WEBHOOK")
+
+if not GOOGLE_SHEET_WEBHOOK:
+    raise RuntimeError("GOOGLE_SHEET_WEBHOOK environment variable not set")
+    
 app = FastAPI()
 
 # Paths to your downloaded databases (adjust if needed)
@@ -17,16 +22,16 @@ asn_reader  = geoip2.database.Reader(str(ASN_DB_PATH))
 
 @app.get("/track")
 async def track_ip(request: Request):
-    # Get client IP (handle proxies)
+
     client_ip = request.headers.get("X-Forwarded-For")
     if client_ip:
         client_ip = client_ip.split(",")[0].strip()
     else:
         client_ip = request.client.host
 
-    # Browser / Device info
     ua_string = request.headers.get("User-Agent", "Unknown")
     ua = parse(ua_string)
+
     browser_info = {
         "browser": ua.browser.family,
         "version": ua.browser.version_string,
@@ -37,47 +42,47 @@ async def track_ip(request: Request):
         "is_bot": ua.is_bot
     }
 
-    # Geolocation lookup (offline!)
-    geo_data = {"error": "IP not found"}
+    geo_data = {}
     asn_data = {}
 
     try:
-        city_response = city_reader.city(client_ip)
+        city = city_reader.city(client_ip)
         geo_data = {
-            "country": city_response.country.name,
-            "country_code": city_response.country.iso_code,
-            "city": city_response.city.name,
-            "region": city_response.subdivisions.most_specific.name,
-            "region_code": city_response.subdivisions.most_specific.iso_code,
-            "postal": city_response.postal.code,
+            "country": city.country.name,
+            "city": city.city.name,
+            "region": city.subdivisions.most_specific.name,
             "location": {
-                "latitude": city_response.location.latitude,
-                "longitude": city_response.location.longitude,
-                "accuracy_radius": city_response.location.accuracy_radius,
-                "timezone": city_response.location.time_zone
+                "latitude": city.location.latitude,
+                "longitude": city.location.longitude
             }
         }
-    except geoip2.errors.AddressNotFoundError:
-        pass
-    except Exception as e:
-        geo_data = {"error": str(e)}
+    except:
+        geo_data = {}
 
     try:
-        asn_response = asn_reader.asn(client_ip)
+        asn = asn_reader.asn(client_ip)
         asn_data = {
-            "asn": asn_response.autonomous_system_number,
-            "org": asn_response.autonomous_system_organization
+            "asn": asn.autonomous_system_number,
+            "org": asn.autonomous_system_organization
         }
     except:
-        pass
+        asn_data = {}
 
-    return {
+    payload = {
         "ip": client_ip,
         "browser": browser_info,
         "geolocation": geo_data,
         "asn": asn_data
     }
 
+    async with httpx.AsyncClient(timeout=5) as client:
+        try:
+            await client.post(GOOGLE_SHEET_WEBHOOK, json=payload)
+        except Exception as e:
+            print("Sheet error:", e)
+
+    return payload
+
 # Optional: close readers on shutdown (good practice)
 @app.on_event("shutdown")
 def close_readers():