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
  20 class FileBrowseWidget(Input):
  21     input_type = 'text'
  22
  23     class Media:
  24         js = (os.path.join(URL_FILEBROWSER_MEDIA, 'js/AddFileBrowser.js'),)
  25
  26     def __init__(self, attrs=None):
  27         self.directory = attrs.get('directory', '')
  28         self.extensions = attrs.get('extensions', '')
  29         self.format = attrs.get('format', '')
  30         if attrs is not None:
  31             self.attrs = attrs.copy()
  32         else:
  33             self.attrs = {}
  34
  35     def render(self, name, value, attrs=None):
  36         if value is None:
  37             value = ""
  38         final_attrs = self.build_attrs(attrs, type=self.input_type, name=name)
  39         final_attrs['search_icon'] = URL_FILEBROWSER_MEDIA + 'img/filebrowser_icon_show.gif'
  40         final_attrs['directory'] = self.directory
  41         final_attrs['extensions'] = self.extensions
  42         final_attrs['format'] = self.format
  43         final_attrs['ADMIN_THUMBNAIL'] = ADMIN_THUMBNAIL
  44         final_attrs['DEBUG'] = DEBUG
  45         if value != "":
  46             try:
  47                 final_attrs['directory'] = os.path.split(value.path_relative_directory)[0]
  48             except:
  49                 pass
  50         return render_to_string("filebrowser/custom_field.html", locals())
  51
  52
  53 class FileBrowseFormField(forms.CharField):
  54     widget = FileBrowseWidget
  55
  56     default_error_messages = {
  57         'extension': _(u'Extension %(ext)s is not allowed. Only %(allowed)s is allowed.'),
  58     }
  59
  60     def __init__(self, max_length=None, min_length=None,
  61                  directory=None, extensions=None, format=None,
  62                  *args, **kwargs):
  63         self.max_length, self.min_length = max_length, min_length
  64         self.directory = directory
  65         self.extensions = extensions
  66         if format:
  67             self.format = format or ''
  68             self.extensions = extensions or EXTENSIONS.get(format)
  69         super(FileBrowseFormField, self).__init__(*args, **kwargs)
  70
  71     def clean(self, value):
  72         value = super(FileBrowseFormField, self).clean(value)
  73         if value == '':
  74             return value
  75         file_extension = os.path.splitext(value)[1].lower()
  76         if self.extensions and not file_extension in self.extensions:
  77             raise forms.ValidationError(self.error_messages['extension'] % {'ext': file_extension, 'allowed': ", ".join(self.extensions)})
  78         return value
  79
  80
  81 class FileBrowseField(Field):
  82     __metaclass__ = models.SubfieldBase
  83
  84     def __init__(self, *args, **kwargs):
  85         self.directory = kwargs.pop('directory', '')
  86         self.extensions = kwargs.pop('extensions', '')
  87         self.format = kwargs.pop('format', '')
  88         return super(FileBrowseField, self).__init__(*args, **kwargs)
  89
  90     def to_python(self, value):
  91         if not value or isinstance(value, FileObject):
  92             return value
  93         return FileObject(_url_to_path(value))
  94
  95     def get_db_prep_value(self, value):
  96         if value is None:
  97             return None
  98         return unicode(value)
  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)