Clean up object type hints

This commit is contained in:
Ciarán Ainsworth 2022-07-18 12:19:48 +02:00
parent 65e4823a09
commit db5a6621f8
3 changed files with 11 additions and 4 deletions

View File

@ -258,7 +258,8 @@ class ChannelSerializer(serializers.ModelSerializer):
"downloads_count",
]
def get_artist(self, obj) -> object:
@extend_schema_field(OpenApiTypes.OBJECT)
def get_artist(self, obj):
return music_serializers.serialize_artist_simple(obj.artist)
def to_representation(self, obj):

View File

@ -10,6 +10,9 @@ from django.core.files.uploadedfile import SimpleUploadedFile
from django.utils.encoding import smart_text
from django.utils.translation import ugettext_lazy as _
from drf_spectacular.utils import extend_schema_field
from drf_spectacular.types import OpenApiTypes
from . import models
from . import utils
@ -270,7 +273,8 @@ class APIMutationSerializer(serializers.ModelSerializer):
"previous_state",
]
def get_target(self, obj) -> object:
@extend_schema_field(OpenApiTypes.OBJECT)
def get_target(self, obj):
target = obj.target
if not target:
return
@ -292,7 +296,8 @@ class AttachmentSerializer(serializers.Serializer):
file = StripExifImageField(write_only=True)
urls = serializers.SerializerMethodField()
def get_urls(self, o) -> object:
@extend_schema_field(OpenApiTypes.OBJECT)
def get_urls(self, o):
urls = {}
urls["source"] = o.url
urls["original"] = o.download_url_original

View File

@ -93,7 +93,8 @@ class ManageUserSerializer(serializers.ModelSerializer):
)
return instance
def get_actor(self, obj) -> object:
@extend_schema_field(OpenApiTypes.OBJECT)
def get_actor(self, obj):
if obj.actor:
return ManageBaseActorSerializer(obj.actor).data