作业11

1

2

3

4

5

6

7

1、实现登录成功记录下用户名

login_user = None
def login():
    inp_user = input('用户名：').strip()
    inp_pwd = input('密码：').strip()
    if inp_user == 'umi' and inp_pwd == '123':
        print("登录成功")
        global login_user
        login_user = inp_user
    else:
        print("登录失败")
login()
print(login_user)

2、针对查询余额的功能，添加额外的逻辑：如果用户没有登录，则先执行登录功能

login_user = None
def login():
    inp_user = input('用户名：').strip()
    inp_pwd = input('密码：').strip()
    if inp_user == 'umi' and inp_pwd == '123':
        print("登录成功")
        global login_user
        login_user = inp_user
    else:
        print("登录失败")

def balance(login_user):
    if login_user is None:
        print("请先登录")
        login()
    else:
        return True
    
while True:
    msg = balance(login_user)
    if msg:
        print("{}的余额".format(login_user))
        break