mirror of
https://github.com/samuelclay/NewsBlur.git
synced 2025-04-13 09:42:01 +00:00
73 lines
No EOL
2.3 KiB
Python
73 lines
No EOL
2.3 KiB
Python
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] |