Source code for fireblog.tags

'Views for managing tags, and viewing all posts with a particular tag.'
from operator import attrgetter
import logging
import fireblog.utils as utils
from fireblog.theme import use_template, TemplateResponseDict
from pyramid.view import view_config
from pyramid.httpexceptions import (
    HTTPFound,
    HTTPNotFound,
)
import paginate
from sqlalchemy.orm.exc import NoResultFound
from fireblog.models import (
    DBSession,
    Tags,
)


log = logging.getLogger(__name__)


@view_config(route_name='tag_view',
             decorator=use_template('multiple_posts.mako'))
def tag_view(request):
    """Display a page similar to that of
    :py:func:`fireblog.views.view_all_posts` but just showing the posts that
    have the supplied tag on them. The tag supplied is
    ``request.matchdict['tag_name']``."""
    page_num = request.params.get('p', None) or 1
    if request.params.get('sort-ascending', False):
        sort_desc = False
        sort_ascending_query_text = '&sort-ascending=true'
    else:
        sort_desc = True
        sort_ascending_query_text = ''
    tag = request.matchdict['tag_name']
    oldest_first_url = request.route_url(
        'tag_view', tag_name=tag,
        _query=(('p', page_num), ('sort-ascending', 'true')))
    newest_first_url = request.route_url('tag_view', tag_name=tag,
                                         _query=[('p', page_num)])
    try:
        tag_obj = DBSession.query(Tags).filter_by(tag=tag).one()
    except NoResultFound:
        log.debug('No tag found for tag name: {}'.format(tag))
        return HTTPNotFound('no such tag exists.')
    posts_obj = tag_obj.posts
    posts_obj = sorted(posts_obj, key=attrgetter("created"), reverse=sort_desc)
    page = paginate.Page(
        posts_obj, page=page_num, items_per_page=20)
    posts, code_styles = utils.create_post_list_from_posts_obj(
        request, page)
    pager = page.pager(
        url=request.route_url('tag_view', tag_name=tag,
                              _query='p=$page' + sort_ascending_query_text))
    return TemplateResponseDict(title='Posts tagged with {}'.format(tag),
                                posts=posts,
                                page_num=page_num,
                                pager=pager,
                                uuid=tag_obj.uuid,
                                code_styles=code_styles,
                                oldest_first_url=oldest_first_url,
                                newest_first_url=newest_first_url)


@view_config(route_name='tag_manager', decorator=use_template(
    'tag_manager.mako'), permission='manage-tags')
def tag_manager(request):
    """Display a page listing all the tags that exist in the db, and how many
    posts have been tagged with each tag. The user can then rename one or
    more tags, and specify if any tags should be deleted."""
    tags = DBSession.query(Tags).order_by(Tags.tag).all()
    if 'form.submitted' in request.params:
        for tag in tags:
            # 1. If it is unchecked, delete it
            keep_tag = request.params.get('check-' + tag.tag)
            if not keep_tag:
                # Delete the tag
                log.info('Deleting tag: {}'.format(tag.tag))
                DBSession.delete(tag)
            # 2. Else, if the name has been changed, update the name
            else:
                new_tag_name = request.params.get('text-' + tag.tag)
                if tag.tag != new_tag_name:
                    # Change the tag name in the db
                    log.info('Changing tag from {} to {}'.format(tag.tag,
                                                                 new_tag_name))
                    tag.tag = new_tag_name
        return HTTPFound(location=request.route_url('tag_manager'))
    tags = [(tag.tag, len(tag.posts)) for tag in tags]
    return TemplateResponseDict(tags=tags,
                                title='Tag manager',
                                save_url=request.route_url('tag_manager'))