from setup_django import *

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

created = skipped = 0

for old in OldModel.objects.all():
    obj, was_created = NewModel.objects.get_or_create(
        id=old.id,
        defaults={
            "bus_id": old.bus_id,
            "school_id": old.school_id,
            "custom_name": old.custom_name,
            "am_service": old.am_service,
            "pm_service": old.pm_service,
        },
    )

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

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