Change folder structure

database.py → database/database.py

@@ -25,5 +25,5 @@ def init_db(engine=engine):
     # import all modules here that might define models so that
     # they will be registered properly on the metadata.  Otherwise
      # you will have to import them first before calling init_db()
-    import models
+    import database.models
     Base.metadata.create_all(bind=engine)

models.py → database/models.py

-from database import Base
+from database.database import Base
 from flask_security import UserMixin, RoleMixin
 from sqlalchemy import create_engine
 from sqlalchemy.orm import relationship, backref

test/test_database.py

@@ -4,8 +4,8 @@ from sqlalchemy import create_engine
 from sqlalchemy.orm import scoped_session, sessionmaker
 from sqlalchemy.ext.declarative import declarative_base
 from flask_security import SQLAlchemySessionUserDatastore
-from models import User, Role, Application
-from database import init_db
+from database.models import User, Role, Application
+from database.database import init_db
 
 class DatabaseSetupTest(unittest.TestCase):