added transfer for advertisement

apps/advertisement/migrations/0004_auto_20191025_0903.py

@@ -0,0 +1,18 @@
+# Generated by Django 2.2.4 on 2019-10-25 09:03
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('advertisement', '0003_auto_20190919_1344'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='advertisement',
+            name='block_level',
+            field=models.CharField(blank=True, max_length=10, null=True, verbose_name='Block level'),
+        ),
+    ]

apps/advertisement/models.py

@@ -15,7 +15,7 @@ class Advertisement(ImageMixin, ProjectBaseMixin, PlatformMixin):
     url = models.URLField(verbose_name=_('Ad URL'))
     width = models.PositiveIntegerField(verbose_name=_('Block width'))
     height = models.PositiveIntegerField(verbose_name=_('Block height'))
-    block_level = models.CharField(verbose_name=_('Block level'), max_length=10)
+    block_level = models.CharField(verbose_name=_('Block level'), max_length=10, blank=True, null=True)
     target_languages = models.ManyToManyField(Language)
 
     class Meta:

apps/advertisement/transfer_data.py

@@ -0,0 +1,20 @@
+from pprint import pprint
+from django.db.models import Value, IntegerField, F
+from transfer.models import Ads
+from transfer.serializers.advertisement import AdvertisementSerializer
+
+
+def transfer_advertisement():
+    queryset = Ads.objects.filter(href__isnull=False)
+
+    serialized_data = AdvertisementSerializer(data=list(queryset.values()), many=True)
+
+    if serialized_data.is_valid():
+        serialized_data.save()
+    else:
+        pprint(f"News serializer errors: {serialized_data.errors}")
+
+
+data_types = {
+    "commercial": [transfer_advertisement]
+}

apps/transfer/management/commands/transfer.py

@@ -15,7 +15,8 @@ class Command(BaseCommand):
         'subscriber',
         'recipe',
         'partner',
-        'gallery'
+        'gallery',
+        'commercial'
     ]
 
     def handle(self, *args, **options):

apps/transfer/models.py

@@ -788,3 +788,23 @@ class PageMetadata(MigrateMixin):
     class Meta:
         managed = False
         db_table = 'page_metadata'
+
+
+class Ads(MigrateMixin):
+    using = 'legacy'
+
+    site_id = models.IntegerField(blank=True, null=True)
+    href = models.CharField(max_length=255, blank=True, null=True)
+    start_at = models.DateTimeField(blank=True, null=True)
+    expire_at = models.DateTimeField(blank=True, null=True)
+    attachment_file_name = models.CharField(max_length=255, blank=True, null=True)
+    attachment_content_type = models.CharField(max_length=255, blank=True, null=True)
+    attachment_file_size = models.IntegerField(blank=True, null=True)
+    attachment_updated_at = models.DateTimeField(blank=True, null=True)
+    geometries = models.CharField(max_length=1024, blank=True, null=True)
+    created_at = models.DateTimeField()
+    updated_at = models.DateTimeField()
+
+    class Meta:
+        managed = False
+        db_table = 'ads'

apps/transfer/serializers/advertisement.py

@@ -0,0 +1,39 @@
+from rest_framework import serializers
+from advertisement.models import Advertisement
+import yaml
+
+
+class AdvertisementSerializer(serializers.ModelSerializer):
+    href = serializers.CharField()
+    geometries = serializers.CharField(max_length=1024)
+
+    class Meta:
+        model = Advertisement
+        fields = (
+            "href",
+            "geometries"
+        )
+
+    def validate(self, data):
+        data["url"] = data["href"]
+        data["width"] = self.get_width(data["geometries"])
+        data["height"] = self.get_height(data["geometries"])
+        data.pop("href")
+        data.pop("geometries")
+        return data
+
+    def create(self, validated_data):
+        return Advertisement.objects.create(**validated_data)
+
+    def get_width(self, data):
+        data = self.parse_geometries(data)
+        return int(float(data["width"]))
+
+    def get_height(self, data):
+        data = self.parse_geometries(data)
+        return int(float(data["height"]))
+
+    def parse_geometries(self, geo_str):
+        clear_str = "!ruby/object:Paperclip::Geometry"
+        content_dict = yaml.safe_load(geo_str.replace(clear_str, ''))
+        return content_dict[':original']