make it run

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__':