From fe2800ecd329798cd64056f178209051c7d27ee4 Mon Sep 17 00:00:00 2001 From: Bruno Alla Date: Thu, 29 Jun 2023 12:18:28 +0200 Subject: [PATCH] Improve type hints for UserSerializer --- .../users/api/serializers.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/api/serializers.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/api/serializers.py index 6b26367d0..83b92eb26 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/api/serializers.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/api/serializers.py @@ -1,10 +1,18 @@ +from __future__ import annotations + +import typing + from django.contrib.auth import get_user_model from rest_framework import serializers +if typing.TYPE_CHECKING: + from {{ cookiecutter.project_slug }}.users.models import User as UserType + + User = get_user_model() -class UserSerializer(serializers.ModelSerializer): +class UserSerializer(serializers.ModelSerializer[UserType]): class Meta: model = User {%- if cookiecutter.username_type == "email" %}