148 lines
5.6 KiB
Python
148 lines
5.6 KiB
Python
from flask import Blueprint, render_template, request, flash, redirect, url_for
|
||
from flask_login import login_required, current_user
|
||
from sqlalchemy.exc import IntegrityError
|
||
from models import db, Course, Category, User, Review # Review добавлен для работы с отзывами
|
||
from tools import CoursesFilter, ImageSaver
|
||
|
||
bp = Blueprint('courses', __name__, url_prefix='/courses')
|
||
|
||
COURSE_PARAMS = [
|
||
'author_id', 'name', 'category_id', 'short_desc', 'full_desc'
|
||
]
|
||
|
||
def params():
|
||
return { p: request.form.get(p) or None for p in COURSE_PARAMS }
|
||
|
||
def search_params():
|
||
return {
|
||
'name': request.args.get('name'),
|
||
'category_ids': [x for x in request.args.getlist('category_ids') if x],
|
||
}
|
||
|
||
@bp.route('/')
|
||
def index():
|
||
courses = CoursesFilter(**search_params()).perform()
|
||
pagination = db.paginate(courses)
|
||
courses = pagination.items
|
||
categories = db.session.execute(db.select(Category)).scalars()
|
||
return render_template('courses/index.html',
|
||
courses=courses,
|
||
categories=categories,
|
||
pagination=pagination,
|
||
search_params=search_params())
|
||
|
||
@bp.route('/new')
|
||
@login_required
|
||
def new():
|
||
course = Course()
|
||
categories = db.session.execute(db.select(Category)).scalars()
|
||
users = db.session.execute(db.select(User)).scalars()
|
||
return render_template('courses/new.html',
|
||
categories=categories,
|
||
users=users,
|
||
course=course)
|
||
|
||
@bp.route('/create', methods=['POST'])
|
||
@login_required
|
||
def create():
|
||
f = request.files.get('background_img')
|
||
img = None
|
||
course = Course()
|
||
try:
|
||
if f and f.filename:
|
||
img = ImageSaver(f).save()
|
||
|
||
image_id = img.id if img else None
|
||
course = Course(**params(), background_image_id=image_id)
|
||
db.session.add(course)
|
||
db.session.commit()
|
||
except IntegrityError as err:
|
||
flash(f'Возникла ошибка при записи данных в БД. Проверьте корректность введённых данных. ({err})', 'danger')
|
||
db.session.rollback()
|
||
categories = db.session.execute(db.select(Category)).scalars()
|
||
users = db.session.execute(db.select(User)).scalars()
|
||
return render_template('courses/new.html',
|
||
categories=categories,
|
||
users=users,
|
||
course=course)
|
||
|
||
flash(f'Курс {course.name} был успешно добавлен!', 'success')
|
||
|
||
return redirect(url_for('courses.index'))
|
||
|
||
@bp.route('/<int:course_id>')
|
||
def show(course_id):
|
||
course = db.get_or_404(Course, course_id)
|
||
# Последние 5 отзывов
|
||
reviews = db.session.execute(
|
||
db.select(Review)
|
||
.filter_by(course_id=course_id)
|
||
.order_by(Review.created_at.desc())
|
||
.limit(5)
|
||
).scalars().all()
|
||
# Отзыв текущего пользователя
|
||
user_review = None
|
||
if current_user.is_authenticated:
|
||
user_review = db.session.execute(
|
||
db.select(Review).filter_by(course_id=course_id, user_id=current_user.id)
|
||
).scalar()
|
||
return render_template('courses/show.html', course=course, reviews=reviews, user_review=user_review)
|
||
|
||
# Страница всех отзывов с пагинацией
|
||
@bp.route('/<int:course_id>/reviews')
|
||
def reviews(course_id):
|
||
course = db.get_or_404(Course, course_id)
|
||
sort = request.args.get('sort', 'new')
|
||
query = db.select(Review).filter_by(course_id=course_id)
|
||
# Порядок сортировки
|
||
if sort == 'positive':
|
||
query = query.order_by(Review.rating.desc(), Review.created_at.desc())
|
||
elif sort == 'negative':
|
||
query = query.order_by(Review.rating.asc(), Review.created_at.desc())
|
||
else:
|
||
query = query.order_by(Review.created_at.desc())
|
||
pagination = db.paginate(query, per_page=5)
|
||
reviews_list = pagination.items
|
||
# Отзыв текущего пользователя
|
||
user_review = None
|
||
if current_user.is_authenticated:
|
||
user_review = db.session.execute(
|
||
db.select(Review).filter_by(course_id=course_id, user_id=current_user.id)
|
||
).scalar()
|
||
return render_template('courses/reviews.html',
|
||
course=course,
|
||
reviews=reviews_list,
|
||
pagination=pagination,
|
||
sort=sort,
|
||
user_review=user_review)
|
||
|
||
# Сохранение отзыва
|
||
@bp.route('/<int:course_id>/reviews/create', methods=['POST'])
|
||
@login_required
|
||
def create_review(course_id):
|
||
course = db.get_or_404(Course, course_id)
|
||
existing = db.session.execute(
|
||
db.select(Review).filter_by(course_id=course_id, user_id=current_user.id)
|
||
).scalar()
|
||
if existing:
|
||
flash('Вы уже оставили отзыв к этому курсу.', 'warning')
|
||
return redirect(url_for('courses.show', course_id=course_id))
|
||
|
||
rating = request.form.get('rating', type=int)
|
||
text = request.form.get('text', '').strip()
|
||
|
||
review = Review(
|
||
rating=rating,
|
||
text=text,
|
||
course_id=course_id,
|
||
user_id=current_user.id
|
||
)
|
||
db.session.add(review)
|
||
# Пересчёт рейтинга курса
|
||
course.rating_sum += rating
|
||
course.rating_num += 1
|
||
db.session.commit()
|
||
flash('Отзыв успешно добавлен!', 'success')
|
||
|
||
next_page = request.form.get('next') or url_for('courses.show', course_id=course_id)
|
||
return redirect(next_page)
|