|
1 """ |
|
2 Extra HTML Widget classes |
|
3 """ |
|
4 |
|
5 import datetime |
|
6 import re |
|
7 |
|
8 from django.forms.widgets import Widget, Select |
|
9 from django.utils.dates import MONTHS |
|
10 from django.utils.safestring import mark_safe |
|
11 |
|
12 __all__ = ('SelectDateWidget',) |
|
13 |
|
14 RE_DATE = re.compile(r'(\d{4})-(\d\d?)-(\d\d?)$') |
|
15 |
|
16 class SelectDateWidget(Widget): |
|
17 """ |
|
18 A Widget that splits date input into three <select> boxes. |
|
19 |
|
20 This also serves as an example of a Widget that has more than one HTML |
|
21 element and hence implements value_from_datadict. |
|
22 """ |
|
23 month_field = '%s_month' |
|
24 day_field = '%s_day' |
|
25 year_field = '%s_year' |
|
26 |
|
27 def __init__(self, attrs=None, years=None): |
|
28 # years is an optional list/tuple of years to use in the "year" select box. |
|
29 self.attrs = attrs or {} |
|
30 if years: |
|
31 self.years = years |
|
32 else: |
|
33 this_year = datetime.date.today().year |
|
34 self.years = range(this_year, this_year+10) |
|
35 |
|
36 def render(self, name, value, attrs=None): |
|
37 try: |
|
38 year_val, month_val, day_val = value.year, value.month, value.day |
|
39 except AttributeError: |
|
40 year_val = month_val = day_val = None |
|
41 if isinstance(value, basestring): |
|
42 match = RE_DATE.match(value) |
|
43 if match: |
|
44 year_val, month_val, day_val = [int(v) for v in match.groups()] |
|
45 |
|
46 output = [] |
|
47 |
|
48 if 'id' in self.attrs: |
|
49 id_ = self.attrs['id'] |
|
50 else: |
|
51 id_ = 'id_%s' % name |
|
52 |
|
53 month_choices = MONTHS.items() |
|
54 month_choices.sort() |
|
55 local_attrs = self.build_attrs(id=self.month_field % id_) |
|
56 select_html = Select(choices=month_choices).render(self.month_field % name, month_val, local_attrs) |
|
57 output.append(select_html) |
|
58 |
|
59 day_choices = [(i, i) for i in range(1, 32)] |
|
60 local_attrs['id'] = self.day_field % id_ |
|
61 select_html = Select(choices=day_choices).render(self.day_field % name, day_val, local_attrs) |
|
62 output.append(select_html) |
|
63 |
|
64 year_choices = [(i, i) for i in self.years] |
|
65 local_attrs['id'] = self.year_field % id_ |
|
66 select_html = Select(choices=year_choices).render(self.year_field % name, year_val, local_attrs) |
|
67 output.append(select_html) |
|
68 |
|
69 return mark_safe(u'\n'.join(output)) |
|
70 |
|
71 def id_for_label(self, id_): |
|
72 return '%s_month' % id_ |
|
73 id_for_label = classmethod(id_for_label) |
|
74 |
|
75 def value_from_datadict(self, data, files, name): |
|
76 y, m, d = data.get(self.year_field % name), data.get(self.month_field % name), data.get(self.day_field % name) |
|
77 if y and m and d: |
|
78 return '%s-%s-%s' % (y, m, d) |
|
79 return data.get(name, None) |