apps/filebrowser/functions.py

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