From d64e403646393089214675e45fba801c2436e4d2 Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Thu, 22 Mar 2018 21:35:23 -0700 Subject: make it run --- config.py | 8 ++++++++ run.py | 17 ++++++++++++++--- 2 files changed, 22 insertions(+), 3 deletions(-) create mode 100644 config.py mode change 100644 => 100755 run.py diff --git a/config.py b/config.py new file mode 100644 index 00000000..bad871dc --- /dev/null +++ b/config.py @@ -0,0 +1,8 @@ + +import os +basedir = os.path.abspath(os.path.dirname(__file__)) + +class Config(object): + SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URI') or \ + 'sqlite:///' + os.path.join(basedir, 'fatcat_dev.sqlite') + SQLALCHEMY_TRACK_MODIFICATIONS = False diff --git a/run.py b/run.py old mode 100644 new mode 100755 index c8f492ea..512d5828 --- a/run.py +++ b/run.py @@ -1,5 +1,7 @@ +#!/usr/bin/env python3 import argparse +import fatcat.sql from fatcat import app, db def main(): @@ -15,12 +17,21 @@ def main(): default=8040, help="listen on this port") parser.add_argument('--database-uri', - default=app.config['DATABASE_URI'], + default=app.config['SQLALCHEMY_DATABASE_URI'], help="sqlalchemy database string") + parser.add_argument('--init-db', + action='store_true', + help="create database tables and insert dummy data") args = parser.parse_args() - app.config['DATABASE_URI'] = args.database_uri - db.create_all() + app.config['SQLALCHEMY_DATABASE_URI'] = args.database_uri + + if args.init_db: + db.create_all() + fatcat.sql.populate_db() + print("Dummy database configured: " + app.config['SQLALCHEMY_DATABASE_URI']) + return + app.run(debug=args.debug, host=args.host, port=args.port) if __name__ == '__main__': -- cgit v1.2.3