cm->m / g->kg widget

1 files changed, 30 insertions, 0 deletions

diff --git a/ishtar_common/widgets.py b/ishtar_common/widgets.py
index f668ee534..326966064 100644
--- a/ishtar_common/widgets.py
+++ b/ishtar_common/widgets.py
@@ -470,6 +470,36 @@ class SquareMeterWidget(forms.TextInput):
         return mark_safe(rendered)
 
 
+class GramKilogramWidget(forms.TextInput):
+    def render(self, name, value, attrs=None, renderer=None):
+        if not value:
+            value = u""
+        final_attrs = flatatt(
+            self.build_attrs(attrs, {"name": name, "value": value}))
+        dct = {'final_attrs': final_attrs,
+               'unit': u"g",
+               'id': attrs['id'],
+               "safe_id": attrs['id'].replace('-', '_')}
+        t = loader.get_template('blocks/GramKilogramWidget.html')
+        rendered = t.render(dct)
+        return mark_safe(rendered)
+
+
+class CentimeterMeterWidget(forms.TextInput):
+    def render(self, name, value, attrs=None, renderer=None):
+        if not value:
+            value = u""
+        final_attrs = flatatt(
+            self.build_attrs(attrs, {"name": name, "value": value}))
+        dct = {'final_attrs': final_attrs,
+               'unit': u"cm",
+               'id': attrs['id'],
+               "safe_id": attrs['id'].replace('-', '_')}
+        t = loader.get_template('blocks/CentimeterMeterWidget.html')
+        rendered = t.render(dct)
+        return mark_safe(rendered)
+
+
 AreaWidget = forms.TextInput
 
 if settings.SURFACE_UNIT == 'square-metre':