Merge pull request #148 from PROCOLLAB-github/dev

Files for project news

Author: yakser

core/models.py

@@ -24,7 +24,6 @@ class Like(Model):
     )
     object_id = models.PositiveIntegerField()
     content_object = GenericForeignKey("content_type", "object_id")
-    # is liked?
 
     class Meta:
         unique_together = ("user", "content_type", "object_id")

files/admin.py

@@ -6,8 +6,32 @@
 @admin.register(UserFile)
 class UserFileAdmin(admin.ModelAdmin):
     list_display = (
+        "short_link",
+        "filename",
+        "user",
+        "datetime_uploaded",
+    )
+    list_display_links = (
+        "short_link",
+        "filename",
+        "user",
+    )
+    search_fields = (
         "link",
+        "user__first_name",
+        "user__last_name",
+    )
+    list_filter = (
         "user",
         "datetime_uploaded",
     )
-    list_display_links = ("link",)
+
+    date_hierarchy = "datetime_uploaded"
+
+    @admin.display(empty_value="Empty filename")
+    def filename(self, obj):
+        return obj.name
+
+    @admin.display(empty_value="Empty link")
+    def short_link(self, obj):
+        return obj.link[15:40] + "..." + obj.link[-20:]

files/models.py

@@ -9,13 +9,16 @@ class UserFile(models.Model):
     UserFile model
 
     Attributes:
-        user: User who uploaded the file
         link: Link to the file on the CDN
+        user: User who uploaded the file
         datetime_uploaded: Datetime when the file was uploaded
+        name: Name of the file
+        extension: Extension of the file
+        size: Size of the file in bytes
     """
 
-    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True)
     link = models.URLField(primary_key=True, null=False)
+    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True)
     datetime_uploaded = models.DateTimeField(auto_now_add=True)
     name = models.TextField(blank=False, default="file")
     extension = models.TextField(blank=True, default="")

projects/admin.py

@@ -28,6 +28,8 @@ class ProjectNewsAdmin(admin.ModelAdmin):
         "project",
         "datetime_created",
     )
+    # todo: set up admin panel for files
+    filter_horizontal = ("files",)
 
 
 @admin.register(Achievement)

projects/migrations/0015_projectnews_files.py

@@ -0,0 +1,21 @@
+# Generated by Django 4.2 on 2023-06-14 08:13
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("files", "0003_userfile_extension_userfile_name_userfile_size"),
+        ("projects", "0014_alter_projectnews_options_and_more"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="projectnews",
+            name="files",
+            field=models.ManyToManyField(
+                related_name="projects_news", to="files.userfile"
+            ),
+        ),
+    ]

projects/migrations/0016_alter_projectnews_files.py

@@ -0,0 +1,21 @@
+# Generated by Django 4.2 on 2023-06-14 08:36
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("files", "0003_userfile_extension_userfile_name_userfile_size"),
+        ("projects", "0015_projectnews_files"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="projectnews",
+            name="files",
+            field=models.ManyToManyField(
+                blank=True, related_name="projects_news", to="files.userfile"
+            ),
+        ),
+    ]

projects/models.py

@@ -6,6 +6,7 @@
 from django.db.models import UniqueConstraint
 
 from core.models import Like, View
+from files.models import UserFile
 from industries.models import Industry
 from projects.constants import VERBOSE_STEPS
 from projects.managers import AchievementManager, ProjectManager
@@ -207,6 +208,9 @@ class ProjectNews(models.Model):
         null=False,
         blank=False,
     )
+    # todo: remove files unused files
+    files = models.ManyToManyField(UserFile, related_name="projects_news", blank=True)
+
     views = GenericRelation(
         View,
         related_query_name="project_views",
@@ -215,7 +219,6 @@ class ProjectNews(models.Model):
         Like,
         related_query_name="project_news",
     )
-    # todo: files
 
     datetime_created = models.DateTimeField(
         verbose_name="Дата создания",

projects/pagination.py

@@ -0,0 +1,15 @@
+from rest_framework import pagination
+
+
+class ProjectNewsPagination(pagination.LimitOffsetPagination):
+    """
+    Pagination for project news
+
+    For example:
+        /projects/1/news/?limit=10&offset=10
+        gets the next 10 news after the first 10 news.
+    """
+
+    default_limit = 5
+    limit_query_param = "limit"
+    offset_query_param = "offset"

projects/serializers.py

@@ -238,6 +238,7 @@ class Meta:
             "datetime_created",
             "views_count",
             "likes_count",
+            "files",
         ]
 
 
@@ -270,4 +271,5 @@ class Meta:
             "datetime_updated",
             "views_count",
             "likes_count",
+            "files",
         ]

projects/views.py

@@ -13,6 +13,7 @@
 from projects.constants import VERBOSE_STEPS
 from projects.helpers import get_recommended_users, check_related_fields_update
 from projects.models import Project, Achievement, ProjectNews
+from projects.pagination import ProjectNewsPagination
 from projects.permissions import (
     IsProjectLeaderOrReadOnlyForNonDrafts,
     HasInvolvementInProjectOrReadOnly,
@@ -233,6 +234,7 @@ def get(self, request, pk):
 class ProjectNewsList(generics.ListCreateAPIView):
     serializer_class = ProjectNewsListSerializer
     permission_classes = [IsNewsAuthorIsProjectLeaderOrReadOnly]
+    pagination_class = ProjectNewsPagination
 
     def perform_create(self, serializer):
         project = Project.objects.get(pk=self.kwargs.get("project_pk"))