Add reviews transfer fix

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": [