loko/streetup/assets/migrations/0012_trafficlightintersection_is_radar.py
2026-07-22 14:48:40 +02:00

28 lines
1 KiB
Python

# Generated by Django 5.2.8 on 2026-06-22 12:43
from django.db import migrations, models
def populate_is_radar(apps, schema_editor):
TrafficLightIntersection = apps.get_model('assets', 'TrafficLightIntersection')
TrafficLightRadar = apps.get_model('assets', 'TrafficLightRadar')
# Intersections linked to at least one radar are radar intersections
radar_intersection_ids = TrafficLightRadar.objects.filter(intersection__isnull=False).values_list('intersection_id', flat=True).distinct()
TrafficLightIntersection.objects.filter(id__in=radar_intersection_ids).update(is_radar=True)
class Migration(migrations.Migration):
dependencies = [
('assets', '0011_create_radar_preventif_category'),
]
operations = [
migrations.AddField(
model_name='trafficlightintersection',
name='is_radar',
field=models.BooleanField(default=False, verbose_name='Est un radar ?'),
),
migrations.RunPython(populate_is_radar, migrations.RunPython.noop),
]