Add a widget to do free input parcel additions (refs #809)

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)