From cd0a14a1235888ec41d062d086e36e92543f0826 Mon Sep 17 00:00:00 2001
From: littlewolf <nickptic@code-artel.com>
Date: Wed, 20 Nov 2019 23:20:28 +0300
Subject: [PATCH] Add reviews transfer fix

---
 apps/location/transfer_data.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/apps/location/transfer_data.py b/apps/location/transfer_data.py
index 0d4e4c96..2406075a 100644
--- a/apps/location/transfer_data.py
+++ b/apps/location/transfer_data.py
@@ -11,6 +11,7 @@ from requests import get
 from main.models import AwardType
 from account.models import Role
 from news.models import News
+from review.models import Review
 
 
 def transfer_countries():
@@ -436,6 +437,18 @@ def fix_news():
         news_item.save()
 
 
+def fix_reviews():
+    reviews = Review.objects.filter(country__isnull=False)
+    for review in reviews:
+        try:
+            correct_country = Country.objects.get(code=review.country.code, mysql_ids__isnull=False)
+        except Country.DoesNotExist:
+            continue
+
+        review.country = correct_country
+        review.save()
+
+
 def fix_location_models():
 
     # try:
@@ -450,7 +463,8 @@ def fix_location_models():
     # fix_location_collection()
     # fix_award_type()
     # fix_role()
-    fix_news()
+    # fix_news()
+    fix_reviews()
 
 data_types = {
     "dictionaries": [