Dodanie aplikacji sorl.thumbnail i filebrowser.
[redakcja.git] / apps / filebrowser / functions.py
diff --git a/apps/filebrowser/functions.py b/apps/filebrowser/functions.py
new file mode 100755 (executable)
index 0000000..cef9dcd
--- /dev/null
@@ -0,0 +1,312 @@
+# coding: utf-8
+
+from django.utils.translation import ugettext as _
+from django.utils.safestring import mark_safe
+from time import gmtime, strftime, localtime, mktime, time
+from django.core.files import File
+from django.core.files.storage import default_storage
+import os, re, decimal
+from urlparse import urlparse
+
+# filebrowser imports
+from filebrowser.fb_settings import *
+
+# PIL import
+if STRICT_PIL:
+    from PIL import Image
+else:
+    try:
+        from PIL import Image
+    except ImportError:
+        import Image
+
+
+def _url_to_path(value):
+    """
+    Change URL to PATH.
+    Value has to be an URL relative to MEDIA URL or a full URL (including MEDIA_URL).
+    
+    Returns a PATH relative to MEDIA_ROOT.
+    """
+    mediaurl_re = re.compile(r'^(%s)' % (MEDIA_URL))
+    value = mediaurl_re.sub('', value)
+    return value
+    
+
+def _path_to_url(value):
+    """
+    Change PATH to URL.
+    Value has to be a PATH relative to MEDIA_ROOT.
+    
+    Return an URL relative to MEDIA_ROOT.
+    """
+    mediaroot_re = re.compile(r'^(%s)' % (MEDIA_ROOT))
+    value = mediaroot_re.sub('', value)
+    return _url_join(MEDIA_URL, value)
+    
+
+def _dir_from_url(value):
+    """
+    Get the relative server directory from a URL.
+    URL has to be an absolute URL including MEDIA_URL or
+    an URL relative to MEDIA_URL.
+    """
+    
+    mediaurl_re = re.compile(r'^(%s)' % (MEDIA_URL))
+    value = mediaurl_re.sub('', value)
+    directory_re = re.compile(r'^(%s)' % (DIRECTORY))
+    value = directory_re.sub('', value)
+    return os.path.split(value)[0]
+    
+
+def _get_version_path(value, version_prefix):
+    """
+    Construct the PATH to an Image version.
+    Value has to be server-path, relative to MEDIA_ROOT.
+    
+    version_filename = filename + version_prefix + ext
+    Returns a path relative to MEDIA_ROOT.
+    """
+    if os.path.isfile(os.path.join(MEDIA_ROOT, value)):
+        path, filename = os.path.split(value)
+        filename, ext = os.path.splitext(filename)
+        version_filename = filename + "_" + version_prefix + ext
+        return os.path.join(VERSIONS_BASEDIR, path, version_filename)
+    else:
+        return None
+    
+
+def _sort_by_attr(seq, attr):
+    """
+    Sort the sequence of objects by object's attribute
+    
+    Arguments:
+    seq  - the list or any sequence (including immutable one) of objects to sort.
+    attr - the name of attribute to sort by
+    
+    Returns:
+    the sorted list of objects.
+    """
+    import operator
+    
+    # Use the "Schwartzian transform"
+    # Create the auxiliary list of tuples where every i-th tuple has form
+    # (seq[i].attr, i, seq[i]) and sort it. The second item of tuple is needed not
+    # only to provide stable sorting, but mainly to eliminate comparison of objects
+    # (which can be expensive or prohibited) in case of equal attribute values.
+    intermed = map(None, map(getattr, seq, (attr,)*len(seq)), xrange(len(seq)), seq)
+    intermed.sort()
+    return map(operator.getitem, intermed, (-1,) * len(intermed))
+    
+
+def _url_join(*args):
+    if args[0].startswith("http://"):
+        url = "http://"
+    else:
+        url = "/"
+    for arg in args:
+        arg = str(arg).replace("\\", "/")
+        arg_split = arg.split("/")
+        for elem in arg_split:
+            if elem != "" and elem != "http:":
+                url = url + elem + "/"
+    # remove trailing slash for filenames
+    if os.path.splitext(args[-1])[1]:
+        url = url.rstrip("/")
+    return url
+    
+
+def _get_path(path):
+    """
+    Get Path.
+    """
+    
+    if os.path.isabs(path) or not os.path.isdir(os.path.join(MEDIA_ROOT, DIRECTORY, path)):
+        return None
+    return path
+    
+
+def _get_file(path, filename):
+    """
+    Get File.
+    """
+    
+    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)):
+        return None
+    return filename
+    
+
+def _get_breadcrumbs(query, path, title):
+    """
+    Get breadcrumbs.
+    """
+    
+    breadcrumbs = []
+    dir_query = ""
+    if path:
+        for item in path.split(os.sep):
+            dir_query = os.path.join(dir_query,item)
+            breadcrumbs.append([item,dir_query])
+    if title:
+        breadcrumbs.append([title,''])
+    return breadcrumbs
+    
+
+def _get_filterdate(filterDate, dateTime):
+    """
+    Get filterdate.
+    """
+    
+    returnvalue = ''
+    dateYear = strftime("%Y", gmtime(dateTime))
+    dateMonth = strftime("%m", gmtime(dateTime))
+    dateDay = strftime("%d", gmtime(dateTime))
+    if filterDate == 'today' and int(dateYear) == int(localtime()[0]) and int(dateMonth) == int(localtime()[1]) and int(dateDay) == int(localtime()[2]): returnvalue = 'true'
+    elif filterDate == 'thismonth' and dateTime >= time()-2592000: returnvalue = 'true'
+    elif filterDate == 'thisyear' and int(dateYear) == int(localtime()[0]): returnvalue = 'true'
+    elif filterDate == 'past7days' and dateTime >= time()-604800: returnvalue = 'true'
+    elif filterDate == '': returnvalue = 'true'
+    return returnvalue
+    
+
+def _get_settings_var():
+    """
+    Get settings variables used for FileBrowser listing.
+    """
+    
+    settings_var = {}
+    # Main
+    settings_var['DEBUG'] = DEBUG
+    settings_var['MEDIA_ROOT'] = MEDIA_ROOT
+    settings_var['MEDIA_URL'] = MEDIA_URL
+    settings_var['DIRECTORY'] = DIRECTORY
+    # FileBrowser
+    settings_var['URL_FILEBROWSER_MEDIA'] = URL_FILEBROWSER_MEDIA
+    settings_var['PATH_FILEBROWSER_MEDIA'] = PATH_FILEBROWSER_MEDIA
+    # TinyMCE
+    settings_var['URL_TINYMCE'] = URL_TINYMCE
+    settings_var['PATH_TINYMCE'] = PATH_TINYMCE
+    # Extensions/Formats (for FileBrowseField)
+    settings_var['EXTENSIONS'] = EXTENSIONS
+    settings_var['SELECT_FORMATS'] = SELECT_FORMATS
+    # Versions
+    settings_var['VERSIONS_BASEDIR'] = VERSIONS_BASEDIR
+    settings_var['VERSIONS'] = VERSIONS
+    settings_var['ADMIN_VERSIONS'] = ADMIN_VERSIONS
+    settings_var['ADMIN_THUMBNAIL'] = ADMIN_THUMBNAIL
+    # FileBrowser Options
+    settings_var['MAX_UPLOAD_SIZE'] = MAX_UPLOAD_SIZE
+    # Convert Filenames
+    settings_var['CONVERT_FILENAME'] = CONVERT_FILENAME
+    return settings_var
+    
+
+def _handle_file_upload(path, file):
+    """
+    Handle File Upload.
+    """
+    
+    file_path = os.path.join(path, file.name)
+    uploadedfile = default_storage.save(file_path, file)
+    return uploadedfile
+    
+
+def _get_file_type(filename):
+    """
+    Get file type as defined in EXTENSIONS.
+    """
+    
+    file_extension = os.path.splitext(filename)[1].lower()
+    file_type = ''
+    for k,v in EXTENSIONS.iteritems():
+        for extension in v:
+            if file_extension == extension.lower():
+                file_type = k
+    return file_type
+    
+
+def _is_selectable(filename, selecttype):
+    """
+    Get select type as defined in FORMATS.
+    """
+    
+    file_extension = os.path.splitext(filename)[1].lower()
+    select_types = []
+    for k,v in SELECT_FORMATS.iteritems():
+        for extension in v:
+            if file_extension == extension.lower():
+                select_types.append(k)
+    return select_types
+    
+
+def _version_generator(value, version_prefix, force=None):
+    """
+    Generate Version for an Image.
+    value has to be a serverpath relative to MEDIA_ROOT.
+    """
+    
+    # PIL's Error "Suspension not allowed here" work around:
+    # s. http://mail.python.org/pipermail/image-sig/1999-August/000816.html
+    if STRICT_PIL:
+        from PIL import ImageFile
+    else:
+        try:
+            from PIL import ImageFile
+        except ImportError:
+            import ImageFile
+    ImageFile.MAXBLOCK = IMAGE_MAXBLOCK # default is 64k
+    
+    try:
+        im = Image.open(os.path.join(MEDIA_ROOT, value))
+        version_path = _get_version_path(value, version_prefix)
+        absolute_version_path = os.path.join(MEDIA_ROOT, version_path)
+        version_dir = os.path.split(absolute_version_path)[0]
+        if not os.path.isdir(version_dir):
+            os.makedirs(version_dir)
+            os.chmod(version_dir, 0775)
+        version = scale_and_crop(im, VERSIONS[version_prefix]['width'], VERSIONS[version_prefix]['height'], VERSIONS[version_prefix]['opts'])
+        try:
+            version.save(absolute_version_path, quality=90, optimize=1)
+        except IOError:
+            version.save(absolute_version_path, quality=90)
+        return version_path
+    except:
+        return None
+    
+
+def scale_and_crop(im, width, height, opts):
+    x, y   = [float(v) for v in im.size]
+    if width:
+        xr = float(width)
+    else:
+        xr = float(x*height/y)
+    if height:
+        yr = float(height)
+    else:
+        yr = float(y*width/x)
+    
+    if 'crop' in opts:
+        r = max(xr/x, yr/y)
+    else:
+        r = min(xr/x, yr/y)
+    
+    if r < 1.0 or (r > 1.0 and 'upscale' in opts):
+        im = im.resize((int(x*r), int(y*r)), resample=Image.ANTIALIAS)
+    
+    if 'crop' in opts:
+        x, y   = [float(v) for v in im.size]
+        ex, ey = (x-min(x, xr))/2, (y-min(y, yr))/2
+        if ex or ey:
+            im = im.crop((int(ex), int(ey), int(x-ex), int(y-ey)))
+    return im
+scale_and_crop.valid_options = ('crop', 'upscale')
+
+
+def _convert_filename(value):
+    if CONVERT_FILENAME:
+        return value.replace(" ", "_").lower()
+    else:
+        return value
+        
+    
+