def test_entities_deletion(self):
dao = Dao()
f1 = FeedInfo("F1")
a1 = Agency("F1",
"A1",
"Agency 1",
agency_url="http://www.agency.fr/",
agency_timezone="Europe/Paris")
dao.add_all([f1, a1])
for feed in dao.feeds():
dao.delete(feed)
self.assertTrue(len(dao.feeds()) == 0)
def main():
arguments = docopt(__doc__,
version='gtfsdbloader %s' % gtfslib.__version__)
if arguments['--id'] is None:
arguments['--id'] = ""
# TODO Configure logging properly?
logger = logging.getLogger('libgtfs')
logger.setLevel(logging.INFO)
logger.addHandler(StreamHandler(sys.stdout))
dao = Dao(arguments['<database>'],
sql_logging=arguments['--logsql'],
schema=arguments['--schema'])
if arguments['--list']:
for feed in dao.feeds():
print(feed.feed_id if feed.feed_id != "" else "(default)")
if arguments['--delete'] or arguments['--load']:
feed_id = arguments['--id']
existing_feed = dao.feed(feed_id)
if existing_feed:
logger.warn("Deleting existing feed ID '%s'" % feed_id)
dao.delete_feed(feed_id)
dao.commit()
if arguments['--load']:
dao.load_gtfs(arguments['--load'],
feed_id=arguments['--id'],
lenient=arguments['--lenient'],
disable_normalization=arguments['--disablenormalize'])
def test_entities_creation(self):
dao = Dao()
f1 = FeedInfo("F1")
a1 = Agency("F1",
"A1",
"Agency 1",
agency_url="http://www.agency.fr/",
agency_timezone="Europe/Paris")
r1 = Route("F1",
"R1",
"A1",
3,
route_short_name="R1",
route_long_name="Route 1")
r2 = Route("F1", "R2", "A1", 3, route_short_name="R2")
c1 = Calendar("F1", "C1")
c1.dates = [
CalendarDate.ymd(2015, 11, 13),
CalendarDate.ymd(2015, 11, 14)
]
dao.add_all([f1, a1, r1, r2, c1])
self.assertTrue(len(dao.feeds()) == 1)
self.assertTrue(len(dao.agencies()) == 1)
a1b = dao.agency("A1", feed_id="F1", prefetch_routes=True)
self.assertTrue(a1b.agency_name == "Agency 1")
self.assertTrue(len(a1b.routes) == 2)
r1b = dao.route("R1", feed_id="F1")
self.assertTrue(r1b.route_short_name == "R1")
self.assertTrue(r1b.route_long_name == "Route 1")
self.assertTrue(r1b.route_type == 3)
r42 = dao.route("R42", feed_id="F1")
self.assertTrue(r42 is None)
