From 4e7248755819ee5a62cb4d688ec3751fe92d7c71 Mon Sep 17 00:00:00 2001
From: littlewolf <nickptic@code-artel.com>
Date: Thu, 14 Nov 2019 20:54:14 +0300
Subject: [PATCH] Add rating.views_count transfer

---
 apps/news/models.py                           |  3 +-
 apps/rating/migrations/0003_viewcount.py      | 27 ++++++++++++++++++
 apps/rating/models.py                         | 18 ++++++++++++
 apps/rating/transfer_data.py                  | 28 +++++++++++++++++++
 apps/transfer/management/commands/transfer.py |  1 +
 project/settings/local.py                     |  5 ++--
 6 files changed, 79 insertions(+), 3 deletions(-)
 create mode 100644 apps/rating/migrations/0003_viewcount.py
 create mode 100644 apps/rating/transfer_data.py

diff --git a/apps/news/models.py b/apps/news/models.py
index 93a2b7cd..d52d04b8 100644
--- a/apps/news/models.py
+++ b/apps/news/models.py
@@ -6,7 +6,7 @@ from django.utils import timezone
 from django.utils.translation import gettext_lazy as _
 from rest_framework.reverse import reverse
 
-from rating.models import Rating
+from rating.models import Rating, ViewCount
 from utils.models import BaseAttributes, TJSONField, TranslatedFieldsMixin, ProjectBaseMixin
 from utils.querysets import TranslationQuerysetMixin
 
@@ -182,6 +182,7 @@ class News(BaseAttributes, TranslatedFieldsMixin):
     tags = models.ManyToManyField('tag.Tag', related_name='news',
                                   verbose_name=_('Tags'))
     gallery = models.ManyToManyField('gallery.Image', through='news.NewsGallery')
+    views_count = generic.GenericRelation(ViewCount, blank=True, null=True)
     ratings = generic.GenericRelation(Rating)
     favorites = generic.GenericRelation(to='favorites.Favorites')
     agenda = models.ForeignKey('news.Agenda', blank=True, null=True,
diff --git a/apps/rating/migrations/0003_viewcount.py b/apps/rating/migrations/0003_viewcount.py
new file mode 100644
index 00000000..2a371371
--- /dev/null
+++ b/apps/rating/migrations/0003_viewcount.py
@@ -0,0 +1,27 @@
+# Generated by Django 2.2.4 on 2019-11-14 17:15
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('contenttypes', '0002_remove_content_type_name'),
+        ('rating', '0002_auto_20191004_0928'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ViewCount',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('object_id', models.PositiveIntegerField()),
+                ('count', models.IntegerField()),
+                ('content_type', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='contenttypes.ContentType')),
+            ],
+            options={
+                'unique_together': {('object_id', 'content_type')},
+            },
+        ),
+    ]
diff --git a/apps/rating/models.py b/apps/rating/models.py
index e1dcec86..7a49857e 100644
--- a/apps/rating/models.py
+++ b/apps/rating/models.py
@@ -20,3 +20,21 @@ class Rating(models.Model):
             return self.content_object.name
         if hasattr(self.content_object, 'title'):
             return self.content_object.title_translated
+
+
+class ViewCount(models.Model):
+    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
+    object_id = models.PositiveIntegerField()
+    content_object = GenericForeignKey('content_type', 'object_id')
+    count = models.IntegerField()
+
+    class Meta:
+        unique_together = ('object_id', 'content_type')
+
+    @property
+    def name(self):
+        # Check if Generic obj has name or title
+        if hasattr(self.content_object, 'name'):
+            return self.content_object.name
+        if hasattr(self.content_object, 'title'):
+            return self.content_object.title_translated
diff --git a/apps/rating/transfer_data.py b/apps/rating/transfer_data.py
new file mode 100644
index 00000000..29d5ad83
--- /dev/null
+++ b/apps/rating/transfer_data.py
@@ -0,0 +1,28 @@
+from transfer.models import PageTexts, PageCounters
+from news.models import News
+from rating.models import ViewCount
+
+
+def transfer_news_view_count():
+    news_list = News.objects.filter(old_id__isnull=False)
+    views_count_list = {}
+    for news_object in news_list:
+        try:
+            mysql_page_text = PageTexts.objects.get(id=news_object.old_id)
+        except PageTexts.DoesNotExist:
+            continue
+
+        if mysql_page_text.page_id not in views_count_list:
+            try:
+                mysql_views_count = PageCounters.objects.get(page_id=mysql_page_text.page_id)
+            except PageCounters.DoesNotExist:
+                continue
+
+            views_count_list[mysql_page_text.page_id] = ViewCount.objects.create(count=mysql_views_count.count)
+
+        news_object.views_count = views_count_list[mysql_page_text.page_id]
+        news_object.save()
+
+data_types = {
+    "rating_count": [transfer_news_view_count]
+}
diff --git a/apps/transfer/management/commands/transfer.py b/apps/transfer/management/commands/transfer.py
index fc01f655..48e3b74d 100644
--- a/apps/transfer/management/commands/transfer.py
+++ b/apps/transfer/management/commands/transfer.py
@@ -36,6 +36,7 @@ class Command(BaseCommand):
         'souvenir',
         'establishment_note',
         'assemblage',
+        'rating_count'
     ]
 
     def handle(self, *args, **options):
diff --git a/project/settings/local.py b/project/settings/local.py
index 09cadd4b..f9a096fe 100644
--- a/project/settings/local.py
+++ b/project/settings/local.py
@@ -36,14 +36,15 @@ THUMBNAIL_DEBUG = True
 
 
 # ADDED TRANSFER APP
-# INSTALLED_APPS.append('transfer.apps.TransferConfig')
+INSTALLED_APPS.append('transfer.apps.TransferConfig')
 
 
 # DATABASES
 DATABASES.update({
     'legacy': {
         'ENGINE': 'django.db.backends.mysql',
-        'HOST': 'mysql_db',
+        'HOST': '172.22.0.1',
+        # 'HOST': 'mysql_db',
         'PORT': 3306,
         'NAME': 'dev',
         'USER': 'dev',