在不停止部分Progam的情况下捕获Python 2.7中的警告

Question

I'm currently working on a Python program which involves logging in with a username and password. I'm using the getpass module to accomplish this. My problem is that, in the case that getpass is unable to control echo, it spews the following into the terminal before continuing with the program.

Warning (from warnings module):
  File "C:\Python27\lib\getpass.py", line 92
    return fallback_getpass(prompt, stream)
GetPassWarning: Can not control echo on the terminal.
Warning: Password input may be echoed.

What I'd like to do is catch the warning and print my own custom message instead. The only code I can think of is the following which prevents the traceback from being shown but does not print the custom message at all.

import getpass

try:
    getpass.getpass("Password: ")
except getpass.GetPassWarning:
    print "Oh no!"

Output:

Warning: Password input may be echoed.
Password: 

I'd like to replace the text, Warning: Password input may be echoed. , with my own message ideally.

Answer 1

getpass uses the python builtin module warnings to show this warning message. You can filter/ignore them using various methods ( Python Docs / PyMOTW ).

You could catch the warning like so:

import getpass
import warnings

# the context manager resets the original
# filterwarnings after it has exited
with warnings.catch_warnings():
    # this will raise warnings of type (or inherited of type)
    # 'getpass.GetPassWarning' in the module 'getpass'
    warnings.filterwarnings(
        'error',
        category=getpass.GetPassWarning,
        module='getpass'
    )
    try:
        password = getpass.getpass('Password: ')
    except getpass.GetPassWarning:
        print 'Cannot get password on insecure platform'

Answer 2

The method suggested by @PM2Ring appears to be the best solution I've found so far. For the sake of others passing by and for the purpose of answering the question, I'll wrap everything up in this post.

The following method overwrites the fallback_getpass function in the getpass module, allowing one to control exactly what happens in the event of the fallback being required. It's a little very hacky but it gets the job done.

import getpass

def custom_fallback(prompt="Password: ",stream=None): #Clone of parameters from getpass.fallback_getpass
    print "Custom message." #Custom message goes here
    return getpass._raw_input(prompt) #Use getpass' custom raw_input function for security

getpass.fallback_getpass = custom_fallback #Replace the getpass.fallback_getpass function with our equivalent

password = getpass.getpass("Password: ") #Prompt for password

Output:

Custom message.
Password: