Zpět na domů

JWT autorizace FastAPI: kód krok za krokem

Průvodce implementací JWT autentifikace v FastAPI pro middle/senior vývojáře. Pokrývá strukturu tokenu, Pydantic schémata, security modul, závislosti a chráněné endpointy s kompletním kódem.

JWT v FastAPI: kód login + chráněné routy
Advertisement 728x90

Implementace JWT autentizace ve FastAPI: kompletní průvodce pro vývojáře

JWT (JSON Web Token) je kompaktní token pro přenos tvrzení mezi stranami. Token se skládá ze tří částí kódovaných v Base64: hlavičky, datové části a podpisu, oddělených tečkami.

Hlavička obsahuje algoritmus podpisu (např. HS256) a typ tokenu (JWT):

{"alg": "HS256", "typ": "JWT"}

Datová část ukládá tvrzení: sub (identifikátor uživatele), iat (čas vydání), exp (čas vypršení platnosti):

Google AdInline article slot
{"sub": "alex", "name": "Alex Tomchok", "iat": 1516239022, "exp": 1516242622}

Podpis se vypočítá jako HMAC-SHA256 z (hlavička.datová část + tajný klíč). Server ověřuje podpis při validaci: jakákoli změna datové části činí token neplatným. Data jsou čitelná, ale chráněná proti úpravám.

Access tokeny mají krátkou životnost (15–30 minut), refresh tokeny dlouhou (dny/měsíce). V příkladu se používá pouze access token.

Příprava projektu a závislosti

Vytvořte strukturu:

Google AdInline article slot
fastapi-jwt-auth/
├── app/
│   ├── main.py
│   ├── core/
│   │   ├── config.py
│   │   └── security.py
│   ├── schemas/
│   │   ├── token.py
│   │   └── user.py
│   ├── api/
│   │   └── v1/
│   │       ├── endpoints/
│   │       │   ├── auth.py
│   │       │   └── users.py
│   │       └── dependencies.py
│   └── services/
│       └── user_service.py
└── requirements.txt

requirements.txt:

fastapi==0.104.1
uvicorn[standard]==0.24.0
python-jose[cryptography]==3.3.0
passlib[bcrypt]==1.7.4
python-multipart==0.0.6
pydantic[email]==2.5.0

Nainstalujte: pip install -r requirements.txt.

Konfigurace a Pydantic schémata

V core/config.py definujte nastavení pomocí Pydantic:

Google AdInline article slot
from pydantic_settings import BaseSettings

class Settings(BaseSettings):
    SECRET_KEY: str = "váš-super-tajný-klíč-změňte-v-produkci"
    ALGORITHM: str = "HS256"
    ACCESS_TOKEN_EXPIRE_MINUTES: int = 30

    class Config:
        env_file = ".env"

settings = Settings()

Schémata v schemas/user.py:

from pydantic import BaseModel, EmailStr

class UserBase(BaseModel):
    username: str
    email: EmailStr

class UserCreate(UserBase):
    password: str

class User(UserBase):
    id: int
    is_active: bool = True

    class Config:
        from_attributes = True

class UserInDB(User):
    hashed_password: str

schemas/token.py:

from pydantic import BaseModel

class Token(BaseModel):
    access_token: str
    token_type: str = "bearer"

class TokenData(BaseModel):
    username: str | None = None

Zabezpečení: hashování a JWT operace

V core/security.py implementujte hashování hesel (bcrypt) a JWT funkce:

from datetime import datetime, timedelta
from jose import JWTError, jwt
from passlib.context import CryptContext
from .config import settings

pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")

def verify_password(plain_password: str, hashed_password: str) -> bool:
    return pwd_context.verify(plain_password, hashed_password)

def get_password_hash(password: str) -> str:
    return pwd_context.hash(password)

def create_access_token(data: dict) -> str:
    to_encode = data.copy()
    expire = datetime.utcnow() + timedelta(minutes=settings.ACCESS_TOKEN_EXPIRE_MINUTES)
    to_encode.update({"exp": expire})
    encoded_jwt = jwt.encode(to_encode, settings.SECRET_KEY, algorithm=settings.ALGORITHM)
    return encoded_jwt

def decode_access_token(token: str) -> dict:
    try:
        payload = jwt.decode(token, settings.SECRET_KEY, algorithms=[settings.ALGORITHM])
        return payload
    except JWTError:
        return None

Uživatelský servis a falešná databáze

services/user_service.py s in-memory úložištěm:

from app.schemas.user import UserInDB
from app.core.security import verify_password, get_password_hash

fake_users_db = {
    "alex": {
        "id": 1,
        "username": "alex",
        "email": "[email protected]",
        "hashed_password": get_password_hash("secret123"),
        "is_active": True,
    }
}

def get_user_by_username(username: str) -> UserInDB | None:
    if username in fake_users_db:
        return UserInDB(**fake_users_db[username])
    return None

def authenticate_user(username: str, password: str) -> UserInDB | None:
    user = get_user_by_username(username)
    if not user:
        return None
    if not verify_password(password, user.hashed_password):
        return None
    return user

Závislosti pro autorizaci

V api/v1/dependencies.py použijte OAuth2PasswordBearer:

from fastapi import Depends, HTTPException, status
from fastapi.security import OAuth2PasswordBearer
from app.core.security import decode_access_token
from app.services.user_service import get_user_by_username
from app.schemas.token import TokenData
from app.schemas.user import User

oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/v1/auth/login")

async def get_current_user(token: str = Depends(oauth2_scheme)) -> User:
    credentials_exception = HTTPException(
        status_code=status.HTTP_401_UNAUTHORIZED,
        detail="Přihlašovací údaje nelze ověřit",
        headers={"WWW-Authenticate": "Bearer"},
    )
    payload = decode_access_token(token)
    if payload is None:
        raise credentials_exception
    username: str = payload.get("sub")
    if username is None:
        raise credentials_exception
    user = get_user_by_username(username)
    if user is None:
        raise credentials_exception
    return user

async def get_current_active_user(current_user: User = Depends(get_current_user)) -> User:
    if not current_user.is_active:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Neaktivní uživatel")
    return current_user

Endpointy: přihlášení a chráněné cesty

api/v1/endpoints/auth.py:

from fastapi import APIRouter, Depends, HTTPException, status
from fastapi.security import OAuth2PasswordRequestForm
from app.schemas.token import Token
from app.services.user_service import authenticate_user
from app.core.security import create_access_token

router = APIRouter(prefix="/auth", tags=["authentication"])

@router.post("/login", response_model=Token)
async def login(form_data: OAuth2PasswordRequestForm = Depends()):
    user = authenticate_user(form_data.username, form_data.password)
    if not user:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail="Nesprávné uživatelské jméno nebo heslo",
            headers={"WWW-Authenticate": "Bearer"},
        )
    access_token = create_access_token(data={"sub": user.username})
    return Token(access_token=access_token, token_type="bearer")

api/v1/endpoints/users.py:

from fastapi import APIRouter, Depends
from app.schemas.user import User
from app.api.v1.dependencies import get_current_active_user

router = APIRouter(prefix="/users", tags=["users"])

@router.get("/me", response_model=User)
async def read_users_me(current_user: User = Depends(get_current_active_user)):
    return current_user

Připojte routery v main.py a spusťte uvicorn app.main:app.

Co je důležité

  • Datová část JWT není šifrovaná, pouze podepsaná — používejte pouze necitlivá data.
  • Uchovávejte SECRET_KEY v proměnných prostředí, generujte délku 32+ znaků.
  • V produkci přidejte refresh tokeny a databázi (SQLAlchemy + Alembic).
  • Závislosti FastAPI umožňují znovupoužití logiky autorizace bez duplikace.
  • Testujte endpointy přes Swagger UI: /docs.

— Editorial Team

Advertisement 728x90

Číst dál