From c61d5321a99e674d7b1f2ae4039e719851fcf63c Mon Sep 17 00:00:00 2001
From: littlewolf <nickptic@code-artel.com>
Date: Fri, 25 Oct 2019 09:52:35 +0300
Subject: [PATCH] Tmp commit

---
 apps/location/transfer_data.py                | 16 +++++++++++-
 apps/transfer/management/commands/transfer.py |  4 ++-
 apps/transfer/serializers/location.py         | 25 +++++++++++++++++--
 3 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/apps/location/transfer_data.py b/apps/location/transfer_data.py
index 04c2b163..745ff23c 100644
--- a/apps/location/transfer_data.py
+++ b/apps/location/transfer_data.py
@@ -1,4 +1,4 @@
-from transfer.serializers.location import CountrySerializer
+from transfer.serializers.location import CountrySerializer, RegionSerializer
 from transfer.models import Cities
 from pprint import pprint
 
@@ -12,10 +12,24 @@ def transfer_countries():
     else:
         pprint(f"Country serializer errors: {serialized_data.errors}")
 
+def transfer_regions():
+    regions_without_subregion_queryset = Cities.objects.\
+        exclude(subregion_code__isnull=False).values('region_code', 'country_code_2', 'subregion_code').distinct()
+
+    serialized_without_subregion = RegionSerializer(data=list(regions_without_subregion_queryset.values()), many=True)
+    if serialized_without_subregion.is_valid():
+        serialized_without_subregion.save()
+    else:
+        pprint(f"Parent regions serializer errors: {serialized_without_subregion.errors}")
+
+
 
 data_types = {
     "dictionaries": [
         transfer_countries,
 
+    ],
+    "tmp": [
+        transfer_regions
     ]
 }
diff --git a/apps/transfer/management/commands/transfer.py b/apps/transfer/management/commands/transfer.py
index d4a1c213..96b4155a 100644
--- a/apps/transfer/management/commands/transfer.py
+++ b/apps/transfer/management/commands/transfer.py
@@ -15,7 +15,9 @@ class Command(BaseCommand):
         'subscriber',
         'recipe',
         'partner',
-        'gallery'
+        'gallery',
+
+        'tmp'
     ]
 
     def handle(self, *args, **options):
diff --git a/apps/transfer/serializers/location.py b/apps/transfer/serializers/location.py
index f3fe44e6..a6d000ef 100644
--- a/apps/transfer/serializers/location.py
+++ b/apps/transfer/serializers/location.py
@@ -1,5 +1,5 @@
 from rest_framework import serializers
-from location.models import Country
+from location.models import Country, Region
 
 
 class CountrySerializer(serializers.ModelSerializer):
@@ -22,5 +22,26 @@ class CountrySerializer(serializers.ModelSerializer):
         return Country
 
     def get_country_code(self, obj):
-        print(f"OBJECT: {obj}")
         return obj.get("country_code_2")
+
+
+class RegionSerializer(serializers.ModelSerializer):
+    region_code = serializers.CharField()
+    subregion_code = serializers.CharField()
+    country_code_2 = serializers.CharField()
+
+    class Meta:
+        model = Region
+        fields = (
+            "region_code",
+            "country_code_2",
+            "subregion_code"
+        )
+
+    def validate(self, data):
+        return data
+
+    def create(self, validated_data):
+        # Some regions may be already in database
+        region, _ = Region.objects.get_or_create(**validated_data)
+        return region