diff options
author | luxagraf <sng@luxagraf.net> | 2010-10-23 19:46:20 -0400 |
---|---|---|
committer | luxagraf <sng@luxagraf.net> | 2010-10-23 19:46:20 -0400 |
commit | c59a2a69fb38b92b6c45bcf7431d2b1a3c5dce3c (patch) | |
tree | 2fc44ca867839d5e591e21467b6e4526f7a9f080 /lib/filebrowser/fields.py | |
parent | ed77da873e675f02f12cbab9be27f342f825444b (diff) |
added grappelli, filebrowser, chunks and tagging to lcal repo
Diffstat (limited to 'lib/filebrowser/fields.py')
-rw-r--r-- | lib/filebrowser/fields.py | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/lib/filebrowser/fields.py b/lib/filebrowser/fields.py new file mode 100644 index 0000000..da19bac --- /dev/null +++ b/lib/filebrowser/fields.py @@ -0,0 +1,127 @@ +# coding: utf-8 + +# imports +import os + +# django imports +from django.db import models +from django import forms +from django.forms.widgets import Input +from django.db.models.fields import Field, CharField +from django.utils.encoding import force_unicode +from django.template.loader import render_to_string +from django.utils.translation import ugettext_lazy as _ + +# filebrowser imports +from filebrowser.settings import * +from filebrowser.base import FileObject +from filebrowser.functions import url_to_path + + +class FileBrowseWidget(Input): + input_type = 'text' + + class Media: + js = (os.path.join(URL_FILEBROWSER_MEDIA, 'js/AddFileBrowser.js'), ) + + def __init__(self, attrs=None): + self.directory = attrs.get('directory', '') + self.extensions = attrs.get('extensions', '') + self.format = attrs.get('format', '') + if attrs is not None: + self.attrs = attrs.copy() + else: + self.attrs = {} + + def render(self, name, value, attrs=None): + if value is None: + value = "" + final_attrs = self.build_attrs(attrs, type=self.input_type, name=name) + final_attrs['search_icon'] = URL_FILEBROWSER_MEDIA + 'img/filebrowser_icon_show.gif' + final_attrs['directory'] = self.directory + final_attrs['extensions'] = self.extensions + final_attrs['format'] = self.format + final_attrs['ADMIN_THUMBNAIL'] = ADMIN_THUMBNAIL + final_attrs['DEBUG'] = DEBUG + if value != "": + try: + final_attrs['directory'] = os.path.split(value.path_relative_directory)[0] + except: + pass + return render_to_string("filebrowser/custom_field.html", locals()) + + +class FileBrowseFormField(forms.CharField): + widget = FileBrowseWidget + + default_error_messages = { + 'extension': _(u'Extension %(ext)s is not allowed. Only %(allowed)s is allowed.'), + } + + def __init__(self, max_length=None, min_length=None, + directory=None, extensions=None, format=None, + *args, **kwargs): + self.max_length, self.min_length = max_length, min_length + self.directory = directory + self.extensions = extensions + if format: + self.format = format or '' + self.extensions = extensions or EXTENSIONS.get(format) + super(FileBrowseFormField, self).__init__(*args, **kwargs) + + def clean(self, value): + value = super(FileBrowseFormField, self).clean(value) + if value == '': + return value + file_extension = os.path.splitext(value)[1].lower() + if self.extensions and not file_extension in self.extensions: + raise forms.ValidationError(self.error_messages['extension'] % {'ext': file_extension, 'allowed': ", ".join(self.extensions)}) + return value + + +class FileBrowseField(Field): + __metaclass__ = models.SubfieldBase + + def __init__(self, *args, **kwargs): + self.directory = kwargs.pop('directory', '') + self.extensions = kwargs.pop('extensions', '') + self.format = kwargs.pop('format', '') + return super(FileBrowseField, self).__init__(*args, **kwargs) + + def to_python(self, value): + if not value or isinstance(value, FileObject): + return value + return FileObject(url_to_path(value)) + + def get_db_prep_value(self, value): + if value is None: + return None + return unicode(value) + + + def get_manipulator_field_objs(self): + return [oldforms.TextField] + + def get_internal_type(self): + return "CharField" + + def formfield(self, **kwargs): + attrs = {} + attrs["directory"] = self.directory + attrs["extensions"] = self.extensions + attrs["format"] = self.format + defaults = { + 'form_class': FileBrowseFormField, + 'widget': FileBrowseWidget(attrs=attrs), + 'directory': self.directory, + 'extensions': self.extensions, + 'format': self.format + } + defaults.update(kwargs) + return super(FileBrowseField, self).formfield(**defaults) + +try: + from south.modelsinspector import add_introspection_rules + add_introspection_rules([], ["^filebrowser\.fields\.FileBrowseField"]) +except: + pass |