Created
February 7, 2013 16:39
-
-
Save codeinthehole/4732233 to your computer and use it in GitHub Desktop.
Basic auth for Django snippet
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import base64 | |
from django.http import HttpResponse | |
from django.contrib.auth import authenticate | |
from django.conf import settings | |
def view_or_basicauth(view, request, *args, **kwargs): | |
# Check for valid basic auth header | |
if 'HTTP_AUTHORIZATION' in request.META: | |
auth = request.META['HTTP_AUTHORIZATION'].split() | |
if len(auth) == 2: | |
if auth[0].lower() == "basic": | |
uname, passwd = base64.b64decode(auth[1]).split(':') | |
user = authenticate(username=uname, password=passwd) | |
if user is not None and user.is_active: | |
request.user = user | |
return view(request, *args, **kwargs) | |
# Either they did not provide an authorization header or | |
# something in the authorization attempt failed. Send a 401 | |
# back to them to ask them to authenticate. | |
response = HttpResponse() | |
response.status_code = 401 | |
response['WWW-Authenticate'] = 'Basic realm="%s"' % settings.BASIC_AUTH_REALM | |
return response | |
def basicauth(view_func): | |
def wrapper(request, *args, **kwargs): | |
return view_or_basicauth(view_func, request, *args, **kwargs) | |
return wrapper(tws) |
What is "tws" here in the last line "return wrapper(tws)" ?
That last function might be better like this:
def basicauth(view_func):
def wrapper(request, *args, **kwargs):
return view_or_basicauth(view_func, request, *args, **kwargs)
wrapper.__name__ = view_func.__name__
return wrapper
Or even better
from functools import wraps
def basicauth(view_func):
@wraps(view_func)
def wrapper(request, *args, **kwargs):
return view_or_basicauth(view_func, request, *args, **kwargs)
return wrapper
note that for python 3, you will have to do
user = authenticate(
username=uname.decode("utf-8"),
password=passwd.decode("utf-8"),
)
otherwise the (not decoded, i.e. bytes) values of uname
and passwd
won't be accepted in the authenticate
call (i.e. they won't match any user!)
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi,
A decode('utf-8') needs to follow b64decode() before using the split() function for strings:
uname,passwd = base64.b64decode(auth[1]).decode('utf-8').split(':')