Add Github login

[cheatdb.git] / setup.py

diff --git a/setup.py b/setup.py
index 0ba2b5a78379c30901c62e1222165c00558b4a7a..9949776bcdf49c4f48e19f33bc07dcb0a39722e3 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -6,12 +6,15 @@ if delete_db and os.path.isfile("app/data.sqlite"):
        os.remove("app/data.sqlite")
 
 if not os.path.isfile("app/data.sqlite"):
-       from app import models
+       from app.models import *
 
        print("Creating database tables...")
-       models.db.create_all()
-
+       db.create_all()
        print("Filling database...")
-       models.db.session.commit()
+
+       ruben = User("rubenwardy")
+       ruben.github_username = "rubenwardy"
+       db.session.add(ruben)
+       db.session.commit()
 else:
        print("Database already exists")