94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
from flask import Blueprint, render_template, redirect, url_for, flash
|
|
from flask_login import login_required, current_user
|
|
from models import db, User, Follow, Post
|
|
|
|
users_bp = Blueprint('users', __name__)
|
|
|
|
|
|
@users_bp.route('/user/<username>')
|
|
def profile(username):
|
|
"""用户个人主页"""
|
|
user = User.query.filter_by(username=username).first_or_404()
|
|
|
|
# 只显示已审核的帖子（除非是自己或管理员）
|
|
if current_user.is_authenticated and (current_user.id == user.id or current_user.is_admin):
|
|
posts = user.posts.order_by(Post.created_at.desc()).all()
|
|
else:
|
|
posts = user.posts.filter_by(is_approved=True).order_by(Post.created_at.desc()).all()
|
|
|
|
follower_count = user.get_follower_count()
|
|
following_count = user.get_following_count()
|
|
|
|
is_following = False
|
|
if current_user.is_authenticated:
|
|
is_following = current_user.is_following(user)
|
|
|
|
return render_template('profile.html',
|
|
user=user,
|
|
posts=posts,
|
|
follower_count=follower_count,
|
|
following_count=following_count,
|
|
is_following=is_following)
|
|
|
|
|
|
@users_bp.route('/user/<username>/follow', methods=['POST'])
|
|
@login_required
|
|
def follow(username):
|
|
"""关注用户"""
|
|
if not current_user.is_approved:
|
|
flash('您的账号尚未通过审核，无法关注', 'error')
|
|
return redirect(url_for('users.profile', username=username))
|
|
|
|
user = User.query.filter_by(username=username).first_or_404()
|
|
|
|
if user.id == current_user.id:
|
|
flash('不能关注自己', 'error')
|
|
return redirect(url_for('users.profile', username=username))
|
|
|
|
if current_user.is_following(user):
|
|
flash('您已经关注了该用户', 'info')
|
|
return redirect(url_for('users.profile', username=username))
|
|
|
|
current_user.follow(user)
|
|
db.session.commit()
|
|
|
|
flash(f'成功关注 {user.username}', 'success')
|
|
return redirect(url_for('users.profile', username=username))
|
|
|
|
|
|
@users_bp.route('/user/<username>/unfollow', methods=['POST'])
|
|
@login_required
|
|
def unfollow(username):
|
|
"""取消关注"""
|
|
user = User.query.filter_by(username=username).first_or_404()
|
|
|
|
if user.id == current_user.id:
|
|
flash('不能取消关注自己', 'error')
|
|
return redirect(url_for('users.profile', username=username))
|
|
|
|
if not current_user.is_following(user):
|
|
flash('您还没有关注该用户', 'info')
|
|
return redirect(url_for('users.profile', username=username))
|
|
|
|
current_user.unfollow(user)
|
|
db.session.commit()
|
|
|
|
flash(f'已取消关注 {user.username}', 'success')
|
|
return redirect(url_for('users.profile', username=username))
|
|
|
|
|
|
@users_bp.route('/user/<username>/followers')
|
|
def followers(username):
|
|
"""粉丝列表"""
|
|
user = User.query.filter_by(username=username).first_or_404()
|
|
followers = [follow.follower for follow in user.followers.all()]
|
|
return render_template('followers.html', user=user, followers=followers)
|
|
|
|
|
|
@users_bp.route('/user/<username>/following')
|
|
def following(username):
|
|
"""关注列表"""
|
|
user = User.query.filter_by(username=username).first_or_404()
|
|
following = [follow.following for follow in user.following.all()]
|
|
return render_template('following.html', user=user, following=following)
|