diff --git a/apps/establishment/transfer_data.py b/apps/establishment/transfer_data.py index 4670e6cf..a5a5bc49 100644 --- a/apps/establishment/transfer_data.py +++ b/apps/establishment/transfer_data.py @@ -4,6 +4,8 @@ from django.db.models import Q from transfer.models import Establishments from transfer.serializers.establishment import EstablishmentSerializer +from establishment.models import Establishment +from location.models import Address def transfer_establishment(): @@ -77,6 +79,30 @@ def transfer_establishment(): pprint(f"Establishment serializer errors: {serialized_data.errors}") +def transfer_establishment_addresses(): + old_establishments = Establishments.objects.only("id", "location_id").exclude( + Q(type='Wineyard') | + Q(location__timezone__isnull=True) + ) + + for old_establishment in old_establishments: + try: + establishment = Establishment.objects.get(old_id=old_establishment.id) + except Establishment.DoesNotExist: + continue + + try: + location = Address.objects.get(old_id=old_establishment.location_id) + except Establishment.DoesNotExist: + continue + + establishment.address = location + establishment.save() + + data_types = { - "establishment": [transfer_establishment] + "establishment": [ + transfer_establishment, + transfer_establishment_addresses + ] }