Silent stderr output during app init for some commands (#293)

Avoid confusing the user for the commands outputing text to the user.

fix #277

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)