from setup_django import *

from coord.models import Bus as OldModel
from transport.models import Bus as NewModel

created = skipped = 0

for old in OldModel.objects.all():
    obj, was_created = NewModel.objects.get_or_create(
        id=old.id,
        defaults={
            "company_id": old.company_id,
            "route_name": old.route_name,
            "contract_number": old.contract_number,
            "registration": old.registration,
            "seating_capacity": old.seating_capacity,
            "make": old.make,
            "model": old.model,
            "notes": old.notes,
        }
    )

    if was_created:
        created += 1
    else:
        skipped += 1

print(f"Buses — created: {created}, skipped: {skipped}")
print("Buses:", OldModel.objects.count(), NewModel.objects.count())