SOY NV AI
commited on
Commit
ยท
a281a37
1
Parent(s):
640d5f8
Fix: Redirect to login page in 404 handler for unauthenticated users
Browse files- app/__init__.py +5 -1
app/__init__.py
CHANGED
|
@@ -91,8 +91,12 @@ def create_app() -> Flask:
|
|
| 91 |
# API ์์ฒญ์ธ ๊ฒฝ์ฐ JSON ์๋ต
|
| 92 |
if request.path.startswith('/api/'):
|
| 93 |
return jsonify({'error': '๋ฆฌ์์ค๋ฅผ ์ฐพ์ ์ ์์ต๋๋ค.', 'path': request.path}), 404
|
| 94 |
-
# ์ผ๋ฐ ์์ฒญ์ธ ๊ฒฝ์ฐ ๋ฆฌ๋ค์ด๋ ํธ
|
| 95 |
from flask import redirect, url_for
|
|
|
|
|
|
|
|
|
|
|
|
|
| 96 |
return redirect(url_for('main.index'))
|
| 97 |
|
| 98 |
# ์์ฒญ ๋ก๊น
๋ฏธ๋ค์จ์ด ์ถ๊ฐ
|
|
|
|
| 91 |
# API ์์ฒญ์ธ ๊ฒฝ์ฐ JSON ์๋ต
|
| 92 |
if request.path.startswith('/api/'):
|
| 93 |
return jsonify({'error': '๋ฆฌ์์ค๋ฅผ ์ฐพ์ ์ ์์ต๋๋ค.', 'path': request.path}), 404
|
| 94 |
+
# ์ผ๋ฐ ์์ฒญ์ธ ๊ฒฝ์ฐ ๋ก๊ทธ์ธ ํ์ด์ง๋ก ๋ฆฌ๋ค์ด๋ ํธ (๋ก๊ทธ์ธ๋์ง ์์ ๊ฒฝ์ฐ)
|
| 95 |
from flask import redirect, url_for
|
| 96 |
+
from flask_login import current_user
|
| 97 |
+
if not current_user.is_authenticated:
|
| 98 |
+
return redirect(url_for('main.login', next=request.path))
|
| 99 |
+
# ๋ก๊ทธ์ธ๋ ๊ฒฝ์ฐ ๋ฉ์ธ ํ์ด์ง๋ก ๋ฆฌ๋ค์ด๋ ํธ
|
| 100 |
return redirect(url_for('main.index'))
|
| 101 |
|
| 102 |
# ์์ฒญ ๋ก๊น
๋ฏธ๋ค์จ์ด ์ถ๊ฐ
|