# - Added wrapper div
import math
from itertools import chain
from django import forms
from django.utils.html import conditional_escape
from django.utils.safestring import mark_safe
class ColumnCheckboxSelectMultiple(forms.CheckboxSelectMultiple):
"""
Widget that renders multiple-select checkboxes in columns.
Constructor takes number of columns and css class to apply
to the <ul> elements that make up the columns.
"""
def __init__(self, columns=2, css_class=None, wrapper_css_class=None, **kwargs):
super(self.__class__, self).__init__(**kwargs)
self.columns = columns
self.css_class = css_class
self.wrapper_css_class = wrapper_css_class
def render(self, name, value, attrs=None, choices=()):
if value is None: value = []
has_id = attrs and 'id' in attrs
final_attrs = self.build_attrs(attrs)
choices_enum = list(enumerate(chain(self.choices, choices)))
# This is the part that splits the choices into columns.
# Slices vertically. Could be changed to slice horizontally, etc.
column_sizes = columnize(len(choices_enum), self.columns)
columns = []
for column_size in column_sizes:
columns.append(choices_enum[:column_size])
choices_enum = choices_enum[column_size:]
output = []
for column in columns:
if self.css_class:
output.append(u'<ul class="%s">' % self.css_class)
else:
output.append(u'<ul>')
# Normalize to strings
str_values = set([v for v in value])
for i, (option_value, option_label) in column:
# If an ID attribute was given, add a numeric index as a suffix,
# so that the checkboxes don't all have the same ID attribute.
if has_id:
final_attrs = dict(final_attrs, id='%s_%s' % (
attrs['id'], i))
label_for = u' for="%s"' % final_attrs['id']
else:
label_for = ''
cb = forms.CheckboxInput(
final_attrs, check_test=lambda value: value in str_values)
option_value = option_value
rendered_cb = cb.render(name, option_value)
option_label = conditional_escape(option_label)
output.append(u'<li><label%s>%s %s</label></li>' % (
label_for, rendered_cb, option_label))
output.append(u'</ul>')
if self.wrapper_css_class:
wrapper = u'<div class="%s">%s</div>' % \
(self.wrapper_css_class, u'\n'.join(output))
else:
wrapper = u'<div>%s</div>' % self.wrapper_css_class
return mark_safe(wrapper)
def columnize(items, columns):
"""
Return a list containing numbers of elements per column if `items` items
are to be divided into `columns` columns.
>>> columnize(10, 1)
[10]
>>> columnize(10, 2)
[5, 5]
>>> columnize(10, 3)
[4, 3, 3]
>>> columnize(3, 4)
[1, 1, 1, 0]
"""
elts_per_column = []
for col in range(columns):
col_size = int(math.ceil(float(items) / columns))
elts_per_column.append(col_size)
items -= col_size
columns -= 1
return elts_per_column
# this is improved from django-form-utils package
# https://bitbucket.org/carljm/django-form-utils/src/c29eb2e1def2aaa5f2a4c302f233b2298a54caf7/form_utils/widgets.py?fileviewer=file-view-default
# Compatible with Django 1.9, 1.10
def thumbnail(image_path, width, height):
absolute_url = posixpath.join(settings.MEDIA_URL, image_path)
return '<img src="%s" alt="%s" class="widget-img" />' % (absolute_url, image_path)
class ImageWidget(forms.ClearableFileInput):
template = '<div>%(image)s</div>' \
'<div>%(clear_template)s</div>' \
'<div>%(input)s</div>'
def __init__(self, attrs=None, template=None, width=200, height=200):
if template is not None:
self.template = template
self.width = width
self.height = height
super(ImageWidget, self).__init__(attrs)
def render(self, name, value, attrs=None):
substitutions = {
'initial_text': self.initial_text,
'input_text': self.input_text,
'clear_template': '',
'clear_checkbox_label': self.clear_checkbox_label,
}
if not self.is_required:
checkbox_name = self.clear_checkbox_name(name)
checkbox_id = self.clear_checkbox_id(checkbox_name)
substitutions['clear_checkbox_name'] = conditional_escape(checkbox_name)
substitutions['clear_checkbox_id'] = conditional_escape(checkbox_id)
substitutions['clear'] = forms.CheckboxInput().render(checkbox_name, False, attrs={'id': checkbox_id})
input_html = super(forms.ClearableFileInput, self).render(name, value, attrs)
if value and hasattr(value, 'width') and hasattr(value, 'height'):
image_html = thumbnail(value.name, self.width, self.height)
output = self.template % {'input': input_html,
'image': image_html,
'clear_template': self.template_with_clear % substitutions}
else:
output = input_html
return mark_safe(output)
class EditProfileForm(forms.ModelForm):
class Meta:
model = models.Profile
fields = '__all__'
widgets = {
'avatar': ImageWidget
}