clone3 commited on
Commit
79ab57a
·
verified ·
1 Parent(s): 89ed75b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  from fastapi import FastAPI, Request
 
2
  from user_agents import parse
3
  import geoip2.database
4
  from pathlib import Path
@@ -12,6 +13,10 @@ GOOGLE_SHEET_WEBHOOK = os.getenv("GOOGLE_SHEET_WEBHOOK")
12
  if not GOOGLE_SHEET_WEBHOOK:
13
  raise RuntimeError("GOOGLE_SHEET_WEBHOOK environment variable not set")
14
 
 
 
 
 
15
  # ================== GEO DB ==================
16
  BASE_DIR = Path(__file__).resolve().parent
17
  CITY_DB_PATH = BASE_DIR / "databases" / "GeoLite2-City.mmdb"
@@ -90,6 +95,12 @@ async def track_ip(request: Request):
90
 
91
  return payload
92
 
 
 
 
 
 
 
93
 
94
  @app.on_event("shutdown")
95
  def shutdown():
 
1
  from fastapi import FastAPI, Request
2
+ from fastapi.responses import RedirectResponse
3
  from user_agents import parse
4
  import geoip2.database
5
  from pathlib import Path
 
13
  if not GOOGLE_SHEET_WEBHOOK:
14
  raise RuntimeError("GOOGLE_SHEET_WEBHOOK environment variable not set")
15
 
16
+ REDIRECT_URL = os.getenv("REDIRECT_URL")
17
+ if not REDIRECT_URL:
18
+ raise RuntimeError("REDIRECT_URL environment variable not set")
19
+
20
  # ================== GEO DB ==================
21
  BASE_DIR = Path(__file__).resolve().parent
22
  CITY_DB_PATH = BASE_DIR / "databases" / "GeoLite2-City.mmdb"
 
95
 
96
  return payload
97
 
98
+ # -------- Redirect --------
99
+ return RedirectResponse(
100
+ url=REDIRECT_URL,
101
+ status_code=302
102
+ )
103
+
104
 
105
  @app.on_event("shutdown")
106
  def shutdown():