import hashlib from django.contrib.auth.models import User from django.core.cache import cache 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 = hashlib.md5(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 hashlib.sha1("".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