optional agency_id #122

bliksemlabs · Sep 22, 2014 · c73548d · c73548d
1 parent 2ce1673
commit c73548d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/gtfsdb.py b/gtfsdb.py
@@ -522,8 +522,12 @@ def agency_timezones(self):
         return list(x[0] for x in self.get_cursor().execute( query,() ))
 
     def agency(self,agency_id):
-        query = "SELECT agency_id,agency_name,agency_url,agency_phone,agency_timezone FROM agency WHERE agency_id = ?"
-        res = list(self.get_cursor().execute( query,(agency_id,) ))
+        if agency_id == "__DEFAULT__": # Likely only one entry
+            query = "SELECT agency_id,agency_name,agency_url,agency_phone,agency_timezone FROM agency"
+            res = list(self.get_cursor().execute( query ))
+        else:
+            query = "SELECT agency_id,agency_name,agency_url,agency_phone,agency_timezone FROM agency WHERE agency_id = ?"
+            res = list(self.get_cursor().execute( query,(agency_id,) ))
         if len(res) == 0:
             raise Exception('feed error: agency_id not found')
         elif len(res) > 1:

diff --git a/timetable.py b/timetable.py
@@ -631,7 +631,10 @@ def time_string_from_seconds(sec) :
     agencyUrls.append('')
 else:
     for agency_id,agency_name,agency_url,agency_phone,agency_timezone in [db.agency(agencyId) for agencyId,idx in sorted_agencyIds]:
-        agencyIds.append(agency_id)
+        if agency_id == None: # agency_id is an optional value in GTFS
+            agencyIds.append('')
+        else:
+            agencyIds.append(agency_id)
         agencyNames.append(agency_name)
         agencyUrls.append(agency_url)
 write_text_comment("AGENCY IDS")