10afd5914e467aaaad36cde3df859486bcdd8387
[wolnelektury.git] / src / search / fields.py
1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
3 #
4 from django import forms
5 from django.forms.utils import flatatt
6 from django.utils.encoding import smart_text
7 from django.utils.safestring import mark_safe
8 from json import dumps
9
10
11 class JQueryAutoCompleteWidget(forms.TextInput):
12     def __init__(self, options, *args, **kwargs):
13         self.options = dumps(options)
14         super(JQueryAutoCompleteWidget, self).__init__(*args, **kwargs)
15
16     def render_js(self, field_id, options):
17         return '$(\'#%s\').autocomplete(%s).result(autocomplete_result_handler);' % (field_id, options)
18
19     def render(self, name, value=None, attrs=None, renderer=None):
20         final_attrs = self.build_attrs(self.attrs, attrs)
21         final_attrs["name"] = name
22         if value:
23             final_attrs['value'] = smart_text(value)
24
25         if 'id' not in self.attrs:
26             final_attrs['id'] = 'id_%s' % name
27
28         html = '''<input type="text" %(attrs)s/>
29             <script type="text/javascript">//<!--
30             %(js)s//--></script>
31             ''' % {
32                 'attrs': flatatt(final_attrs),
33                 'js': self.render_js(final_attrs['id'], self.options),
34             }
35
36         return mark_safe(html)
37
38
39 class JQueryAutoCompleteSearchWidget(JQueryAutoCompleteWidget):
40     def __init__(self, *args, **kwargs):
41         super(JQueryAutoCompleteSearchWidget, self).__init__(*args, **kwargs)
42
43     def render_js(self, field_id, options):
44         return ""
45
46
47 class JQueryAutoCompleteField(forms.CharField):
48     def __init__(self, source, options=None, *args, **kwargs):
49         if options is None:
50             options = {}
51         if 'widget' not in kwargs:
52             options['source'] = source
53             kwargs['widget'] = JQueryAutoCompleteWidget(options)
54
55         super(JQueryAutoCompleteField, self).__init__(*args, **kwargs)
56
57
58 class JQueryAutoCompleteSearchField(forms.CharField):
59     def __init__(self, options=None, *args, **kwargs):
60         if options is None:
61             options = {}
62         if 'widget' not in kwargs:
63             kwargs['widget'] = JQueryAutoCompleteSearchWidget(options)
64
65         super(JQueryAutoCompleteSearchField, self).__init__(*args, **kwargs)