maple-bbs/maple/tag/views.py

#!/usr/bin/env python
# -*- coding=UTF-8 -*-
# **************************************************************************
# Copyright © 2016 jianglin
# File Name: views.py
# Author: jianglin
# Email: xiyang0807@gmail.com
# Created: 2016-05-20 13:18:19 (CST)
# Last Update:星期五 2016-7-15 18:33:44 (CST)
#          By:
# Description:
# **************************************************************************
from flask import render_template, request, url_for
from maple import app
from maple.helpers import is_num
from maple.topic.models import Tags, Topic
from maple.filters import Filters
from urllib.parse import urljoin
from werkzeug.utils import escape
from werkzeug.contrib.atom import AtomFeed

# site = Blueprint('tag', __name__)


# @site.route('', defaults={'tag': None})
# @site.route('/<tag>')
def tag(tag):
    if tag is None:
        tags = Tags.query.distinct(Tags.tagname).all()
        data = {'title': '所有标签 - ', 'tags': tags}
        return render_template('tag/tag_list.html', **data)
    else:
        page = is_num(request.args.get('page'))
        topic_base = Topic.query.join(Topic.tags).filter(Tags.tagname == tag)
        topics = topic_base.filter(Topic.is_top == False).paginate(
            page, app.config['PER_PAGE'],
            error_out=True)
        top_topics = topic_base.filter(Topic.is_top == True).limit(5).all()
        tag = Tags.query.filter_by(tagname=tag).first_or_404()
        data = {'title': '%s - ' % tag,
                'tag': tag,
                'topics': topics,
                'top_topics': top_topics}
        return render_template('tag/tag.html', **data)

# @site.route('/hot')
# def hot():
#     tags = Tags.query.order_by(Tags.time.desc()).limit(10).all()
#     return tags

# @site.route('/recent')
# def recent():
#     tags = Tags.query.order_by(Tags.time.desc()).limit(10).all()
#     return tags


# @site.route('/<tag>/feed')
def rss(tag):
    feed = AtomFeed('%s·HonMaple社区' % tag,
                    feed_url=request.url,
                    url=request.url_root,
                    subtitle='I like solitude, yearning for freedom')
    topics = Topic.query.join(Topic.tags).filter(Tags.tagname == tag).limit(
        10).all()
    for topic in topics:
        feed.add(
            topic.title,
            escape(Filters.safe_markdown(topic.content) if topic.is_markdown
                   else topic.content),
            content_type='html',
            author=topic.author.username,
            url=urljoin(request.url_root,
                        url_for('topic.topic', uid=topic.uid)),
            updated=topic.publish,
            published=topic.publish)
    return feed.get_response()