50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
from flask import Blueprint, redirect, url_for, render_template, request, session, abort
|
|
|
|
from ..models import Topics, Threads, Subscriptions
|
|
from ..auth import get_active_user
|
|
import math
|
|
|
|
bp = Blueprint('topics', __name__, url_prefix = '/topics/')
|
|
|
|
@bp.get('/')
|
|
def all_topics():
|
|
topic_list = Topics.get_list()
|
|
return render_template('topics/topics.html', topics=topic_list)
|
|
|
|
@bp.get('/<int:topic_id>/')
|
|
def topic_by_id(topic_id):
|
|
topic = Topics.find({'id': topic_id})
|
|
if not topic:
|
|
abort(404)
|
|
return redirect(url_for('.topic', topic_id=topic_id, slug=topic.slug, **request.args))
|
|
|
|
@bp.get('/<int:topic_id>/<slug>/')
|
|
def topic(topic_id, slug):
|
|
topic = Topics.find({'id': topic_id})
|
|
if not topic:
|
|
abort(404)
|
|
if topic.slug != slug:
|
|
return redirect(url_for('.topic', topic_id=topic_id, slug=topic.slug, **request.args))
|
|
|
|
sort_by = request.args.get('sort_by', default=session.get('sort_by', default='activity'))
|
|
PER_PAGE = 10
|
|
threads_count = Threads.count({'topic_id': topic.id})
|
|
page_count = max(1, math.ceil(threads_count / PER_PAGE))
|
|
try:
|
|
page = max(1, min(int(request.args.get('page', default=1)), page_count))
|
|
except ValueError:
|
|
abort(404)
|
|
threads = topic.get_threads(PER_PAGE, page, sort_by)
|
|
subscriptions = {}
|
|
user = get_active_user()
|
|
if user:
|
|
for thread in threads:
|
|
subscription = Subscriptions.find({'user_id': user.id, 'thread_id': thread['id']})
|
|
if subscription:
|
|
subscriptions[thread['id']] = subscription.get_unread_count()
|
|
return render_template('topics/topic.html', topic=topic, threads=threads, sort_by=sort_by, page=page, page_count=page_count, subscriptions=subscriptions)
|
|
|
|
@bp.get('/<int:topic_id>/feed.atom/')
|
|
def feed(topic_id):
|
|
return 'stub'
|