gestioCOF/gestioncof/widgets.py

21 lines
782 B
Python
Raw Normal View History

2012-07-11 17:39:20 +02:00
from django.forms.widgets import Widget
from django.forms.utils import flatatt
2012-07-11 17:39:20 +02:00
from django.utils.safestring import mark_safe
class TriStateCheckbox(Widget):
2012-07-11 17:39:20 +02:00
def __init__(self, attrs=None, choices=()):
2018-01-16 16:22:52 +01:00
super().__init__(attrs)
2012-07-11 17:39:20 +02:00
# choices can be any iterable, but we may need to render this widget
# multiple times. Thus, collapse it into a list so it can be consumed
# more than once.
self.choices = list(choices)
def render(self, name, value, attrs=None, choices=()):
if value is None:
value = 'none'
attrs['value'] = value
final_attrs = self.build_attrs(self.attrs, attrs)
output = ["<span class=\"tristate\"%s></span>" % flatatt(final_attrs)]
2012-07-11 17:39:20 +02:00
return mark_safe('\n'.join(output))