如何在登录成功或失败后显示函数定义的消息

def displayMessage(status, message):
    if status == 'success':
        print('登录成功！' + message)
    elif status == 'failure':
        print('登录失败！' + message)
    else:
        print('未知状态！' + message)

def login(username, password):
    # 登录验证逻辑
    if isValidCredentials(username, password):
        displayMessage('success', '欢迎回来，' + username)
    else:
        displayMessage('failure', '用户名或密码错误')

username = input('请输入用户名：')
password = input('请输入密码：')

login(username, password)