From 629037c861c2ceecb43cd7503cbabc21524bd3ea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alexis=20M=C3=A9taireau?= <alexis@notmyidea.org>
Date: Wed, 28 Jun 2017 20:48:56 +0200
Subject: [PATCH] Fix manage.py imports

---
 budget/manage.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/budget/manage.py b/budget/manage.py
index 8bb44452..0a662840 100755
--- a/budget/manage.py
+++ b/budget/manage.py
@@ -5,6 +5,8 @@ from flask_script import Manager, Command
 from flask_migrate import Migrate, MigrateCommand
 from werkzeug.security import generate_password_hash
 
+from .run import app
+from .models import db
 
 
 class GeneratePasswordHash(Command):
@@ -20,5 +22,10 @@ manager = Manager(app)
 manager.add_command('db', MigrateCommand)
 manager.add_command('generate_password_hash', GeneratePasswordHash)   
 
-if __name__ == '__main__':
+
+def main():
     manager.run()
+
+
+if __name__ == '__main__':
+    main()