Likes, views and more...

Author: yakser

core/admin.py

@@ -1,8 +1,14 @@
 from django.contrib import admin
-from core.models import Like
+from core.models import Like, View
 
 
 @admin.register(Like)
 class LikeAdmin(admin.ModelAdmin):
     list_display = ("id", "user", "content_type", "object_id", "content_object")
     list_display_links = ("id", "user", "content_type", "object_id", "content_object")
+
+
+@admin.register(View)
+class ViewAdmin(admin.ModelAdmin):
+    list_display = ("id", "user", "content_type", "object_id", "content_object")
+    list_display_links = ("id", "user", "content_type", "object_id", "content_object")

core/migrations/0002_view.py

@@ -0,0 +1,53 @@
+# Generated by Django 4.2 on 2023-06-11 10:28
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("contenttypes", "0002_remove_content_type_name"),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ("core", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="View",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("object_id", models.PositiveIntegerField()),
+                (
+                    "content_type",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="views",
+                        to="contenttypes.contenttype",
+                    ),
+                ),
+                (
+                    "user",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="views",
+                        to=settings.AUTH_USER_MODEL,
+                    ),
+                ),
+            ],
+            options={
+                "verbose_name": "Просмотр",
+                "verbose_name_plural": "Просмотры",
+                "unique_together": {("user", "content_type", "object_id")},
+            },
+        ),
+    ]

core/models.py

@@ -24,6 +24,7 @@ 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")
@@ -32,3 +33,33 @@ class Meta:
 
     def __str__(self):
         return f"Like<{self.user} - {self.content_object}>"
+
+
+class View(Model):
+    """
+    Generic View model based on contenttype
+
+    Indicates if user has viewed the object
+
+    """
+
+    user = models.ForeignKey(
+        User,
+        on_delete=models.CASCADE,
+        related_name="views",
+    )
+    content_type = models.ForeignKey(
+        ContentType,
+        on_delete=models.CASCADE,
+        related_name="views",
+    )
+    object_id = models.PositiveIntegerField()
+    content_object = GenericForeignKey("content_type", "object_id")
+
+    class Meta:
+        unique_together = ("user", "content_type", "object_id")
+        verbose_name = "Просмотр"
+        verbose_name_plural = "Просмотры"
+
+    def __str__(self):
+        return f"View<{self.user} - {self.content_object}>"

core/serializers.py

@@ -0,0 +1,9 @@
+from rest_framework import serializers
+
+
+class SetLikedSerializer(serializers.Serializer):
+    is_liked = serializers.BooleanField()
+
+
+class SetViewedSerializer(serializers.Serializer):
+    is_viewed = serializers.BooleanField()

core/services.py

@@ -0,0 +1,86 @@
+from django.contrib.auth import get_user_model
+from django.contrib.contenttypes.models import ContentType
+
+from core.models import Like, View
+
+User = get_user_model()
+
+
+def add_like(obj, user):
+    obj_type = ContentType.objects.get_for_model(obj)
+    like, is_created = Like.objects.get_or_create(
+        content_type=obj_type, object_id=obj.id, user=user
+    )
+    return like
+
+
+def remove_like(obj, user):
+    obj_type = ContentType.objects.get_for_model(obj)
+    Like.objects.filter(content_type=obj_type, object_id=obj.id, user=user).delete()
+
+
+def is_fan(obj, user) -> bool:
+    if not user.is_authenticated:
+        return False
+    obj_type = ContentType.objects.get_for_model(obj)
+    likes = Like.objects.filter(content_type=obj_type, object_id=obj.id, user=user)
+    return likes.exists()
+
+
+def get_fans(obj):
+    obj_type = ContentType.objects.get_for_model(obj)
+    return User.objects.filter(likes__content_type=obj_type, likes__object_id=obj.id)
+
+
+def get_likes_count(obj):
+    obj_type = ContentType.objects.get_for_model(obj)
+    return User.objects.filter(
+        likes__content_type=obj_type, likes__object_id=obj.id
+    ).count()
+
+
+def set_like(obj, user, is_liked):
+    if is_liked:
+        add_like(obj, user)
+    else:
+        remove_like(obj, user)
+
+
+def add_view(obj, user):
+    obj_type = ContentType.objects.get_for_model(obj)
+    view, is_created = View.objects.get_or_create(
+        content_type=obj_type, object_id=obj.id, user=user
+    )
+    return view
+
+
+def remove_view(obj, user):
+    obj_type = ContentType.objects.get_for_model(obj)
+    View.objects.filter(content_type=obj_type, object_id=obj.id, user=user).delete()
+
+
+def is_viewer(obj, user) -> bool:
+    if not user.is_authenticated:
+        return False
+    obj_type = ContentType.objects.get_for_model(obj)
+    views = View.objects.filter(content_type=obj_type, object_id=obj.id, user=user)
+    return views.exists()
+
+
+def get_viewers(obj):
+    obj_type = ContentType.objects.get_for_model(obj)
+    return User.objects.filter(views__content_type=obj_type, views__object_id=obj.id)
+
+
+def get_views_count(obj):
+    obj_type = ContentType.objects.get_for_model(obj)
+    return User.objects.filter(
+        views__content_type=obj_type, views__object_id=obj.id
+    ).count()
+
+
+def set_viewed(obj, user, is_viewed):
+    if is_viewed:
+        add_view(obj, user)
+    else:
+        remove_view(obj, user)

projects/migrations/0014_alter_projectnews_options_and_more.py

@@ -0,0 +1,24 @@
+# Generated by Django 4.2 on 2023-06-11 10:28
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("projects", "0013_projectnews"),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name="projectnews",
+            options={
+                "verbose_name": "Новость проекта",
+                "verbose_name_plural": "Новости проекта",
+            },
+        ),
+        migrations.RemoveField(
+            model_name="projectnews",
+            name="views_count",
+        ),
+    ]

projects/models.py

@@ -5,7 +5,7 @@
 from django.db import models
 from django.db.models import UniqueConstraint
 
-from core.models import Like
+from core.models import Like, View
 from industries.models import Industry
 from projects.constants import VERBOSE_STEPS
 from projects.managers import AchievementManager, ProjectManager
@@ -192,6 +192,7 @@ class Meta:
         ]
 
 
+# fixme: move project news to another app?
 class ProjectNews(models.Model):
     """
     Project news model
@@ -202,17 +203,29 @@ class ProjectNews(models.Model):
         on_delete=models.CASCADE,
         related_name="news",
     )
-    text = models.TextField(null=False, blank=False)
-    # todo
-    views_count = models.PositiveIntegerField(default=0)
-    likes = GenericRelation(Like, related_query_name="project_news")
+    text = models.TextField(
+        null=False,
+        blank=False,
+    )
+    views = GenericRelation(
+        View,
+        related_query_name="project_views",
+    )
+    likes = GenericRelation(
+        Like,
+        related_query_name="project_news",
+    )
     # todo: files
 
     datetime_created = models.DateTimeField(
-        verbose_name="Дата создания", null=False, auto_now_add=True
+        verbose_name="Дата создания",
+        null=False,
+        auto_now_add=True,
     )
     datetime_updated = models.DateTimeField(
-        verbose_name="Дата изменения", null=False, auto_now=True
+        verbose_name="Дата изменения",
+        null=False,
+        auto_now=True,
     )
 
     class Meta:

projects/permissions.py

@@ -67,7 +67,7 @@ def has_object_permission(self, request, view, obj):
             return False
 
 
-class IsNewsAuthorIsProjectLeader(BasePermission):
+class IsNewsAuthorIsProjectLeaderOrReadOnly(BasePermission):
     """
     Allows access to update project news only to leader.
     """

projects/serializers.py

@@ -1,8 +1,8 @@
 from django.contrib.auth import get_user_model
-from django.contrib.contenttypes.models import ContentType
 from rest_framework import serializers
 
 from core.fields import CustomListField
+from core.services import get_views_count, get_likes_count
 from industries.models import Industry
 from projects.models import Project, Achievement, Collaborator, ProjectNews
 from projects.validators import validate_project
@@ -211,6 +211,7 @@ class Meta:
 
 
 class ProjectNewsListSerializer(serializers.ModelSerializer):
+    views_count = serializers.SerializerMethodField()
     likes_count = serializers.SerializerMethodField()
     project_name = serializers.SerializerMethodField()
     project_image_address = serializers.SerializerMethodField()
@@ -221,11 +222,11 @@ def get_project_name(self, obj):
     def get_project_image_address(self, obj):
         return obj.project.image_address
 
+    def get_views_count(self, obj):
+        return get_views_count(obj)
+
     def get_likes_count(self, obj):
-        obj_type = ContentType.objects.get_for_model(obj)
-        return User.objects.filter(
-            likes__content_type=obj_type, likes__object_id=obj.id
-        ).count()
+        return get_likes_count(obj)
 
     class Meta:
         model = ProjectNews
@@ -241,6 +242,7 @@ class Meta:
 
 
 class ProjectNewsDetailSerializer(serializers.ModelSerializer):
+    views_count = serializers.SerializerMethodField()
     likes_count = serializers.SerializerMethodField()
     project_name = serializers.SerializerMethodField()
     project_image_address = serializers.SerializerMethodField()
@@ -251,11 +253,11 @@ def get_project_name(self, obj):
     def get_project_image_address(self, obj):
         return obj.project.image_address
 
+    def get_views_count(self, obj):
+        return get_views_count(obj)
+
     def get_likes_count(self, obj):
-        obj_type = ContentType.objects.get_for_model(obj)
-        return User.objects.filter(
-            likes__content_type=obj_type, likes__object_id=obj.id
-        ).count()
+        return get_likes_count(obj)
 
     class Meta:
         model = ProjectNews

projects/urls.py

@@ -13,15 +13,23 @@
     SetLikeOnProject,
     ProjectNewsList,
     ProjectNewsDetail,
+    ProjectNewsDetailSetViewed,
+    ProjectNewsDetailSetLiked,
 )
 
 app_name = "projects"
 
 urlpatterns = [
     path("", ProjectList.as_view()),
     path("<int:pk>/like/", SetLikeOnProject.as_view()),
-    path("<int:pk>/news/", ProjectNewsList.as_view()),
+    path("<int:project_pk>/news/", ProjectNewsList.as_view()),
     path("<int:project_pk>/news/<int:pk>/", ProjectNewsDetail.as_view()),
+    path(
+        "<int:project_pk>/news/<int:pk>/set_viewed/", ProjectNewsDetailSetViewed.as_view()
+    ),
+    path(
+        "<int:project_pk>/news/<int:pk>/set_liked/", ProjectNewsDetailSetLiked.as_view()
+    ),
     path("<int:pk>/collaborators/", ProjectCollaborators.as_view()),
     path("<int:pk>/", ProjectDetail.as_view()),
     path("<int:pk>/recommended_users", ProjectRecommendedUsers.as_view()),