create_tables.py

#!/usr/bin/python3
"""Usage: create_tables.py [-c <file>]

Create tables for the GTFS database
Options:
  -c <file>, --config <file>  Use the specified configuration file.
"""
# Copyright (C) 2018 Elad Alfassa <elad@fedoraproject.org>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import asyncio
from docopt import docopt
from curlbus.gtfs import model
import configparser
import gino
db = model.db


async def main():
    arguments = docopt(__doc__)
    configfile = arguments['--config'] or "config.ini"
    config = configparser.ConfigParser()
    config.read(configfile)
    engine = await gino.create_engine(config['gino']['dsn'])
    db.bind = engine
    await db.gino.create_all()

if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())
    loop.close()