make it run

2 files changed, 22 insertions, 3 deletions

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