From 785d2a2f0987ba09c1fd5907245d217097fc75d4 Mon Sep 17 00:00:00 2001
From: evgeniy-st <stoyushko.evgeniy@gmail.com>
Date: Fri, 27 Sep 2019 16:56:26 +0300
Subject: [PATCH] update news serializer, RUD for backoffice

---
 apps/news/migrations/0016_news_template.py | 18 ++++++++++++++++++
 apps/news/models.py                        | 12 ++++++++++++
 apps/news/serializers.py                   |  5 +++++
 3 files changed, 35 insertions(+)
 create mode 100644 apps/news/migrations/0016_news_template.py

diff --git a/apps/news/migrations/0016_news_template.py b/apps/news/migrations/0016_news_template.py
new file mode 100644
index 00000000..f85959ba
--- /dev/null
+++ b/apps/news/migrations/0016_news_template.py
@@ -0,0 +1,18 @@
+# Generated by Django 2.2.4 on 2019-09-27 13:54
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('news', '0015_auto_20190927_0853'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='news',
+            name='template',
+            field=models.PositiveIntegerField(choices=[(0, 'newspaper'), (1, 'main.pdf.erb'), (2, 'main')], default=0),
+        ),
+    ]
diff --git a/apps/news/models.py b/apps/news/models.py
index 4fb8e118..e0c5fbf4 100644
--- a/apps/news/models.py
+++ b/apps/news/models.py
@@ -49,6 +49,17 @@ class NewsQuerySet(models.QuerySet):
 class News(BaseAttributes, TranslatedFieldsMixin):
     """News model."""
 
+    # template choices
+    NEWSPAPER = 0
+    MAIN_PDF_ERB = 1
+    MAIN = 2
+
+    TEMPLATE_CHOICES = (
+        (NEWSPAPER, 'newspaper'),
+        (MAIN_PDF_ERB, 'main.pdf.erb'),
+        (MAIN, 'main'),
+    )
+
     news_type = models.ForeignKey(NewsType, on_delete=models.PROTECT,
                                   verbose_name=_('news type'))
     title = TJSONField(blank=True, null=True, default=None,
@@ -78,6 +89,7 @@ class News(BaseAttributes, TranslatedFieldsMixin):
                                 verbose_name=_('Image URL path'))
     preview_image_url = models.URLField(blank=True, null=True, default=None,
                                         verbose_name=_('Preview image URL path'))
+    template = models.PositiveIntegerField(choices=TEMPLATE_CHOICES, default=NEWSPAPER)
     address = models.ForeignKey('location.Address', blank=True, null=True,
                                 default=None, verbose_name=_('address'),
                                 on_delete=models.SET_NULL)
diff --git a/apps/news/serializers.py b/apps/news/serializers.py
index 3d15cff9..82f9980b 100644
--- a/apps/news/serializers.py
+++ b/apps/news/serializers.py
@@ -89,6 +89,9 @@ class NewsBackOfficeDetailSerializer(NewsBackOfficeBaseSerializer,
         source='country', write_only=True,
         queryset=location_models.Country.objects.all())
 
+    template_display = serializers.CharField(source='get_template_display',
+                                             read_only=True)
+
     class Meta(NewsBackOfficeBaseSerializer.Meta, NewsDetailSerializer.Meta):
         """Meta class."""
 
@@ -97,5 +100,7 @@ class NewsBackOfficeDetailSerializer(NewsBackOfficeBaseSerializer,
                     'description',
                     'news_type_id',
                     'country_id',
+                    'template',
+                    'template_display',
                  )