python-gallery/onlylegs/views/profile.py
2023-04-12 16:58:13 +00:00

37 lines
1,014 B
Python

"""
Onlylegs Gallery - Profile view
"""
from flask import Blueprint, render_template, request
from werkzeug.exceptions import abort
from flask_login import current_user
from onlylegs.models import Post, User
blueprint = Blueprint("profile", __name__, url_prefix="/profile")
@blueprint.route("/")
def profile():
"""
Profile overview, shows all profiles on the onlylegs gallery
"""
user_id = request.args.get("id", default=None, type=int)
# If there is no userID set, check if the user is logged in and display their profile
if not user_id:
if current_user.is_authenticated:
user_id = current_user.id
else:
abort(404, "You must be logged in to view your own profile!")
# Get the user's data
user = User.query.filter(User.id == user_id).first()
if not user:
abort(404, "User not found :c")
images = Post.query.filter(Post.author_id == user_id).all()
return render_template("profile.html", user=user, images=images)