diff --git a/backend/api/posts/serializers.py b/backend/api/posts/serializers.py index 7216aaa..c2e91db 100644 --- a/backend/api/posts/serializers.py +++ b/backend/api/posts/serializers.py @@ -62,16 +62,9 @@ class PostPreviewSerializer(serializers.ModelSerializer): def get_media(self, obj: PostModel) -> List[Dict[str, str]]: data = [] for i in obj.files.all(): - if i.mimetype.startswith("video/"): - f_type = "video" - elif i.mimetype.startswith("image/gif"): - f_type = "gif" - elif i.mimetype.startswith("image/"): - f_type = "image" - else: - f_type = "other" - - data.append({"type": f_type, "mimetype": i.mimetype, "src": i.file.url}) + data.append( + {"type": i.file_type, "mimetype": i.mimetype, "src": i.file.url} + ) return data def get_media_count(self, obj: PostModel) -> int: @@ -142,5 +135,7 @@ class PostSerializer(serializers.ModelSerializer): def get_media(self, obj) -> List[Dict[str, str]]: data = [] for i in obj.files.all(): - data.append({"type": i.mimetype, "url": i.file.url}) + data.append( + {"type": i.file_type, "mimetype": i.mimetype, "src": i.file.url} + ) return data