39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
from flask import Blueprint, render_template, request, redirect, url_for
|
|
from flask_login import login_required, current_user
|
|
from ..extensions import db
|
|
from ..models import User, Profile, Notification, Follow
|
|
|
|
bp = Blueprint("users", __name__, url_prefix="/users")
|
|
|
|
@bp.route("/<int:user_id>")
|
|
def profile(user_id):
|
|
user = User.query.get_or_404(user_id)
|
|
followers_count = Follow.query.filter_by(followee_id=user.id).count()
|
|
following_count = Follow.query.filter_by(follower_id=user.id).count()
|
|
is_following = False
|
|
try:
|
|
from flask_login import current_user
|
|
if current_user.is_authenticated:
|
|
is_following = Follow.query.filter_by(follower_id=current_user.id, followee_id=user.id).first() is not None
|
|
except Exception:
|
|
pass
|
|
return render_template("users/profile.html", user=user, followers_count=followers_count, following_count=following_count, is_following=is_following)
|
|
|
|
@bp.route("/me/edit", methods=["GET", "POST"])
|
|
@login_required
|
|
def edit_profile():
|
|
profile = Profile.query.filter_by(user_id=current_user.id).first()
|
|
if request.method == "POST":
|
|
profile.bio = request.form.get("bio")
|
|
profile.grade = request.form.get("grade")
|
|
profile.class_name = request.form.get("class_name")
|
|
db.session.commit()
|
|
return redirect(url_for("users.profile", user_id=current_user.id))
|
|
return render_template("users/edit.html", profile=profile)
|
|
|
|
@bp.route("/me/notifications")
|
|
@login_required
|
|
def notifications():
|
|
items = Notification.query.filter_by(user_id=current_user.id).order_by(Notification.created_at.desc()).limit(50).all()
|
|
return render_template("users/notifications.html", items=items)
|