From a080a2c5ee529aa8066e4314b7313cced7eb9c86 Mon Sep 17 00:00:00 2001 From: midoks Date: Sun, 19 Jun 2022 23:06:42 +0800 Subject: [PATCH] Update __init__.py --- route/__init__.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/route/__init__.py b/route/__init__.py index 7dacc3f05..d5a032720 100755 --- a/route/__init__.py +++ b/route/__init__.py @@ -246,13 +246,16 @@ def get_admin_safe(): return (False, '') -def admin_safe_path(path, req, data): +def admin_safe_path(path, req, data, pageFile): if path != req and not isLogined(): return render_template('path.html') if not isLogined(): return render_template('login.html', data=data) + if not req in pageFile: + return redirect('/') + return render_template(req + '.html', data=data) @@ -289,12 +292,12 @@ def index(reqClass=None, reqAction=None, reqData=None): session['overdue'] = 0 if ainfo[0]: - return admin_safe_path(ainfo[1], reqClass, data) + return admin_safe_path(ainfo[1], reqClass, data, pageFile) return render_template('login.html', data=data) if ainfo[0]: - return admin_safe_path(ainfo[1], reqClass, data) + return admin_safe_path(ainfo[1], reqClass, data, pageFile) if not reqClass in pageFile: return redirect('/')