From a4a8cd9e9f77500c99f15b539deff8c9d9337b0e Mon Sep 17 00:00:00 2001 From: Emmy D'Anello Date: Mon, 27 May 2024 23:49:12 +0200 Subject: [PATCH] Drop transfers before populating new ones --- trainvel/gtfs/management/commands/update_trainvel_gtfs.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/trainvel/gtfs/management/commands/update_trainvel_gtfs.py b/trainvel/gtfs/management/commands/update_trainvel_gtfs.py index 862436d..ea550a0 100644 --- a/trainvel/gtfs/management/commands/update_trainvel_gtfs.py +++ b/trainvel/gtfs/management/commands/update_trainvel_gtfs.py @@ -9,6 +9,7 @@ from zoneinfo import ZoneInfo import requests from django.core.management import BaseCommand from django.db import transaction +from django.db.models import Q from tqdm import tqdm from trainvel.gtfs.models import Agency, Calendar, CalendarDate, FeedInfo, GTFSFeed, Route, Stop, StopTime, \ @@ -368,6 +369,7 @@ class Command(BaseCommand): stop_times.clear() if os.path.exists(os.path.join(zip_dir, "transfers.txt")): + Transfer.objects.filter(Q(from_stop__gtfs_feed_id=gtfs_code) | Q(to_stop__gtfs_feed_id=gtfs_code)).delete() transfers = [] for transfer_dict in read_csv("transfers.txt"): transfer_dict: dict