Echo-ai commited on
Commit
346dbbb
·
verified ·
1 Parent(s): 29bd4ca

Update routes.py

Browse files
Files changed (1) hide show
  1. routes.py +4 -18
routes.py CHANGED
@@ -93,8 +93,9 @@ def login():
93
  username = request.form.get('username')
94
  password = request.form.get('password')
95
 
96
- user = db.get_user(username)
97
- if user and check_password_hash(user.password, password):
 
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'))