files.py

import os
import json
import pickle
from random import randint
from django.conf import settings
from django.core.files.storage import FileSystemStorage
from django.contrib.staticfiles import finders


class OverwriteStorage(FileSystemStorage):

    def get_available_name(self, name, max_length=None):
        """
        Returns a filename that's free on the target storage system, and
        available for new content to be written to.
        This file storage solves overwrite on upload problem.
        Found at http://djangosnippets.org/snippets/976/
        """
        if self.exists(name):
            os.remove(os.path.join(settings.MEDIA_ROOT, name))
        return name


def upload_pic(uploaded_file, to_file):
    destination = open(os.path.join(settings.MEDIA_ROOT, to_file), 'wb')
    for chunk in uploaded_file.chunks():
        destination.write(chunk)
    destination.close()


def read_all(path):
    with open(path) as f:
        return f.read()


def read_lines(f, remove_empty=False):
    lines = f.read().splitlines()
    lines = [line.strip() for line in lines]
    if remove_empty: lines = [s for s in lines if s]
    return lines


def get_extension(filename):
    parts = filename.split('.')
    return parts[-1].lower()


def random_filename(prefix, extension):
    while True:
        out_file = '%s_%d.%s' % (prefix, randint(0, 1000000), extension)
        if not os.path.exists(out_file):
            return out_file


def read_static_file(path):
    path = finders.find(path)
    with open(path) as f:
        return f.read()


def get_extension(filename):
    ext = filename.rsplit('.', maxsplit=1)
    return ext[1].lower() if len(ext) == 2 else None


def quick_write_json(filename, data):
    with open(filename, 'w') as f:
        f.write(json.dumps(data))


def quick_write_bin(filename, data):
    with open(filename, 'wb') as f:
        pickle.dump(data, f)


def quick_read_json(filename):
    with open(filename) as f:
        return json.loads(f.read())


def quick_read_bin(filename):
    with open(filename, 'rb') as f:
        return pickle.load(f)