85e981b9ea79dc7d79ff1f2745d2e289a2049952
[redakcja.git] / 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