pasteque/paste/tools.py

import string
import random
import shortuuid
import os
from webtools import settings
from .models import Paste


def random_id(model):
    """Returns a short uuid for the slug of the given model."""
    uuid = random.choice('0123456789') + shortuuid.uuid()
    for i in range(3, len(uuid)):
        potential_uuid = uuid[:i]
        if not model.objects.filter(slug=potential_uuid):
            return potential_uuid
    return uuid


def cache_get_filepath(key):
    """Returns cache path."""
    return os.path.join(settings.CACHE_PATH, key)


def cache_exists(key):
    """Says if cache exists for key."""
    return os.path.isfile(cache_get_filepath(key))


def cache_store(key, value):
    """Store cache value for key."""
    with open(cache_get_filepath(key), 'w') as cache_file:
        cache_file.write(value)


def cache_fetch(key):
    """Fetch cache value for key."""
    with open(cache_get_filepath(key), 'r') as cache_file:
        return cache_file.read()