summaryrefslogtreecommitdiffstats
path: root/askbot/templatetags/askbot_django.py
blob: 5a90af165890f6df6260869a6cc26731af78adba (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
from django import template
from django.core.urlresolvers import reverse
from django.contrib.humanize.templatetags.humanize import intcomma
from askbot.conf import settings as askbot_settings
from askbot.utils.pluralization import py_pluralize as _py_pluralize
from askbot.utils import functions
from askbot.search.state_manager import SearchState
from askbot.utils.slug import slugify

register = template.Library()

@register.filter
def as_js_bool(some_object):
    if bool(some_object):
        return 'true'
    return 'false'

@register.filter
def set_sort_method(search_state, sort_method):
    """sets sort method on search state"""
    return search_state.change_sort(sort_method).full_url()

@register.filter
def remove_author(search_state):
    return search_state.remove_author()

@register.filter
def remove_tags(search_state):
    return search_state.remove_tags()

@register.filter
def change_page(search_state, page):
    return search_state.change_page(page)

@register.filter
def get_url(search_state):
    return search_state.full_url()

@register.filter
def get_answer_count(thread, visitor):
    return thread.get_answer_count(visitor)

@register.filter
def get_latest_revision(thread, visitor):
    return thread.get_latest_revision(visitor)

setup_paginator = register.filter(functions.setup_paginator)

@register.filter
def reverse(iterable):
    return reversed(iterable)

@register.filter
def tag_used_count(count):
    return '<span class="tag-number">&#215; %s</span>' % intcomma(count)

@register.filter
def get_avatar_url(user):
    return user.get_avatar_url()

@register.filter
def question_absolute_url(question, thread):
    return question.get_absolute_url(thread=thread)

@register.filter
def show_block_to(block_name, user):
    block = getattr(askbot_settings, block_name)
    if block:
        flag_name = block_name + '_ANON_ONLY'
        require_anon = getattr(askbot_settings, flag_name, False)
        return (require_anon is False) or user.is_anonymous()
    return False

@register.filter
def py_pluralize(source, count):
    plural_forms = source.strip().split('\n')
    return _py_pluralize(plural_forms, count)

class ThreadSummaryNode(template.Node):
    def __init__(self, thread_token, search_state_token, visitor_token):
        self.thread = template.Variable(thread_token)
        self.search_state = template.Variable(search_state_token)
        self.visitor = template.Variable(visitor_token)

    def render(self, context):
        thread = self.thread.resolve(context)
        search_state = self.search_state.resolve(context)
        visitor = self.visitor.resolve(context)
        return thread.get_summary_html(search_state, visitor)

@register.tag
def thread_summary_html(thread, tokens): 
    tag_name, thread_token, search_state_token, visitor_token = tokens.split_contents()
    return ThreadSummaryNode(thread_token, search_state_token, visitor_token)

@register.inclusion_tag('widgets/avatar.html')
def avatar(user, size):
    return {'user': user, 'size': size}

@register.inclusion_tag('widgets/tag_list.html')
def tag_list_widget(tags, **kwargs):
    kwargs['tags'] = tags
    return kwargs

@register.inclusion_tag('widgets/tag.html')
def tag_widget(tag, **kwargs):
    kwargs['tag'] = tag
    if kwargs.get('search_state') is None:
        kwargs['search_state'] = SearchState.get_empty()
    return kwargs

@register.inclusion_tag('widgets/radio_select.html')
def radio_select(name=None, value=None, choices=None):
    choices_data = list()
    for choice in choices:
        choice_datum = {
            'id': 'id_%s_%s' % (name, choice[0]),
            'value': choice[0],
            'label': choice[1]
        }
        choices_data.append(choice_datum)
    return {
        'name': name,
        'value': value,
        'choices': choices_data
    }

@register.inclusion_tag('widgets/tag_cloud.html')
def tag_cloud(tags=None, font_sizes=None, search_state=None):
    tags_data = list()
    for tag in tags:
        tag_datum = {
            'name': tag.name,
            'font_size': font_sizes[tag.name]
        }
        tags_data.append(tag_datum)
    return {
        'tags': tags_data,
        'search_state': search_state
    }

@register.inclusion_tag('widgets/user_country_flag.html')
def user_country_flag(user):
    context = {
        'user': user,
    }
    if user.country and user.show_counry:
        context['flag_url'] = '/images/flags/' + user.country.code.lower() + '.gif'
    return context

@register.inclusion_tag('widgets/user_primary_group.html')
def user_primary_group(user):
    group = user.get_primary_group()    
    group_name = group.name.replace('-', ' ')
    group_url = reverse('users_by_group', args=(group.id, slugify(group_name)))
    return {
        'group_name': group_name,
        'group_url': group_url
    }