starlette/docs/authentication.md

4.3 KiB

Starlette offers a simple but powerful interface for handling authentication and permissions. Once you've installed AuthenticationMiddleware with an appropriate authentication backend the request.user and request.auth interfaces will be available in your endpoints.

from starlette.authentication import (
    AuthenticationBackend, AuthenticationError, SimpleUser, UnauthenticatedUser,
    AuthCredentials
)
from starlette.middleware.authentication import AuthenticationMiddleware
from starlette.responses import PlainTextResponse
import base64
import binascii


class BasicAuthBackend(AuthenticationBackend):
    async def authenticate(self, request):
        if "Authorization" not in request.headers:
            return

        auth = request.headers["Authorization"]
        try:
            scheme, credentials = auth.split()
            if scheme.lower() != 'basic':
                return
            decoded = base64.b64decode(credentials).decode("ascii")
        except (ValueError, UnicodeDecodeError, binascii.Error) as exc:
            raise AuthenticationError('Invalid basic auth credentials')

        username, _, password = decoded.partition(":")
        # TODO: You'd want to verify the username and password here,
        #       possibly by installing `DatabaseMiddleware`
        #       and retrieving user information from `request.database`.
        return AuthCredentials(["authenticated"]), SimpleUser(username)


app = Starlette()
app.add_middleware(AuthenticationMiddleware, backend=BasicAuthBackend())


@app.route('/')
async def homepage(request):
    if request.user.is_authenticated:
        return PlainTextResponse('hello, ' + request.user.display_name)
    return PlainTextResponse('hello, you')

Users

Once AuthenticationMiddleware is installed the request.user interface will be available to endpoints or other middleware.

This interface should subclass BaseUser, which provides two properties, as well as whatever other information your user model includes.

  • .is_authenticated
  • .display_name

Starlette provides two built-in user implementations: UnauthenticatedUser(), and SimpleUser(username).

AuthCredentials

It is important that authentication credentials are treated as separate concept from users. An authentication scheme should be able to restrict or grant particular privileges independently of the user identity.

The AuthCredentials class provides the basic interface that request.auth exposes:

  • .scopes

Permissions

Permissions are implemented as an endpoint decorator, that enforces that the incoming request includes the required authentication scopes.

from starlette.authentication import requires


@app.route('/dashboard')
@requires('authenticated')
async def dashboard(request):
    ...

You can include either one or multiple required scopes:

from starlette.authentication import requires


@app.route('/dashboard')
@requires(['authenticated', 'admin'])
async def dashboard(request):
    ...

By default 403 responses will be returned when permissions are not granted. In some cases you might want to customize this, for example to hide information about the URL layout from unauthenticated users.

from starlette.authentication import requires


@app.route('/dashboard')
@requires(['authenticated', 'admin'], status_code=404)
async def dashboard(request):
    ...

Alternatively you might want to redirect unauthenticated users to a different page.

from starlette.authentication import requires


@app.route('/homepage')
async def homepage(request):
    ...


@app.route('/dashboard')
@requires('authenticated', redirect='homepage')
async def dashboard(request):
    ...

For class-based endpoints, you should wrap the decorator around a method on the class.

@app.route("/dashboard")
class Dashboard(HTTPEndpoint):
    @requires("authenticated")
    async def get(self, request):
        ...

Custom authentication error responses

You can customise the error response sent when a AuthenticationError is raised by an auth backend:

def on_auth_error(request: Request, exc: Exception):
    return JSONResponse({"error": str(exc)}, status_code=401)

app.add_middleware(AuthenticationMiddleware, backend=BasicAuthBackend(), on_error=on_auth_error)