apps/filebrowser/functions.py

   1 # -*- coding: utf-8
   2
   3 import os
   4 import re
   5 import decimal
   6
   7 from django.utils.translation import ugettext as _
   8 from django.utils.safestring import mark_safe
   9 from time import gmtime, strftime, localtime, mktime, time
  10 from django.core.files import File
  11 from django.core.files.storage import default_storage
  12
  13 from urlparse import urlparse
  14
  15 # filebrowser imports
  16 from filebrowser.fb_settings import *
  17
  18 # PIL import
  19 if STRICT_PIL:
  20     from PIL import Image
  21 else:
  22     try:
  23         from PIL import Image
  24     except ImportError:
  25         import Image
  26
  27
  28 def _url_to_path(value):
  29     """
  30     Change URL to PATH.
  31     Value has to be an URL relative to MEDIA URL or a full URL (including MEDIA_URL).
  32
  33     Returns a PATH relative to MEDIA_ROOT.
  34     """
  35     print "URL2PATH", repr(value)
  36     mediaurl_re = re.compile(r'^(%s)' % (MEDIA_URL))
  37     value = mediaurl_re.sub(u'', value)
  38     return value
  39
  40
  41 def _path_to_url(value):
  42     """
  43     Change PATH to URL.
  44     Value has to be a PATH relative to MEDIA_ROOT.
  45
  46     Return an URL relative to MEDIA_ROOT.
  47     """
  48     mediaroot_re = re.compile(r'^(%s)' % (MEDIA_ROOT))
  49     value = mediaroot_re.sub('', value)
  50     return _url_join(MEDIA_URL, value)
  51
  52
  53 def _dir_from_url(value):
  54     """
  55     Get the relative server directory from a URL.
  56     URL has to be an absolute URL including MEDIA_URL or
  57     an URL relative to MEDIA_URL.
  58     """
  59
  60     mediaurl_re = re.compile(r'^(%s)' % (MEDIA_URL))
  61     value = mediaurl_re.sub('', value)
  62     directory_re = re.compile(r'^(%s)' % (DIRECTORY))
  63     value = directory_re.sub('', value)
  64     return os.path.split(value)[0]
  65
  66
  67 def _get_version_path(value, version_prefix):
  68     """
  69     Construct the PATH to an Image version.
  70     Value has to be server-path, relative to MEDIA_ROOT.
  71
  72     version_filename = filename + version_prefix + ext
  73     Returns a path relative to MEDIA_ROOT.
  74     """
  75     if os.path.isfile(os.path.join(MEDIA_ROOT, value)):
  76         path, filename = os.path.split(value)
  77         filename, ext = os.path.splitext(filename)
  78         version_filename = filename + u"_" + version_prefix + ext
  79         return os.path.join(VERSIONS_BASEDIR, path, version_filename)
  80     else:
  81         return None
  82
  83
  84 def _url_join(*args):
  85     if args[0].startswith(u"http://"):
  86         url = u"http://"
  87     else:
  88         url = u"/"
  89     for arg in args:
  90         arg = unicode(arg).replace(u"\\", u"/")
  91         arg_split = arg.split(u"/")
  92         for elem in arg_split:
  93             if elem != u"" and elem != u"http:":
  94                 url = url + elem + u"/"
  95     # remove trailing slash for filenames
  96     if os.path.splitext(args[-1])[1]:
  97         url = url.rstrip(u"/")
  98     return url
  99
 100
 101 def _get_path(path):
 102     """
 103     Get Path.
 104     """
 105
 106     if os.path.isabs(path) or not os.path.isdir(os.path.join(MEDIA_ROOT, DIRECTORY, path)):
 107         return None
 108     return path
 109
 110
 111 def _get_file(path, filename):
 112     """
 113     Get File.
 114     """
 115
 116     if not os.path.isfile(os.path.join(MEDIA_ROOT, DIRECTORY, path, filename)) and not os.path.isdir(os.path.join(MEDIA_ROOT, DIRECTORY, path, filename)):
 117         return None
 118     return filename
 119
 120
 121 def _get_breadcrumbs(query, path, title):
 122     """
 123     Get breadcrumbs.
 124     """
 125
 126     breadcrumbs = []
 127     dir_query = ""
 128     if path:
 129         for item in path.split(os.sep):
 130             dir_query = os.path.join(dir_query, item)
 131             breadcrumbs.append([item, dir_query])
 132     if title:
 133         breadcrumbs.append([title, ''])
 134     return breadcrumbs
 135
 136
 137 def _get_filterdate(filterDate, dateTime):
 138     """
 139     Get filterdate.
 140     """
 141
 142     dateYear = strftime("%Y", gmtime(dateTime))
 143     dateMonth = strftime("%m", gmtime(dateTime))
 144     dateDay = strftime("%d", gmtime(dateTime))
 145
 146     if filterDate == 'today' and int(dateYear) == int(localtime()[0]) and int(dateMonth) == int(localtime()[1]) and int(dateDay) == int(localtime()[2]):
 147         return 'true'
 148
 149     if filterDate == 'thismonth' and dateTime >= time() - 2592000:
 150         return 'true'
 151
 152     if filterDate == 'thisyear' and int(dateYear) == int(localtime()[0]):
 153         return 'true'
 154
 155     if filterDate == 'past7days' and dateTime >= time() - 604800:
 156         return 'true'
 157
 158     return 'true' if filterDate == '' else ''
 159
 160
 161 def _get_settings_var():
 162     """
 163     Get settings variables used for FileBrowser listing.
 164     """
 165
 166     settings_var = {}
 167     # Main
 168     settings_var['DEBUG'] = DEBUG
 169     settings_var['MEDIA_ROOT'] = MEDIA_ROOT
 170     settings_var['MEDIA_URL'] = MEDIA_URL
 171     settings_var['DIRECTORY'] = DIRECTORY
 172     # FileBrowser
 173     settings_var['URL_FILEBROWSER_MEDIA'] = URL_FILEBROWSER_MEDIA
 174     settings_var['PATH_FILEBROWSER_MEDIA'] = PATH_FILEBROWSER_MEDIA
 175     # TinyMCE
 176     settings_var['URL_TINYMCE'] = URL_TINYMCE
 177     settings_var['PATH_TINYMCE'] = PATH_TINYMCE
 178     # Extensions/Formats (for FileBrowseField)
 179     settings_var['EXTENSIONS'] = EXTENSIONS
 180     settings_var['SELECT_FORMATS'] = SELECT_FORMATS
 181     # Versions
 182     settings_var['VERSIONS_BASEDIR'] = VERSIONS_BASEDIR
 183     settings_var['VERSIONS'] = VERSIONS
 184     settings_var['ADMIN_VERSIONS'] = ADMIN_VERSIONS
 185     settings_var['ADMIN_THUMBNAIL'] = ADMIN_THUMBNAIL
 186     # FileBrowser Options
 187     settings_var['MAX_UPLOAD_SIZE'] = MAX_UPLOAD_SIZE
 188     # Convert Filenames
 189     settings_var['CONVERT_FILENAME'] = CONVERT_FILENAME
 190     return settings_var
 191
 192
 193 def _handle_file_upload(path, file):
 194     """
 195     Handle File Upload.
 196     """
 197
 198     file_path = os.path.join(path, file.name)
 199     uploadedfile = default_storage.save(file_path, file)
 200     return uploadedfile
 201
 202
 203 def _get_file_type(filename):
 204     """
 205     Get file type as defined in EXTENSIONS.
 206     """
 207     file_extension = os.path.splitext(filename)[1].lower()
 208     for ftype, ext_list in EXTENSIONS.iteritems():
 209         if file_extension in map(str.lower, ext_list):
 210             return ftype
 211     return u''
 212
 213
 214 def _is_selectable(filename, selecttype):
 215     """
 216     Get select type as defined in FORMATS.
 217     """
 218
 219     file_extension = os.path.splitext(filename)[1].lower()
 220     select_types = []
 221     for k, v in SELECT_FORMATS.iteritems():
 222         for extension in v:
 223             if file_extension == extension.lower():
 224                 select_types.append(k)
 225     return select_types
 226
 227
 228 def _version_generator(value, version_prefix, force=None):
 229     """
 230     Generate Version for an Image.
 231     value has to be a serverpath relative to MEDIA_ROOT.
 232     """
 233
 234     # PIL's Error "Suspension not allowed here" work around:
 235     # s. http://mail.python.org/pipermail/image-sig/1999-August/000816.html
 236     if STRICT_PIL:
 237         from PIL import ImageFile
 238     else:
 239         try:
 240             from PIL import ImageFile
 241         except ImportError:
 242             import ImageFile
 243     ImageFile.MAXBLOCK = IMAGE_MAXBLOCK  # default is 64k
 244
 245     try:
 246         im = Image.open(os.path.join(MEDIA_ROOT, value))
 247         version_path = _get_version_path(value, version_prefix)
 248         absolute_version_path = os.path.join(MEDIA_ROOT, version_path)
 249         version_dir = os.path.split(absolute_version_path)[0]
 250         if not os.path.isdir(version_dir):
 251             os.makedirs(version_dir)
 252             os.chmod(version_dir, 0775)
 253         version = scale_and_crop(im, VERSIONS[version_prefix]['width'], VERSIONS[version_prefix]['height'], VERSIONS[version_prefix]['opts'])
 254         try:
 255             version.save(absolute_version_path, quality=90, optimize=1)
 256         except IOError:
 257             version.save(absolute_version_path, quality=90)
 258         return version_path
 259     except:
 260         return None
 261
 262
 263 def scale_and_crop(im, width, height, opts):
 264     x, y = [float(v) for v in im.size]
 265     if width:
 266         xr = float(width)
 267     else:
 268         xr = float(x * height / y)
 269     if height:
 270         yr = float(height)
 271     else:
 272         yr = float(y * width / x)
 273
 274     if 'crop' in opts:
 275         r = max(xr / x, yr / y)
 276     else:
 277         r = min(xr / x, yr / y)
 278
 279     if r < 1.0 or (r > 1.0 and 'upscale' in opts):
 280         im = im.resize((int(x * r), int(y * r)), resample=Image.ANTIALIAS)
 281
 282     if 'crop' in opts:
 283         x, y = [float(v) for v in im.size]
 284         ex, ey = (x - min(x, xr)) / 2, (y - min(y, yr)) / 2
 285         if ex or ey:
 286             im = im.crop((int(ex), int(ey), int(x - ex), int(y - ey)))
 287     return im
 288 scale_and_crop.valid_options = ('crop', 'upscale')
 289
 290
 291 def _convert_filename(value):
 292     if CONVERT_FILENAME:
 293         return value.replace(" ", "_").lower()
 294     else:
 295         return value