summaryrefslogtreecommitdiffstats
path: root/run.py
diff options
context:
space:
mode:
Diffstat (limited to 'run.py')
-rwxr-xr-x[-rw-r--r--]run.py17
1 files changed, 14 insertions, 3 deletions
diff --git a/run.py b/run.py
index c8f492ea..512d5828 100644..100755
--- 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__':