mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-09-22 03:49:06 +03:00
100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
"""
|
|
The ``authentication`` module provides a set of pluggable authentication classes.
|
|
|
|
Authentication behavior is provided by adding the ``AuthMixin`` class to a ``View`` .
|
|
|
|
The set of authentication methods which are used is then specified by setting
|
|
``authentication`` attribute on the ``View`` class, and listing a set of authentication classes.
|
|
"""
|
|
|
|
from django.contrib.auth import authenticate
|
|
from django.middleware.csrf import CsrfViewMiddleware
|
|
from djangorestframework.utils import as_tuple
|
|
import base64
|
|
|
|
__all__ = (
|
|
'BaseAuthenticaton',
|
|
'BasicAuthenticaton',
|
|
'UserLoggedInAuthenticaton'
|
|
)
|
|
|
|
|
|
class BaseAuthenticaton(object):
|
|
"""
|
|
All authentication classes should extend BaseAuthentication.
|
|
"""
|
|
|
|
def __init__(self, view):
|
|
"""
|
|
Authentication classes are always passed the current view on creation.
|
|
"""
|
|
self.view = view
|
|
|
|
def authenticate(self, request):
|
|
"""
|
|
Authenticate the request and return a ``User`` instance or None. (*)
|
|
|
|
This function must be overridden to be implemented.
|
|
|
|
(*) The authentication context _will_ typically be a ``User`` object,
|
|
but it need not be. It can be any user-like object so long as the
|
|
permissions classes on the view can handle the object and use
|
|
it to determine if the request has the required permissions or not.
|
|
|
|
This can be an important distinction if you're implementing some token
|
|
based authentication mechanism, where the authentication context
|
|
may be more involved than simply mapping to a ``User``.
|
|
"""
|
|
return None
|
|
|
|
|
|
class BasicAuthenticaton(BaseAuthenticaton):
|
|
"""
|
|
Use HTTP Basic authentication.
|
|
"""
|
|
|
|
def authenticate(self, request):
|
|
from django.utils.encoding import smart_unicode, DjangoUnicodeDecodeError
|
|
|
|
if 'HTTP_AUTHORIZATION' in request.META:
|
|
auth = request.META['HTTP_AUTHORIZATION'].split()
|
|
if len(auth) == 2 and auth[0].lower() == "basic":
|
|
try:
|
|
auth_parts = base64.b64decode(auth[1]).partition(':')
|
|
except TypeError:
|
|
return None
|
|
|
|
try:
|
|
uname, passwd = smart_unicode(auth_parts[0]), smart_unicode(auth_parts[2])
|
|
except DjangoUnicodeDecodeError:
|
|
return None
|
|
|
|
user = authenticate(username=uname, password=passwd)
|
|
if user is not None and user.is_active:
|
|
return user
|
|
return None
|
|
|
|
|
|
class UserLoggedInAuthenticaton(BaseAuthenticaton):
|
|
"""
|
|
Use Django's session framework for authentication.
|
|
"""
|
|
|
|
def authenticate(self, request):
|
|
# TODO: Switch this back to request.POST, and let MultiPartParser deal with the consequences.
|
|
if getattr(request, 'user', None) and request.user.is_active:
|
|
# If this is a POST request we enforce CSRF validation.
|
|
if request.method.upper() == 'POST':
|
|
# Temporarily replace request.POST with .RAW_CONTENT,
|
|
# so that we use our more generic request parsing
|
|
request._post = self.view.RAW_CONTENT
|
|
resp = CsrfViewMiddleware().process_view(request, None, (), {})
|
|
del(request._post)
|
|
if resp is not None: # csrf failed
|
|
return None
|
|
return request.user
|
|
return None
|
|
|
|
|
|
# TODO: TokenAuthentication, DigestAuthentication, OAuthAuthentication
|