forum/settings/forms.py

   1 import os
   2 from string import strip
   3 from django import forms
   4 from base import Setting
   5 from django.utils.translation import ugettext as _
   6 from django.core.files.storage import FileSystemStorage
   7
   8 class DummySetting:
   9     pass
  10
  11 class UnfilteredField(forms.CharField):
  12     def clean(self, value):
  13             return value
  14
  15
  16 class SettingsSetForm(forms.Form):
  17     def __init__(self, set, data=None, *args, **kwargs):
  18         if data is None:
  19             data = dict([(setting.name, setting.value) for setting in set])
  20
  21         super(SettingsSetForm, self).__init__(data, *args, **kwargs)
  22
  23         for setting in set:
  24             if isinstance(setting, (Setting.emulators.get(str, DummySetting), Setting.emulators.get(unicode, DummySetting))):
  25                 if not setting.field_context.get('widget', None):
  26                     setting.field_context['widget'] = forms.TextInput(attrs={'class': 'longstring'})
  27                 field = forms.CharField(**setting.field_context)
  28             elif isinstance(setting, Setting.emulators.get(float, DummySetting)):
  29                 field = forms.FloatField(**setting.field_context)
  30             elif isinstance(setting, Setting.emulators.get(int, DummySetting)):
  31                 field = forms.IntegerField(**setting.field_context)
  32             elif isinstance(setting, Setting.emulators.get(bool, DummySetting)):
  33                 field = forms.BooleanField(**setting.field_context)
  34             else:
  35                 field = UnfilteredField(**setting.field_context)
  36
  37             self.fields[setting.name] = field
  38
  39         self.set = set
  40
  41     def save(self):
  42         for setting in self.set:
  43             setting.set_value(self.cleaned_data[setting.name])
  44
  45 class ImageFormWidget(forms.Widget):
  46     def render(self, name, value, attrs=None):
  47         return """
  48             <img src="%(value)s" /><br />
  49             %(change)s: <input type="file" name="%(name)s" />
  50             <input type="hidden" name="%(name)s_old" value="%(value)s" />
  51             """ % {'name': name, 'value': value, 'change': _('Change this:')}
  52
  53     def value_from_datadict(self, data, files, name):
  54         if name in files:
  55             f = files[name]
  56
  57             # check file type
  58             file_name_suffix = os.path.splitext(f.name)[1].lower()
  59
  60             if not file_name_suffix in ('.jpg', '.jpeg', '.gif', '.png', '.bmp', '.tiff', '.ico'):
  61                 raise Exception('File type not allowed')
  62
  63             from forum.settings import UPFILES_FOLDER, UPFILES_ALIAS
  64
  65             storage = FileSystemStorage(str(UPFILES_FOLDER), str(UPFILES_ALIAS))
  66             new_file_name = storage.save(f.name, f)
  67             return str(UPFILES_ALIAS) + new_file_name
  68         else:
  69             if "%s_old" % name in data:
  70                 return data["%s_old" % name]
  71             elif name in data:
  72                 return data[name]
  73
  74 class StringListWidget(forms.Widget):
  75     def render(self, name, value, attrs=None):
  76         ret = ""
  77         for s in value:
  78             ret += """
  79             <div>
  80                 <input type="text" name="%(name)s" value="%(value)s" />
  81                 <button class="string_list_widget_button">-</button>
  82             </div>
  83             """  % {'name': name, 'value': s}
  84
  85         return """
  86             <div class="string_list_widgets">
  87                 %(ret)s
  88                 <div><button name="%(name)s" class="string_list_widget_button add">+</button></div>
  89             </div>
  90             """ % dict(name=name, ret=ret)
  91
  92     def value_from_datadict(self, data, files, name):
  93         if 'submit' in data:
  94             return data.getlist(name)
  95         else:
  96             return data[name]
  97
  98 class CommaStringListWidget(forms.Textarea):
  99     def value_from_datadict(self, data, files, name):
 100         if 'submit' in data:
 101             return map(strip, data[name].split(','))
 102         else:
 103             return ', '.join(data[name])
 104
 105
 106
 107