Merge pull request #154 from PROCOLLAB-github/feature/partner-programs

Feature/partner programs

Author: VeryBigSad

core/migrations/0003_link.py

@@ -0,0 +1,44 @@
+# Generated by Django 4.1.3 on 2023-07-08 17:05
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("contenttypes", "0002_remove_content_type_name"),
+        ("core", "0002_view"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Link",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("link", models.URLField(verbose_name="Ссылка")),
+                ("object_id", models.PositiveIntegerField()),
+                (
+                    "content_type",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="links",
+                        to="contenttypes.contenttype",
+                    ),
+                ),
+            ],
+            options={
+                "verbose_name": "Ссылка",
+                "verbose_name_plural": "Ссылки",
+                "unique_together": {("content_type", "object_id")},
+            },
+        ),
+    ]

core/models.py

@@ -7,6 +7,31 @@
 User = get_user_model()
 
 
+class Link(Model):
+    """
+    Generic Link model based on contenttype
+    """
+
+    link = models.URLField(
+        verbose_name="Ссылка",
+    )
+    content_type = models.ForeignKey(
+        ContentType,
+        on_delete=models.CASCADE,
+        related_name="links",
+    )
+    object_id = models.PositiveIntegerField()
+    content_object = GenericForeignKey("content_type", "object_id")
+
+    class Meta:
+        unique_together = ("content_type", "object_id")
+        verbose_name = "Ссылка"
+        verbose_name_plural = "Ссылки"
+
+    def __str__(self):
+        return f"Link for {self.content_object} - {self.link}"
+
+
 class Like(Model):
     """
     Generic Like model based on contenttype

core/services.py

@@ -1,7 +1,7 @@
 from django.contrib.auth import get_user_model
 from django.contrib.contenttypes.models import ContentType
 
-from core.models import Like, View
+from core.models import Like, View, Link
 
 User = get_user_model()
 
@@ -86,3 +86,21 @@ def set_viewed(obj, user, is_viewed):
         add_view(obj, user)
     else:
         remove_view(obj, user)
+
+
+def add_link(obj, link: str):
+    obj_type = ContentType.objects.get_for_model(obj)
+    like, is_created = Link.objects.get_or_create(
+        content_type=obj_type, object_id=obj.id, link=link
+    )
+    return like
+
+
+def remove_link(obj, link):
+    obj_type = ContentType.objects.get_for_model(obj)
+    Like.objects.filter(content_type=obj_type, object_id=obj.id, link=link).delete()
+
+
+def get_links(obj):
+    obj_type = ContentType.objects.get_for_model(obj)
+    return Link.objects.filter(content_type=obj_type, object_id=obj.id)

news/managers.py

@@ -0,0 +1,13 @@
+from django.contrib.contenttypes.models import ContentType
+from django.db import models
+
+
+class NewsManager(models.Manager):
+    def get_news(self, obj):
+        obj_type = ContentType.objects.get_for_model(obj)
+        return self.get_queryset().filter(content_type=obj_type, object_id=obj.id)
+
+    def add_news(self, obj, **kwargs):
+        obj_type = ContentType.objects.get_for_model(obj)
+        news = self.create(content_type=obj_type, object_id=obj.id, **kwargs)
+        return news

news/migrations/0005_news_content_type_news_object_id.py

@@ -0,0 +1,31 @@
+# Generated by Django 4.1.3 on 2023-07-08 18:58
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("contenttypes", "0002_remove_content_type_name"),
+        ("news", "0004_alter_news_tags"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="news",
+            name="content_type",
+            field=models.ForeignKey(
+                default=0,
+                on_delete=django.db.models.deletion.CASCADE,
+                to="contenttypes.contenttype",
+            ),
+            preserve_default=False,
+        ),
+        migrations.AddField(
+            model_name="news",
+            name="object_id",
+            field=models.PositiveIntegerField(default=0),
+            preserve_default=False,
+        ),
+    ]

news/models.py

@@ -1,5 +1,8 @@
+from django.contrib.contenttypes.models import ContentType
 from django.db import models
 
+from news.managers import NewsManager
+
 
 class NewsTag(models.Model):
     """News tag model
@@ -52,13 +55,19 @@ class News(models.Model):
 
     tags = models.ManyToManyField(NewsTag, blank=True, verbose_name="Список тегов")
 
+    # generic relation to project/program/ something else possibly
+    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
+    object_id = models.PositiveIntegerField()
+
     datetime_created = models.DateTimeField(
         verbose_name="Дата создания", null=False, auto_now_add=True
     )
     datetime_updated = models.DateTimeField(
         verbose_name="Дата изменения", null=False, auto_now=True
     )
 
+    objects = NewsManager()
+
     @property
     def tags_str(self):
         """Formats tags to string