Skip to content

Commit

Permalink
Cleans GTFS importer
Browse files Browse the repository at this point in the history
  • Loading branch information
pveigadecamargo committed May 28, 2024
1 parent 04bb956 commit ee4b967
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions aequilibrae/transit/transit_elements/pattern.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def __init__(self, route_id, gtfs_feed) -> None:
self.network_candidates = []
self.full_path: List[int] = []
self.fpath_dir: List[int] = []
self.pattern_mapping = []
self.pattern_mapping = pd.DataFrame([])
self.stops = []
self.__map_matching_error = {}

Expand Down Expand Up @@ -94,13 +94,14 @@ def save_to_database(self, conn: Connection, commit=True) -> None:
total_capacity, geometry) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ST_Multi(GeomFromWKB(?, ?)));"""
conn.execute(sql, data)

if self.pattern_mapping and self.shape:
if self.pattern_mapping.shape[0]:
sqlgeo = """insert into pattern_mapping (pattern_id, seq, link, dir, geometry)
values (?, ?, ?, ?, GeomFromWKB(?, ?));"""
sql = "insert into pattern_mapping (pattern_id, seq, link, dir) values (?, ?, ?, ?);"

if "wkb" in self.pattern_mapping.columns:
data = self.pattern_mapping[["pattern_id", "seq", "link_id", "dir", "wkb", "srid"]].to_records()
cols = ["pattern_id", "seq", "link_id", "dir", "wkb", "srid"]
data = self.pattern_mapping[cols].to_records(index=False)
conn.executemany(sqlgeo, data)
else:
data = self.pattern_mapping[["pattern_id", "seq", "link_id", "dir"]].to_records()
Expand Down

0 comments on commit ee4b967

Please sign in to comment.