board.py
49 lines
| 1.2 KiB
| text/x-python
|
PythonLexer
neko259
|
r320 | from django.core.urlresolvers import reverse | ||
from django.shortcuts import get_object_or_404 | ||||
from boards.models import Post | ||||
from boards.views import thread | ||||
from django import template | ||||
register = template.Library() | ||||
neko259
|
r460 | actions = [ | ||
{ | ||||
'name': 'google', | ||||
'link': 'http://google.com/searchbyimage?image_url=%s', | ||||
}, | ||||
{ | ||||
'name': 'iqdb', | ||||
'link': 'http://iqdb.org/?url=%s', | ||||
}, | ||||
] | ||||
neko259
|
r320 | |||
@register.simple_tag(name='post_url') | ||||
def post_url(*args, **kwargs): | ||||
post_id = args[0] | ||||
post = get_object_or_404(Post, id=post_id) | ||||
neko259
|
r400 | if not post.is_opening(): | ||
link = reverse(thread, kwargs={ | ||||
'post_id': post.thread_new.get_opening_post().id}) + '#' + str( | ||||
post_id) | ||||
neko259
|
r320 | else: | ||
neko259
|
r322 | link = reverse(thread, kwargs={'post_id': post_id}) | ||
neko259
|
r320 | |||
return link | ||||
neko259
|
r460 | |||
@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 | ||||