Added links field for user

Created `UserLink` model

Author: yakser

users/admin.py

@@ -2,7 +2,15 @@
 from django.contrib import admin
 
 from .helpers import send_verification_completed_email
-from .models import CustomUser, UserAchievement, Member, Mentor, Expert, Investor
+from .models import (
+    CustomUser,
+    UserAchievement,
+    Member,
+    Mentor,
+    Expert,
+    Investor,
+    UserLink,
+)
 
 
 @admin.register(CustomUser)
@@ -136,3 +144,8 @@ def save_model(self, request, obj, form, change):
 @admin.register(UserAchievement)
 class UserAchievementAdmin(admin.ModelAdmin):
     list_display = ("id", "title", "status", "user")
+
+
+@admin.register(UserLink)
+class UserLinkAdmin(admin.ModelAdmin):
+    list_display = ("id", "user", "link")

users/helpers.py

@@ -5,7 +5,7 @@
 
 from core.utils import Email
 from users.constants import PROTOCOL
-from users.models import UserAchievement
+from users.models import UserAchievement, UserLink
 
 User = get_user_model()
 
@@ -91,7 +91,23 @@ def send_verification_completed_email(user: User):
     Email.send_email(data)
 
 
+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 user
+    """
+
     # delete all old achievements
     UserAchievement.objects.filter(user_id=pk).delete()
     # create new achievements
@@ -105,3 +121,22 @@ def update_achievements(achievements, pk):
             for achievement in achievements
         ]
     )
+
+
+def update_links(links, pk):
+    """
+    Bootleg version of updating links via user
+    """
+
+    # delete all old links
+    UserLink.objects.filter(user_id=pk).delete()
+    # create new links
+    UserLink.objects.bulk_create(
+        [
+            UserLink(
+                user_id=pk,
+                link=link,
+            )
+            for link in links
+        ]
+    )

users/migrations/0036_userlink.py

@@ -0,0 +1,43 @@
+# Generated by Django 4.2 on 2023-05-16 19:48
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("users", "0035_alter_customuser_onboarding_stage"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="UserLink",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("link", models.URLField()),
+                (
+                    "user",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="links",
+                        to=settings.AUTH_USER_MODEL,
+                    ),
+                ),
+            ],
+            options={
+                "verbose_name": "Ссылка пользователя",
+                "verbose_name_plural": "Ссылки пользователей",
+                "unique_together": {("user", "link")},
+            },
+        ),
+    ]

users/models.py

@@ -324,3 +324,30 @@ def create_or_update_user_types(sender, instance, created, **kwargs):
             Expert.objects.create(user=instance)
         elif instance.user_type == CustomUser.INVESTOR:
             Investor.objects.create(user=instance)
+
+
+class UserLink(models.Model):
+    """
+    UserLink model
+
+    Represents the user's link to some resource.
+
+    Attributes:
+            user: ForeignKey instance of the CustomUser model.
+            link: URLField instance of the user's link to some resource.
+    """
+
+    user = models.ForeignKey(
+        CustomUser,
+        on_delete=models.CASCADE,
+        related_name="links",
+    )
+    link = models.URLField(null=False, blank=False)
+
+    def __str__(self):
+        return f"UserLink<{self.id}> - {self.user.first_name} {self.user.last_name}"
+
+    class Meta:
+        verbose_name = "Ссылка пользователя"
+        verbose_name_plural = "Ссылки пользователей"
+        unique_together = ("user", "link")

users/serializers.py

@@ -83,8 +83,13 @@ class UserDetailSerializer(serializers.ModelSerializer):
     mentor = MentorSerializer(required=False)
     achievements = AchievementListSerializer(required=False, many=True)
     key_skills = KeySkillsField(required=False)
+    links = serializers.SerializerMethodField()
     is_online = serializers.SerializerMethodField()
 
+    @classmethod
+    def get_links(cls, user: CustomUser):
+        return [user_link.link for user_link in user.links.all()]
+
     @classmethod
     def get_is_online(cls, user: CustomUser):
         cache_key = get_user_online_cache_key(user)
@@ -106,6 +111,7 @@ class Meta:
             "organization",
             "about_me",
             "avatar",
+            "links",
             "city",
             "is_active",
             "is_online",

users/views.py

@@ -29,7 +29,7 @@
 from users.helpers import (
     reset_email,
     verify_email,
-    update_achievements,
+    check_related_fields_update,
 )
 from users.constants import (
     VERBOSE_ROLE_TYPES,
@@ -117,16 +117,13 @@ class UserDetail(RetrieveUpdateDestroyAPIView):
 
     @transaction.atomic
     def put(self, request, pk):
-        # bootleg version of updating achievements via user
-        if request.data.get("achievements") is not None:
-            update_achievements(request.data.get("achievements"), pk)
+        check_related_fields_update(request.data, pk)
         return super().put(request, pk)
 
     @transaction.atomic
     def patch(self, request, pk):
-        # bootleg version of updating achievements via user
-        if request.data.get("achievements") is not None:
-            update_achievements(request.data.get("achievements"), pk)
+        check_related_fields_update(request.data, pk)
+
         return super().patch(request, pk)