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