Add command

apps/location/transfer_data.py

@@ -476,22 +476,22 @@ def fix_chosen_tag():
 
 def fix_location_models():
 
-    # try:
-    #     ruby_data_file = open(f"{settings.PROJECT_ROOT}/apps/location/ruby_data.py", "r")
-    #     ruby_data = json.loads(ruby_data_file.read())
-    # except FileNotFoundError:
-    #     ruby_data = get_ruby_data()
-    #
-    # add_correct_location_models(ruby_data)
-    #
-    # fix_location_address()
-    # fix_location_collection()
-    # fix_award_type()
-    # fix_role()
-    # fix_news()
-    # fix_reviews()
+    try:
+        ruby_data_file = open(f"{settings.PROJECT_ROOT}/apps/location/ruby_data.py", "r")
+        ruby_data = json.loads(ruby_data_file.read())
+    except FileNotFoundError:
+        ruby_data = get_ruby_data()
+
+    add_correct_location_models(ruby_data)
+
+    fix_location_address()
+    fix_location_collection()
+    fix_award_type()
+    fix_role()
+    fix_news()
+    fix_reviews()
     fix_tag_category()
-fix_chosen_tag()
+    fix_chosen_tag()
 
 
 data_types = {
@@ -513,7 +513,7 @@ data_types = {
     "migrate_city_gallery": [
         migrate_city_photos
     ],
-    "tmp": [
+    "fix_location": [
         fix_location_models
     ]
 

apps/transfer/management/commands/transfer.py

@@ -40,7 +40,8 @@ class Command(BaseCommand):
         'product_review',
         'newsletter_subscriber',  # подписчики на рассылку - переносить после переноса пользователей №1
         'update_city_info',
-        'migrate_city_gallery'
+        'migrate_city_gallery',
+        'fix_location'
     ]
 
     def handle(self, *args, **options):