akarpov/files/views.py

@@ -1,6 +1,5 @@
 import os
 
-import elastic_transport
 import structlog
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.core.exceptions import PermissionDenied
@@ -92,9 +91,7 @@ def get_queryset(self):
         ):
             return self.filter(BaseFileItem.objects.none())
         return self.filter(
-            BaseFileItem.objects.cache().filter(
+            BaseFileItem.objects.filter(user=self.request.user, parent__isnull=True)
-                user=self.request.user, parent__isnull=True
-            )
         )
 
 
@@ -316,25 +313,18 @@ def on_completion(self, uploaded_file, request):
                 }
                 prepared = False
         if prepared and uploaded_file.size <= request.user.left_file_upload:
-            try:
+            f = File.objects.create(
-                f = File.objects.create(
+                user=request.user,
-                    user=request.user,
+                file_obj=uploaded_file,
-                    file_obj=uploaded_file,
+                name=uploaded_file.name,
-                    name=uploaded_file.name,
+                parent=folder,
-                    parent=folder,
+            )
-                )
+            request.user.left_file_upload -= uploaded_file.size
-                request.user.left_file_upload -= uploaded_file.size
+            request.user.save()
-                request.user.save()
+            self.message = {
-                self.message = {
+                "message": f"File {f.file.name.split('/')[-1]} successfully uploaded",
-                    "message": f"File {f.file.name.split('/')[-1]} successfully uploaded",
+                "status": True,
-                    "status": True,
+            }
-                }
-            except elastic_transport.ConnectionError:
-                self.message = {
-                    "message": "Service is down, please try again later or contact support",
-                    "status": False,
-                }
-                logger.error("Elasticsearch is down")
         elif prepared:
             self.message = {
                 "message": "File is too large, please increase disk space",

akarpov/music/api/serializers.py

@@ -16,13 +16,13 @@
 class AuthorSerializer(serializers.ModelSerializer):
     class Meta:
         model = Author
-        fields = ["name", "slug", "image_cropped"]
+        fields = ["name", "slug", "link", "image_cropped"]
 
 
 class AlbumSerializer(serializers.ModelSerializer):
     class Meta:
         model = Album
-        fields = ["name", "slug", "image_cropped"]
+        fields = ["name", "slug", "link", "image_cropped"]
 
 
 class SongSerializer(serializers.ModelSerializer):
@@ -32,12 +32,10 @@ class SongSerializer(serializers.ModelSerializer):
 
     @extend_schema_field(serializers.BooleanField)
     def get_liked(self, obj):
-        if "request" in self.context:
+        if self.context["request"].user.is_authenticated:
-            if self.context["request"].user.is_authenticated:
+            return SongUserRating.objects.filter(
-                return SongUserRating.objects.filter(
+                song=obj, user=self.context["request"].user, like=True
-                    song=obj, user=self.context["request"].user, like=True
+            ).exists()
-                ).exists()
-        return None
 
     class Meta:
         model = Song
@@ -61,30 +59,18 @@ class Meta:
 
 
 class ListSongSerializer(SetUserModelSerializer):
-    album = AlbumSerializer(read_only=True)
+    album = serializers.CharField(source="album.name", read_only=True)
-    authors = AuthorSerializer(many=True, read_only=True)
     liked = serializers.SerializerMethodField(method_name="get_liked")
 
     @extend_schema_field(serializers.BooleanField)
     def get_liked(self, obj):
         if "likes" in self.context:
             return self.context["likes"]
-        if "likes_ids" in self.context:
+        return obj.id in self.context["likes_ids"]
-            return obj.id in self.context["likes_ids"]
-        return None
 
     class Meta:
         model = Song
-        fields = [
+        fields = ["name", "slug", "file", "image_cropped", "length", "album", "liked"]
-            "name",
-            "slug",
-            "file",
-            "image_cropped",
-            "length",
-            "album",
-            "authors",
-            "liked",
-        ]
         extra_kwargs = {
             "slug": {"read_only": True},
             "image_cropped": {"read_only": True},

akarpov/music/tasks.py

@@ -48,8 +48,7 @@ def process_file(path, user_id):
 
 @shared_task
 def load_ym_file_meta(track, user_id):
-    yb = yandex.load_file_meta(track, user_id)
+    return yandex.load_file_meta(track, user_id)
-    return yb
 
 
 @shared_task()