almet/ihatemoney
1
0
Fork 0
mirror of https://github.com/spiral-project/ihatemoney.git synced 2025-05-02 03:02:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

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

fix #277
This commit is contained in:
JocelynDelalande 2017-12-27 17:28:18 +01:00 committed by Alexis Metaireau
parent f2a53eb1e3
commit 1c9120e68b
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
ihatemoney/manage.py
View file

@ -3,6 +3,7 @@
import os import os
import pkgutil import pkgutil
import random import random
import sys
from getpass import getpass from getpass import getpass
from flask_script import Manager, Command, Option from flask_script import Manager, Command, Option
@ -56,9 +57,20 @@ class ConfigTemplate(Command):
def main(): 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() app = create_app()
Migrate(app, db) Migrate(app, db)
# Restore stderr (among other: to be able to display help)
sys.stderr = backup_stderr
manager = Manager(app) manager = Manager(app)
manager.add_command('db', MigrateCommand) manager.add_command('db', MigrateCommand)
manager.add_command('generate_password_hash', GeneratePasswordHash) manager.add_command('generate_password_hash', GeneratePasswordHash)