From 17ca27c7515e6b2fb004f26c933bc7e05aa7e653 Mon Sep 17 00:00:00 2001
From: littlewolf <wolfverwolf@mail.ru>
Date: Fri, 6 Dec 2019 15:28:58 +0300
Subject: [PATCH] Temp fiz

---
 apps/location/transfer_data.py                | 74 ++++++++++++++++++-
 apps/transfer/management/commands/transfer.py |  3 +-
 2 files changed, 75 insertions(+), 2 deletions(-)

diff --git a/apps/location/transfer_data.py b/apps/location/transfer_data.py
index 178c4312..aaddf25d 100644
--- a/apps/location/transfer_data.py
+++ b/apps/location/transfer_data.py
@@ -541,6 +541,77 @@ def transfer_city_gallery():
           f'Already added: {gallery_obj_exists_counter}')
 
 
+def add_fake_country():
+    # add country
+    country_data = {
+        "name": '{"en-GB": "Antilles Guyane West Indies"}',
+        "code": "aa"
+    }
+    country_mysql_ids = transfer_models.Cities.objects.filter(country_code_2="AA").only("id")
+    if len(country_mysql_ids) < 1:
+        raise ValueError("Incorrect country code")
+
+    country_data['mysql_ids'] = list(country_mysql_ids.values_list('id', flat=True))
+
+    country = Country.objects.create(**country_data)
+
+    regions_data = [
+        {
+            "name": "Guadeloupe",
+            "code": "GP",
+            "subregions": [
+                {
+                    "name": "Basse Terre",
+                    "code": "",
+                },
+                {
+                    "name": "Grande Terre",
+                    "code": "",
+                },
+            ]
+        },
+        {
+            "name": "Martinique",
+            "code": "MQ",
+        },
+        {
+            "name": "Guyane",
+            "code": "GY",
+        },
+        {
+            "name": "St Barthelemy",
+            "code": "BL",
+        },
+        {
+            "name": "St Martin",
+            "code": "MF",
+        },
+        {
+            "name": "Sainte-Lucie",
+            "code": "LC",
+        },
+    ]
+    regions = {}
+
+    for region_data in regions_data:
+        subregions = region_data['subregions'] if "subregions" in region_data else False
+
+        region_name = region_data['name']
+        region_data['name'] = '{"en-GB": "' + region_name + '"}'
+        region_data['country'] = country
+
+        regions[region_name] = Region.objects.create(**region_data)
+
+        if subregions:
+            for subregion_data in subregions:
+                subregion_name = subregion_data['name']
+                subregion_data['name'] = '{"en-GB": "' + subregion_name + '"}'
+                subregion_data['country'] = country
+                subregion_data['parent_region'] = regions[region_name]
+
+                regions[subregion_name] = Region.objects.create(**subregion_data)
+
+
 data_types = {
     "dictionaries": [
         transfer_countries,
@@ -565,6 +636,7 @@ data_types = {
     ],
     "remove_old_locations":[remove_old_records],
     
-    "fill_city_gallery": [transfer_city_gallery]
+    "fill_city_gallery": [transfer_city_gallery],
+    "add_fake_country": [add_fake_country]
 
 }
diff --git a/apps/transfer/management/commands/transfer.py b/apps/transfer/management/commands/transfer.py
index b2193d4c..5d69412a 100644
--- a/apps/transfer/management/commands/transfer.py
+++ b/apps/transfer/management/commands/transfer.py
@@ -44,7 +44,8 @@ class Command(BaseCommand):
         'update_city_info',
         'migrate_city_gallery',
         'fix_location',
-        'remove_old_locations'
+        'remove_old_locations',
+        'add_fake_country'
     ]
 
     def handle(self, *args, **options):