diff options
| -rwxr-xr-x | ihatemoney/manage.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/ihatemoney/manage.py b/ihatemoney/manage.py index c1821c5..797e6c4 100755 --- a/ihatemoney/manage.py +++ b/ihatemoney/manage.py @@ -3,6 +3,7 @@ import os import pkgutil import random +import sys from getpass import getpass from flask_script import Manager, Command, Option @@ -56,9 +57,20 @@ class ConfigTemplate(Command): def main(): + QUIET_COMMANDS = ('generate_password_hash', 'generate-config') + + backup_stderr = sys.stderr + # Hack to divert stderr for commands generating content to stdout + # to avoid confusing the user + if len(sys.argv) > 1 and sys.argv[1] in QUIET_COMMANDS: + sys.stderr = open(os.devnull, 'w') + app = create_app() Migrate(app, db) + # Restore stderr (among other: to be able to display help) + sys.stderr = backup_stderr + manager = Manager(app) manager.add_command('db', MigrateCommand) manager.add_command('generate_password_hash', GeneratePasswordHash) |