SOY NV AI commited on
Commit
a281a37
ยท
1 Parent(s): 640d5f8

Fix: Redirect to login page in 404 handler for unauthenticated users

Browse files
Files changed (1) hide show
  1. 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
  # ์š”์ฒญ ๋กœ๊น… ๋ฏธ๋“ค์›จ์–ด ์ถ”๊ฐ€