refactor location transfer 1

apps/location/transfer_data.py

@@ -779,9 +779,9 @@ def set_unused_regions():
 
 data_types = {
     "dictionaries": [
-        # transfer_countries,
-        # transfer_regions,
-        # transfer_cities,
+        # transfer_countries, # !!! см setup_clean_db
+        # transfer_regions, # !!! см setup_clean_db
+        # transfer_cities, # !!! см setup_clean_db
         transfer_addresses,
         transfer_wine_region,
         transfer_wine_sub_region,

apps/review/transfer_data.py

@@ -1,9 +1,5 @@
 from pprint import pprint
 
-from django.db.models import Q
-
-from account.models import User
-from account.transfer_data import STOP_LIST
 from establishment.models import Establishment
 from product.models import Product
 from review.models import Inquiries as NewInquiries, Review
@@ -77,10 +73,7 @@ def make_en_text_review():
 
 def transfer_inquiries():
     reviews = Review.objects.all().values_list('old_id', flat=True)
-    inquiries = Inquiries.objects.exclude(
-        Q(account__confirmed_at__isnull=True) |
-        Q(account__email__in=STOP_LIST)
-    ).filter(review_id__in=list(reviews))
+    inquiries = Inquiries.objects.exclude(account__confirmed_at__isnull=True).filter(review_id__in=list(reviews))
 
     serialized_data = InquiriesSerializer(data=list(inquiries.values()), many=True)
     if serialized_data.is_valid():

apps/transfer/serializers/location.py

@@ -224,13 +224,11 @@ class AddressSerializer(serializers.ModelSerializer):
         return data
 
     def set_city(self, data):
-        try:
         city = models.City.objects.filter(mysql_id=data['city_id']).first()
-        except models.City.DoesNotExist as e:
-            raise ValueError(f"City not found with {data}: {e}")
+        if not city:
+            raise ValueError(f"City not found with ID {data['city_id']}")
 
-        data['city'] = city
-        del(data['city_id'])
+        data['city_id'] = city.id
         return data
 
     def set_address(self, data):
@@ -280,7 +278,6 @@ class AddressSerializer(serializers.ModelSerializer):
 
 
 class WineRegionSerializer(TransferSerializerMixin):
-
     id = serializers.IntegerField()
     name = serializers.CharField()
     desc = serializers.CharField(allow_null=True)
@@ -307,9 +304,7 @@ class WineRegionSerializer(TransferSerializerMixin):
         return attrs
 
 
-
 class WineSubRegionSerializer(WineRegionSerializer):
-
     id = serializers.IntegerField()
     name = serializers.CharField()
     parent_id = serializers.IntegerField()
@@ -336,7 +331,6 @@ class WineSubRegionSerializer(WineRegionSerializer):
 
 
 class WineVillageSerializer(TransferSerializerMixin):
-
     id = serializers.IntegerField()
     name = serializers.CharField()
     parent_id = serializers.IntegerField()
@@ -432,7 +426,6 @@ class CityMapCorrectSerializer(CityMapSerializer):
 
 
 class CepageWineRegionSerializer(TransferSerializerMixin):
-
     CATEGORY_LABEL = 'Cepage'
     CATEGORY_INDEX_NAME = slugify(CATEGORY_LABEL)
 

make_data_migration.sh

@@ -41,9 +41,41 @@
 # requester -> requester
 #./manage.py add_ownership
 
-
+# Заполнение моделей Country, Region, City из ruby_data
+# для City перенос из старой бд, заполнение флагов и фото
+# --------------------------
 ./manage.py transfer --setup_clean_db
+
+# Заполнение модели Address из Locations
+# --------------------------
+# id
+# city_id
+# zip_code
+# latitude
+# longitude
+# address
+
+# Заполнение модели WineRegion из WineLocations
+# --------------------------
+# id
+# name
+# desc
+# latitude
+# longitude
+
+# Заполнение модели WineSubRegion из WineVillage
+# --------------------------
+# id
+# name
+# parent_id
+
+# Заполнение модели WineSubRegion из WineLocations
+# --------------------------
+# id
+# name
+# parent_id
 ./manage.py transfer -d
+
 ./manage.py transfer -e
 ./manage.py transfer -n
 ./manage.py rm_empty_images  # команда для удаления картинок с относительным урлом из news.description