Added links field for projects

Created `ProjectLink` model

Author: yakser

projects/admin.py

@@ -1,6 +1,6 @@
 from django.contrib import admin
 
-from projects.models import Project, Achievement, Collaborator
+from projects.models import Project, Achievement, Collaborator, ProjectLink
 
 
 @admin.register(Project)
@@ -21,6 +21,12 @@ class AchievementAdmin(admin.ModelAdmin):
     list_display = ("id", "title", "status", "project")
 
 
+@admin.register(ProjectLink)
+class ProjectLinkAdmin(admin.ModelAdmin):
+    list_display = ("id", "link", "project")
+    list_display_links = ("id", "link", "project")
+
+
 @admin.register(Collaborator)
 class CollaboratorAdmin(admin.ModelAdmin):
     list_display = (

projects/helpers.py

@@ -3,7 +3,7 @@
 from django.contrib.auth import get_user_model
 
 from projects.constants import RECOMMENDATIONS_COUNT
-from projects.models import Project
+from projects.models import Project, ProjectLink, Achievement
 
 User = get_user_model()
 
@@ -33,3 +33,54 @@ def get_recommended_users(project: Project) -> list[User]:
     )
 
     return sampled_recommended_users
+
+
+def check_related_fields_update(data, pk):
+    """
+    Check if achievements or links were updated and update them.
+    """
+
+    if data.get("achievements") is not None:
+        update_achievements(data.get("achievements"), pk)
+
+    if data.get("links") is not None:
+        update_links(data.get("links"), pk)
+
+
+def update_achievements(achievements, pk):
+    """
+    Bootleg version of updating achievements via project
+    """
+
+    # delete all old achievements
+    Achievement.objects.filter(project_id=pk).delete()
+    # create new achievements
+    Achievement.objects.bulk_create(
+        [
+            Achievement(
+                project_id=pk,
+                title=achievement.get("title"),
+                status=achievement.get("status"),
+            )
+            for achievement in achievements
+        ]
+    )
+
+
+def update_links(links, pk):
+    """
+    Bootleg version of updating links via project
+    """
+
+    # delete all old links
+    ProjectLink.objects.filter(project_id=pk).delete()
+    # create new links
+    ProjectLink.objects.bulk_create(
+        [
+            ProjectLink(
+                project_id=pk,
+                link=link,
+            )
+            for link in links
+        ]
+    )

projects/migrations/0012_projectlink.py

@@ -0,0 +1,42 @@
+# Generated by Django 4.2 on 2023-05-16 21:02
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("projects", "0011_project_views_count"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="ProjectLink",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("link", models.URLField()),
+                (
+                    "project",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="links",
+                        to="projects.project",
+                    ),
+                ),
+            ],
+            options={
+                "verbose_name": "Ссылка проекта",
+                "verbose_name_plural": "Ссылки проектов",
+                "unique_together": {("project", "link")},
+            },
+        ),
+    ]

projects/models.py

@@ -95,6 +95,31 @@ class Meta:
         verbose_name_plural = "Проекты"
 
 
+class ProjectLink(models.Model):
+    """
+    Project link model
+
+    Attributes:
+        project: A ForeignKey referring to the Project model.
+        link: A URLField link to the project.
+    """
+
+    project = models.ForeignKey(
+        Project,
+        on_delete=models.CASCADE,
+        related_name="links",
+    )
+    link = models.URLField(null=False, blank=False)
+
+    def __str__(self):
+        return f"ProjectLink<{self.id}> - {self.project.name}"
+
+    class Meta:
+        verbose_name = "Ссылка проекта"
+        verbose_name_plural = "Ссылки проектов"
+        unique_together = ("project", "link")
+
+
 class Achievement(models.Model):
     """
     Achievement model

projects/serializers.py

@@ -67,6 +67,11 @@ class ProjectDetailSerializer(serializers.ModelSerializer):
     short_description = serializers.SerializerMethodField()
     industry_id = serializers.IntegerField(required=False)
     likes_count = serializers.SerializerMethodField(method_name="count_likes")
+    links = serializers.SerializerMethodField()
+
+    @classmethod
+    def get_links(cls, project):
+        return [link.link for link in project.links.all()]
 
     def validate(self, data):
         super().validate(data)
@@ -92,6 +97,7 @@ class Meta:
             "description",
             "short_description",
             "achievements",
+            "links",
             "region",
             "step",
             "industry",

projects/views.py

@@ -9,7 +9,7 @@
 from core.permissions import IsStaffOrReadOnly
 from projects.filters import ProjectFilter
 from projects.constants import VERBOSE_STEPS
-from projects.helpers import get_recommended_users
+from projects.helpers import get_recommended_users, check_related_fields_update
 from projects.models import Project, Achievement
 from projects.permissions import (
     IsProjectLeaderOrReadOnlyForNonDrafts,
@@ -91,23 +91,11 @@ def retrieve(self, request, *args, **kwargs):
         return Response(serializer.data)
 
     def put(self, request, pk, **kwargs):
-        # bootleg version of updating achievements via project
-        if request.data.get("achievements") is not None:
-            achievements = request.data.get("achievements")
-            # delete all old achievements
-            Achievement.objects.filter(project_id=pk).delete()
-            # create new achievements
-            Achievement.objects.bulk_create(
-                [
-                    Achievement(
-                        project_id=pk,
-                        title=achievement.get("title"),
-                        status=achievement.get("status"),
-                    )
-                    for achievement in achievements
-                ]
-            )
+        check_related_fields_update(request.data, pk)
+        return super(ProjectDetail, self).put(request, pk)
 
+    def patch(self, request, pk, **kwargs):
+        check_related_fields_update(request.data, pk)
         return super(ProjectDetail, self).put(request, pk)
 
 

users/admin.py

@@ -149,3 +149,4 @@ class UserAchievementAdmin(admin.ModelAdmin):
 @admin.register(UserLink)
 class UserLinkAdmin(admin.ModelAdmin):
     list_display = ("id", "user", "link")
+    list_display_links = ("id", "user", "link")

users/views.py

@@ -123,7 +123,6 @@ def put(self, request, pk):
     @transaction.atomic
     def patch(self, request, pk):
         check_related_fields_update(request.data, pk)
-
         return super().patch(request, pk)