Spaces:
Sleeping
Sleeping
Echo-ai
commited on
Update routes.py
Browse files
routes.py
CHANGED
@@ -93,8 +93,9 @@ def login():
|
|
93 |
username = request.form.get('username')
|
94 |
password = request.form.get('password')
|
95 |
|
96 |
-
|
97 |
-
if
|
|
|
98 |
login_user(user)
|
99 |
flash('Logged in successfully.')
|
100 |
return redirect(url_for('admin.index'))
|
@@ -106,19 +107,4 @@ def login():
|
|
106 |
def logout():
|
107 |
logout_user()
|
108 |
flash('Logged out successfully.')
|
109 |
-
return redirect(url_for('main.home'))
|
110 |
-
|
111 |
-
@main.route('/create-admin')
|
112 |
-
def create_admin():
|
113 |
-
if db.get_user('admin'):
|
114 |
-
flash('Admin user already exists')
|
115 |
-
return redirect(url_for('main.home'))
|
116 |
-
|
117 |
-
admin = {
|
118 |
-
'username': os.getenv("USERNAME"),
|
119 |
-
'password': generate_password_hash(os.getenv("PASSWORD")),
|
120 |
-
'is_admin': True
|
121 |
-
}
|
122 |
-
db.add_user(admin)
|
123 |
-
flash('Admin user created successfully')
|
124 |
-
return redirect(url_for('main.home'))
|
|
|
93 |
username = request.form.get('username')
|
94 |
password = request.form.get('password')
|
95 |
|
96 |
+
# Check against environment variables
|
97 |
+
if username == os.getenv('ADMIN_USERNAME') and password == os.getenv('ADMIN_PASSWORD'):
|
98 |
+
user = db.get_user(username)
|
99 |
login_user(user)
|
100 |
flash('Logged in successfully.')
|
101 |
return redirect(url_for('admin.index'))
|
|
|
107 |
def logout():
|
108 |
logout_user()
|
109 |
flash('Logged out successfully.')
|
110 |
+
return redirect(url_for('main.home'))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|