diff --git a/rest_framework-stubs/fields.pyi b/rest_framework-stubs/fields.pyi index 0ccae0e22..b6c17806a 100644 --- a/rest_framework-stubs/fields.pyi +++ b/rest_framework-stubs/fields.pyi @@ -22,6 +22,7 @@ from typing import ( MutableMapping, ) +from django.contrib.auth.models import AnonymousUser, User from django.core.exceptions import ValidationError from django.db import models from django.db.models import Model @@ -39,6 +40,10 @@ class CreateOnlyDefault(object): def set_context(self, serializer_field: Field) -> None: ... def __call__(self) -> Any: ... +class CurrentUserDefault(object): + def set_context(self, serializer_field: Field) -> None: ... + def __call__(self) -> Union[AnonymousUser, User]: ... + class SkipField(Exception): ... def set_value(dictionary: MutableMapping[str, Any], keys: Sequence[str], value: Any) -> None: ... diff --git a/rest_framework-stubs/serializers.pyi b/rest_framework-stubs/serializers.pyi index ee67ebaf8..bed3af567 100644 --- a/rest_framework-stubs/serializers.pyi +++ b/rest_framework-stubs/serializers.pyi @@ -47,6 +47,7 @@ from rest_framework.fields import ( ImageField as ImageField, SkipField as SkipField, CreateOnlyDefault as CreateOnlyDefault, + CurrentUserDefault as CurrentUserDefault, ) from rest_framework.exceptions import ( APIException as APIException,