snipt/snipts/utils.py

import uuid

from django.contrib.auth import authenticate, login
from django.template.defaultfilters import slugify
from pygments.lexers import get_all_lexers
from registration.signals import user_registered


def slugify_uniquely(value, model, slugfield="slug"):
    suffix = False
    potential = base = slugify(value)[:255]

    while True:
        if suffix:
            if value:
                potential = "-".join([base, str(suffix)])
            else:
                potential = str(suffix)
        if not model.objects.filter(**{slugfield: potential}).count():
            return potential
        suffix = str(uuid.uuid4()).split('-')[0]


def activate_user(user, request, **kwargs):
    user = authenticate(username=request.POST['username'],
                        password=request.POST['password1'])
    login(request, user)


def get_lexers_list():
    lexers = list(get_all_lexers())

    for l in lexers:
        if l[0] == 'ANTLR With Java Target':
            lexers.remove(l)

    lexers.append(('Markdown', ('markdown',),))
    lexers = sorted(lexers)

    return lexers

user_registered.connect(activate_user)