boards/templatetags/board.py

from django.shortcuts import get_object_or_404
from django import template


register = template.Library()

actions = [
    {
        'name': 'google',
        'link': 'http://google.com/searchbyimage?image_url=%s',
    },
    {
        'name': 'iqdb',
        'link': 'http://iqdb.org/?url=%s',
    },
]


@register.simple_tag(name='post_url')
def post_url(*args, **kwargs):
    post_id = args[0]

    post = get_object_or_404('Post', id=post_id)

    return post.get_url()


@register.simple_tag(name='post_object_url')
def post_object_url(*args, **kwargs):
    post = args[0]

    if 'thread' in kwargs:
        post_thread = kwargs['thread']
    else:
        post_thread = None

    return post.get_url(thread=post_thread)


@register.simple_tag(name='image_actions')
def image_actions(*args, **kwargs):
    image_link = args[0]
    if len(args) > 1:
        image_link = 'http://' + args[1] + image_link # TODO https?

    result = ''

    for action in actions:
        result += '[<a href="' + action['link'] % image_link + '">' + \
                  action['name'] + '</a>]'

    return result


# TODO Use get_view of a post instead of this
@register.inclusion_tag('boards/post.html', name='post_view')
def post_view(post, moderator=False, need_open_link=False, truncated=False,
              **kwargs):
    """
    Get post
    """

    if 'is_opening' in kwargs:
        is_opening = kwargs['is_opening']
    else:
        is_opening = post.is_opening()

    if 'thread' in kwargs:
        thread = kwargs['thread']
    else:
        thread = post.get_thread()

    if 'can_bump' in kwargs:
        can_bump = kwargs['can_bump']
    else:
        can_bump = thread.can_bump()

    opening_post_id = thread.get_opening_post_id()

    return {
        'post': post,
        'moderator': moderator,
        'is_opening': is_opening,
        'thread': thread,
        'bumpable': can_bump,
        'need_open_link': need_open_link,
        'truncated': truncated,
        'opening_post_id': opening_post_id,
    }