From 0b6dafec62f22383bf501e1594d3b9a8c93f2337 Mon Sep 17 00:00:00 2001 From: Parker Date: Sun, 3 Nov 2024 21:29:10 -0600 Subject: Fix link redirects and remove BASE_URL --- app/main.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'app/main.py') diff --git a/app/main.py b/app/main.py index 266c589..78a65c8 100644 --- a/app/main.py +++ b/app/main.py @@ -15,7 +15,6 @@ import random from models import User, Link from database import * from app.util.log import log -from config import BASE_URL class FlaskUser(UserMixin): @@ -158,14 +157,14 @@ def log_redirect(link: str): link = link.upper() # If `link` is not exactly 5 characters, return redirect to base url if len(link) != 5: - return redirect(BASE_URL) + return redirect(url_for("login")) # Make sure the link exists in the database db = SessionLocal() link_record = db.query(Link).filter(Link.link == link).first() if not link_record: db.close() - return redirect(BASE_URL) + return redirect(url_for("login")) else: # Log the visit if request.headers.get("X-Real-IP"): -- cgit v1.2.3-70-g09d2