summaryrefslogtreecommitdiff
path: root/lib/filebrowser/fields.py
diff options
context:
space:
mode:
authorluxagraf <sng@luxagraf.net>2010-10-23 19:46:20 -0400
committerluxagraf <sng@luxagraf.net>2010-10-23 19:46:20 -0400
commitc59a2a69fb38b92b6c45bcf7431d2b1a3c5dce3c (patch)
tree2fc44ca867839d5e591e21467b6e4526f7a9f080 /lib/filebrowser/fields.py
parented77da873e675f02f12cbab9be27f342f825444b (diff)
added grappelli, filebrowser, chunks and tagging to lcal repo
Diffstat (limited to 'lib/filebrowser/fields.py')
-rw-r--r--lib/filebrowser/fields.py127
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