31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
from django.core.management.base import BaseCommand
|
|
|
|
from establishment.models import Establishment
|
|
from transfer.models import Descriptions
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Add description values from old db to new db'
|
|
|
|
def handle(self, *args, **kwargs):
|
|
count = 0
|
|
|
|
queryset = Descriptions.objects.all()
|
|
for obj in queryset:
|
|
try:
|
|
establishment = Establishment.objects.get(old_id=obj.establishment.id)
|
|
except Establishment.DoesNotExist:
|
|
continue
|
|
except Establishment.MultipleObjectsReturned:
|
|
establishment = Establishment.objects.filter(old_id=obj.establishment.id).first()
|
|
else:
|
|
description = establishment.description
|
|
description.update({
|
|
obj.locale: obj.text
|
|
})
|
|
establishment.description = description
|
|
establishment.save()
|
|
count += 1
|
|
|
|
self.stdout.write(self.style.WARNING(f'Updated {count} objects.'))
|