Nie powiększanie pierwszej litery w tytule.
[redakcja.git] / apps / filebrowser / base.py
1 # coding: utf-8
2
3 import os, re, datetime
4 from time import gmtime, strftime
5 from django.conf import settings
6
7 # filebrowser imports
8 from filebrowser.fb_settings import *
9 from filebrowser.functions import _get_file_type, _url_join, _is_selectable, _get_version_path
10
11 # PIL import
12 if STRICT_PIL:
13     from PIL import Image
14 else:
15     try:
16         from PIL import Image
17     except ImportError:
18         import Image
19     
20
21 class FileObject(object):
22     """
23     The FileObject represents a File on the Server.
24     
25     PATH has to be relative to MEDIA_ROOT.
26     """
27     
28     def __init__(self, path):
29         self.path = path
30         self.head = os.path.split(path)[0]
31         self.filename = os.path.split(path)[1]
32         self.filename_lower = self.filename.lower() # important for sorting
33         self.filetype = _get_file_type(self.filename)
34     
35     def _filesize(self):
36         """
37         Filesize.
38         """
39         if os.path.isfile(os.path.join(MEDIA_ROOT, self.path)) or os.path.isdir(os.path.join(MEDIA_ROOT, self.path)):
40             return os.path.getsize(os.path.join(MEDIA_ROOT, self.path))
41         return ""
42     filesize = property(_filesize)
43         
44     def _date(self):
45         """
46         Date.
47         """
48         if os.path.isfile(os.path.join(MEDIA_ROOT, self.path)) or os.path.isdir(os.path.join(MEDIA_ROOT, self.path)):
49             return os.path.getmtime(os.path.join(MEDIA_ROOT, self.path))
50         return ""
51     date = property(_date)
52     
53     def _datetime(self):
54         """
55         Datetime Object.
56         """
57         return datetime.datetime.fromtimestamp(self.date)
58     datetime = property(_datetime)
59     
60     def _extension(self):
61         """
62         Extension.
63         """
64         return u"%s" % os.path.splitext(self.filename)[1]
65     extension = property(_extension)
66     
67     def _filetype_checked(self):
68         if self.filetype == "Folder" and os.path.isdir(self.path_full):
69             return self.filetype
70         elif self.filetype != "Folder" and os.path.isfile(self.path_full):
71             return self.filetype
72         else:
73             return ""
74     filetype_checked = property(_filetype_checked)
75     
76     def _path_full(self):
77         """
78         Full server PATH including MEDIA_ROOT.
79         """
80         return u"%s" % os.path.join(MEDIA_ROOT, self.path)
81     path_full = property(_path_full)
82     
83     def _path_relative(self):
84         return self.path
85     path_relative = property(_path_relative)
86     
87     def _path_relative_directory(self):
88         """
89         Path relative to initial directory.
90         """
91         directory_re = re.compile(r'^(%s)' % (DIRECTORY))
92         value = directory_re.sub('', self.path)
93         return u"%s" % value
94     path_relative_directory = property(_path_relative_directory)
95     
96     def _url_relative(self):
97         return self.path
98     url_relative = property(_url_relative)
99     
100     def _url_full(self):
101         """
102         Full URL including MEDIA_URL.
103         """
104         return u"%s" % _url_join(MEDIA_URL, self.path)
105     url_full = property(_url_full)
106     
107     def _url_save(self):
108         """
109         URL used for the filebrowsefield.
110         """
111         if SAVE_FULL_URL:
112             return self.url_full
113         else:
114             return self.path
115     url_save = property(_url_save)
116     
117     def _url_thumbnail(self):
118         """
119         Thumbnail URL.
120         """
121         if self.filetype == "Image":
122             return u"%s" % _url_join(MEDIA_URL, _get_version_path(self.path, 'fb_thumb'))
123         else:
124             return ""
125     url_thumbnail = property(_url_thumbnail)
126     
127     def url_admin(self):
128         if self.filetype_checked == "Folder":
129             directory_re = re.compile(r'^(%s)' % (DIRECTORY))
130             value = directory_re.sub('', self.path)
131             return u"%s" % value
132         else:
133             return u"%s" % _url_join(MEDIA_URL, self.path)
134     
135     def _dimensions(self):
136         """
137         Image Dimensions.
138         """
139         if self.filetype == 'Image':
140             try:
141                 im = Image.open(os.path.join(MEDIA_ROOT, self.path))
142                 return im.size
143             except:
144                 pass
145         else:
146             return False
147     dimensions = property(_dimensions)
148     
149     def _width(self):
150         """
151         Image Width.
152         """
153         return self.dimensions[0]
154     width = property(_width)
155     
156     def _height(self):
157         """
158         Image Height.
159         """
160         return self.dimensions[1]
161     height = property(_height)
162     
163     def _orientation(self):
164         """
165         Image Orientation.
166         """
167         if self.dimensions:
168             if self.dimensions[0] >= self.dimensions[1]:
169                 return "Landscape"
170             else:
171                 return "Portrait"
172         else:
173             return None
174     orientation = property(_orientation)
175     
176     def _is_empty(self):
177         """
178         True if Folder is empty, False if not.
179         """
180         if os.path.isdir(self.path_full):
181             if not os.listdir(self.path_full):
182                 return True
183             else:
184                 return False
185         else:
186             return None
187     is_empty = property(_is_empty)
188     
189     def __repr__(self):
190         return u"%s" % self.url_save
191     
192     def __str__(self):
193         return u"%s" % self.url_save
194     
195     def __unicode__(self):
196         return u"%s" % self.url_save
197     
198
199