user_functions.py

from django.utils.hashcompat import sha_constructor
from django.contrib.auth.models import User
from django.core.cache import cache
from django.utils.hashcompat import md5_constructor
from django.utils.http import urlquote
from django.http import HttpResponseForbidden
from django.conf import settings

def ajax_login_required(function=None):
    def _dec(view_func):
        def _view(request, *args, **kwargs):
            if request.user.is_anonymous():
                return HttpResponseForbidden()
            else:
                return view_func(request, *args, **kwargs)

        _view.__name__ = view_func.__name__
        _view.__dict__ = view_func.__dict__
        _view.__doc__ = view_func.__doc__

        return _view

    if function is None:
        return _dec
    else:
        return _dec(function)

def admin_only(function=None):
    def _dec(view_func):
        def _view(request, *args, **kwargs):
            if not request.user.is_staff:
                return HttpResponseForbidden()
            else:
                return view_func(request, *args, **kwargs)

        _view.__name__ = view_func.__name__
        _view.__dict__ = view_func.__dict__
        _view.__doc__ = view_func.__doc__

        return _view

    if function is None:
        return _dec
    else:
        return _dec(function)
        
def get_user(request):
    if not hasattr(request, 'user'):
        user = request
    else:
        user = request.user
        
    if user.is_anonymous():
        user = cache.get('user:%s' % settings.HOMEPAGE_USERNAME, None)
        if not user:
            try:
                user = User.objects.get(username=settings.HOMEPAGE_USERNAME)
                cache.set('user:%s' % user, user)
            except User.DoesNotExist:
                user = User.objects.create(username=settings.HOMEPAGE_USERNAME)
                user.set_password('')
                user.save()
    return user
    
def invalidate_template_cache(fragment_name, *variables):
    args = md5_constructor(u':'.join([urlquote(var) for var in variables]))
    cache_key = 'template.cache.%s.%s' % (fragment_name, args.hexdigest())
    cache.delete(cache_key)
    
def generate_secret_token(phrase, size=12):
    """Generate a (SHA1) security hash from the provided info."""
    info = (phrase, settings.SECRET_KEY)
    return sha_constructor("".join(info)).hexdigest()[:size]

def extract_user_agent(request):
    user_agent = request.environ.get('HTTP_USER_AGENT', '')
    platform = '------'
    if 'iPad App' in user_agent:
        platform = 'iPad'
    elif 'iPhone App' in user_agent:
        platform = 'iPhone'
    elif 'Blar' in user_agent:
        platform = 'Blar'
    elif 'Android' in user_agent:
        platform = 'Androd'
    elif 'Metroblur' in user_agent:
        platform = 'Metrob'
    elif 'Pluggio' in user_agent:
        platform = 'Plugio'
    elif 'MSIE' in user_agent:
        platform = 'IE'
        if 'MSIE 9' in user_agent:
            platform += '9'
        elif 'MSIE 10' in user_agent:
            platform += '10'
        elif 'MSIE 8' in user_agent:
            platform += '8'
    elif 'Chrome' in user_agent:
        platform = 'Chrome'
    elif 'Safari' in user_agent:
        platform = 'Safari'
    elif 'MeeGo' in user_agent:
        platform = 'MeeGo'
    elif 'Firefox' in user_agent:
        platform = 'FF'
    elif 'Opera' in user_agent:
        platform = 'Opera'
    elif 'WP7' in user_agent:
        platform = 'WP7'
    elif 'WP8' in user_agent:
        platform = 'WP8'
    elif 'Tafiti' in user_agent:
        platform = 'Tafiti'
    
    return platform