|
29 | 29 | from collections import OrderedDict |
30 | 30 | from shared.common.db_utils import get_gtfs_feeds_query, get_gtfs_rt_feeds_query |
31 | 31 |
|
32 | | -from helpers.database import Database |
| 32 | +from shared.helpers.database import Database |
33 | 33 |
|
34 | 34 | load_dotenv() |
35 | 35 | csv_default_file_path = "./output.csv" |
@@ -112,28 +112,28 @@ def export_csv(request=None): |
112 | 112 |
|
113 | 113 | print(f"Retrieved {len(gtfs_rt_feeds)} GTFS realtime feeds.") |
114 | 114 |
|
115 | | - except Exception as error: |
116 | | - print(f"Error retrieving feeds: {error}") |
117 | | - raise Exception(f"Error retrieving feeds: {error}") |
| 115 | + data_collector = DataCollector() |
118 | 116 |
|
119 | | - data_collector = DataCollector() |
| 117 | + for feed in gtfs_feeds: |
| 118 | + # print(f"Processing feed {feed.stable_id}") |
| 119 | + data = get_feed_csv_data(feed) |
120 | 120 |
|
121 | | - for feed in gtfs_feeds: |
122 | | - # print(f"Processing feed {feed.stable_id}") |
123 | | - data = get_feed_csv_data(feed) |
| 121 | + for key, value in data.items(): |
| 122 | + data_collector.add_data(key, value) |
| 123 | + data_collector.finalize_row() |
| 124 | + print(f"Procewssed {len(gtfs_feeds)} GTFS feeds.") |
124 | 125 |
|
125 | | - for key, value in data.items(): |
126 | | - data_collector.add_data(key, value) |
127 | | - data_collector.finalize_row() |
128 | | - print(f"Procewssed {len(gtfs_feeds)} GTFS feeds.") |
| 126 | + for feed in gtfs_rt_feeds: |
| 127 | + # print(f"Processing rt feed {feed.stable_id}") |
| 128 | + data = get_gtfs_rt_feed_csv_data(feed) |
| 129 | + for key, value in data.items(): |
| 130 | + data_collector.add_data(key, value) |
| 131 | + data_collector.finalize_row() |
| 132 | + print(f"Processed {len(gtfs_rt_feeds)} GTFS realtime feeds.") |
129 | 133 |
|
130 | | - for feed in gtfs_rt_feeds: |
131 | | - # print(f"Processing rt feed {feed.stable_id}") |
132 | | - data = get_gtfs_rt_feed_csv_data(feed) |
133 | | - for key, value in data.items(): |
134 | | - data_collector.add_data(key, value) |
135 | | - data_collector.finalize_row() |
136 | | - print(f"Processed {len(gtfs_rt_feeds)} GTFS realtime feeds.") |
| 134 | + except Exception as error: |
| 135 | + print(f"Error retrieving feeds: {error}") |
| 136 | + raise Exception(f"Error retrieving feeds: {error}") |
137 | 137 |
|
138 | 138 | data_collector.write_csv(csv_file_path) |
139 | 139 |
|
|
0 commit comments