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 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):
|
class LoginForm(FlaskForm):
|
||||||
username = StringField("Username", validators=[validators.DataRequired()])
|
username = StringField("Username", validators=[DataRequired()])
|
||||||
password = PasswordField("Password")
|
password = PasswordField("Password")
|
||||||
submit = SubmitField("Login")
|
submit = SubmitField("Login")
|
||||||
|
|||||||
@@ -16,6 +16,6 @@ def index():
|
|||||||
|
|
||||||
|
|
||||||
@dash_blueprint.route("/admin", methods=["GET", "POST"])
|
@dash_blueprint.route("/admin", methods=["GET", "POST"])
|
||||||
# @admin_required
|
@admin_required
|
||||||
def admin():
|
def admin():
|
||||||
return render_template("admin.html")
|
return render_template("admin.html")
|
||||||
|
|||||||
@@ -30,7 +30,7 @@
|
|||||||
<li class="nav-item">
|
<li class="nav-item">
|
||||||
<a class="nav-link" href="#">Info</a>
|
<a class="nav-link" href="#">Info</a>
|
||||||
</li>
|
</li>
|
||||||
{% if current_user.is_authenticated %}
|
{% if current_user.is_admin %}
|
||||||
<li class="nav-item">
|
<li class="nav-item">
|
||||||
<a class="nav-link" href="{{url_for('dash.admin')}}">Add user</a>
|
<a class="nav-link" href="{{url_for('dash.admin')}}">Add user</a>
|
||||||
</li>
|
</li>
|
||||||
|
|||||||
18
seed.py
18
seed.py
@@ -11,9 +11,19 @@ new_strikers = [
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
new_user = User(
|
new_users = [
|
||||||
username="admin", password=generate_password_hash("admin"), is_admin=True
|
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")
|
new_services = Service(name="test123", url="http://google.com")
|
||||||
|
|
||||||
with app.app_context():
|
with app.app_context():
|
||||||
@@ -23,5 +33,5 @@ with app.app_context():
|
|||||||
db.session.commit()
|
db.session.commit()
|
||||||
# Then add new
|
# Then add new
|
||||||
db.session.add(new_services)
|
db.session.add(new_services)
|
||||||
db.session.add(new_user)
|
db.session.add_all(new_users)
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|||||||
Reference in New Issue
Block a user