diff options
Diffstat (limited to 'archaeological_operations/widgets.py')
| -rw-r--r-- | archaeological_operations/widgets.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/archaeological_operations/widgets.py b/archaeological_operations/widgets.py index 0e84b2047..6c4ebe79a 100644 --- a/archaeological_operations/widgets.py +++ b/archaeological_operations/widgets.py @@ -19,6 +19,8 @@ from django import forms from django.forms import widgets +from django.utils.safestring import mark_safe +from django.utils.translation import ugettext_lazy as _ class ParcelWidget(widgets.MultiWidget): def __init__(self, attrs=None): @@ -41,3 +43,10 @@ class ParcelWidget(widgets.MultiWidget): def format_output(self, rendered_widgets): return u' / '.join(rendered_widgets) + +class SelectParcelWidget(widgets.TextInput): + def render(self, *args, **kwargs): + render = super(SelectParcelWidget, self).render(*args, **kwargs) + render += u" <button name='formset_add' value='add'>%s</button>" \ + % _(u"Add") + return mark_safe(render) |