mirror of
https://github.com/StefBuwalda/WebTech.git
synced 2025-10-30 11:19:58 +00:00
Fixed is_admin privileges
This commit is contained in:
@@ -1,8 +1,9 @@
|
||||
from flask_wtf import FlaskForm # type: ignore
|
||||
from wtforms import StringField, SubmitField, PasswordField, validators
|
||||
from wtforms import StringField, SubmitField, PasswordField
|
||||
from wtforms.validators import DataRequired
|
||||
|
||||
|
||||
class LoginForm(FlaskForm):
|
||||
username = StringField("Username", validators=[validators.DataRequired()])
|
||||
username = StringField("Username", validators=[DataRequired()])
|
||||
password = PasswordField("Password")
|
||||
submit = SubmitField("Login")
|
||||
|
||||
@@ -16,6 +16,6 @@ def index():
|
||||
|
||||
|
||||
@dash_blueprint.route("/admin", methods=["GET", "POST"])
|
||||
# @admin_required
|
||||
@admin_required
|
||||
def admin():
|
||||
return render_template("admin.html")
|
||||
|
||||
@@ -30,7 +30,7 @@
|
||||
<li class="nav-item">
|
||||
<a class="nav-link" href="#">Info</a>
|
||||
</li>
|
||||
{% if current_user.is_authenticated %}
|
||||
{% if current_user.is_admin %}
|
||||
<li class="nav-item">
|
||||
<a class="nav-link" href="{{url_for('dash.admin')}}">Add user</a>
|
||||
</li>
|
||||
|
||||
18
seed.py
18
seed.py
@@ -11,9 +11,19 @@ new_strikers = [
|
||||
|
||||
"""
|
||||
|
||||
new_user = User(
|
||||
username="admin", password=generate_password_hash("admin"), is_admin=True
|
||||
)
|
||||
new_users = [
|
||||
User(
|
||||
username="admin",
|
||||
password=generate_password_hash("admin"),
|
||||
is_admin=True,
|
||||
),
|
||||
User(
|
||||
username="test",
|
||||
password=generate_password_hash("test123"),
|
||||
is_admin=False,
|
||||
),
|
||||
]
|
||||
|
||||
new_services = Service(name="test123", url="http://google.com")
|
||||
|
||||
with app.app_context():
|
||||
@@ -23,5 +33,5 @@ with app.app_context():
|
||||
db.session.commit()
|
||||
# Then add new
|
||||
db.session.add(new_services)
|
||||
db.session.add(new_user)
|
||||
db.session.add_all(new_users)
|
||||
db.session.commit()
|
||||
|
||||
Reference in New Issue
Block a user