apps/filebrowser/fields.py

   1 # coding: utf-8
   2
   3 import os
   4
   5 # django imports
   6 from django.db import models
   7 from django import forms
   8 from django.forms.widgets import Input
   9 from django.db.models.fields import Field, CharField
  10 from django.utils.encoding import force_unicode
  11 from django.template.loader import render_to_string
  12 from django.utils.translation import ugettext_lazy as _
  13
  14 # filebrowser imports
  15 from filebrowser.functions import _url_to_path, _dir_from_url, _get_version_path, _url_join
  16 from filebrowser.fb_settings import *
  17 from filebrowser.base import FileObject
  18
  19 class FileBrowseWidget(Input):
  20     input_type = 'text'
  21
  22     class Media:
  23         js = (os.path.join(URL_FILEBROWSER_MEDIA, 'js/AddFileBrowser.js'), )
  24
  25     def __init__(self, attrs=None):
  26         self.directory = attrs.get('directory', '')
  27         self.extensions = attrs.get('extensions', '')
  28         self.format = attrs.get('format', '')
  29         if attrs is not None:
  30             self.attrs = attrs.copy()
  31         else:
  32             self.attrs = {}
  33
  34     def render(self, name, value, attrs=None):
  35         if value is None:
  36             value = ""
  37         final_attrs = self.build_attrs(attrs, type=self.input_type, name=name)
  38         final_attrs['search_icon'] = URL_FILEBROWSER_MEDIA + 'img/filebrowser_icon_show.gif'
  39         final_attrs['directory'] = self.directory
  40         final_attrs['extensions'] = self.extensions
  41         final_attrs['format'] = self.format
  42         final_attrs['ADMIN_THUMBNAIL'] = ADMIN_THUMBNAIL
  43         final_attrs['DEBUG'] = DEBUG
  44         if value != "":
  45             try:
  46                 final_attrs['directory'] = os.path.split(value.path_relative_directory)[0]
  47             except:
  48                 pass
  49         return render_to_string("filebrowser/custom_field.html", locals())
  50
  51
  52 class FileBrowseFormField(forms.CharField):
  53     widget = FileBrowseWidget
  54
  55     default_error_messages = {
  56         'extension': _(u'Extension %(ext)s is not allowed. Only %(allowed)s is allowed.'),
  57     }
  58
  59     def __init__(self, max_length=None, min_length=None,
  60                  directory=None, extensions=None, format=None,
  61                  *args, **kwargs):
  62         self.max_length, self.min_length = max_length, min_length
  63         self.directory = directory
  64         self.extensions = extensions
  65         if format:
  66             self.format = format or ''
  67             self.extensions = extensions or EXTENSIONS.get(format)
  68         super(FileBrowseFormField, self).__init__(*args, **kwargs)
  69
  70     def clean(self, value):
  71         value = super(FileBrowseFormField, self).clean(value)
  72         if value == '':
  73             return value
  74         file_extension = os.path.splitext(value)[1].lower()
  75         if self.extensions and not file_extension in self.extensions:
  76             raise forms.ValidationError(self.error_messages['extension'] % {'ext': file_extension, 'allowed': ", ".join(self.extensions)})
  77         return value
  78
  79
  80 class FileBrowseField(Field):
  81     __metaclass__ = models.SubfieldBase
  82
  83     def __init__(self, *args, **kwargs):
  84         self.directory = kwargs.pop('directory', '')
  85         self.extensions = kwargs.pop('extensions', '')
  86         self.format = kwargs.pop('format', '')
  87         return super(FileBrowseField, self).__init__(*args, **kwargs)
  88
  89     def to_python(self, value):
  90         if not value or isinstance(value, FileObject):
  91             return value
  92         return FileObject(_url_to_path(value))
  93
  94     def get_db_prep_value(self, value):
  95         if value is None:
  96             return None
  97         return unicode(value)
  98
  99
 100     def get_manipulator_field_objs(self):
 101         return [oldforms.TextField]
 102
 103     def get_internal_type(self):
 104         return "CharField"
 105
 106     def formfield(self, **kwargs):
 107         attrs = {}
 108         attrs["directory"] = self.directory
 109         attrs["extensions"] = self.extensions
 110         attrs["format"] = self.format
 111         defaults = {
 112             'form_class': FileBrowseFormField,
 113             'widget': FileBrowseWidget(attrs=attrs),
 114             'directory': self.directory,
 115             'extensions': self.extensions,
 116             'format': self.format
 117         }
 118         defaults.update(kwargs)
 119         return super(FileBrowseField, self).formfield(**defaults)
 120
 121