测试学习笔记（P72P76教程from凡云教育）之习题python代码

P72 

编写注册、登录代码

username = ['zhangsan', 'lisi', 'wangwu']
password = ['111111', '222222', '333333']

def reg():
    print('欢迎使用蜗牛ATM系统')
    un = input('请输入注册用户名：')
    if un in username:
        print('用户已注册！')
        return False #结束reg函数
    else:
        pw = input('请输入注册密码')
        if len(pw)<6:
            print('密码长度小于6位，请检查')
            return False
        else:
            username.append(un)
            password.append(pw)
            print('注册成功！')
            return True

def login():
    un = input('请输入用户名：')
    pw = input('请输入密码：')
    if un in username:
        i = username.index(un)
        if pw == password[i]:
            print('登录成功！')
            return True
        else:
            print('用户名或密码错误！')
            return
    else:
        print('用户名或密码错误！')

if reg():
login()

P73

优化，添加循环

def reg():
    print('欢迎使用蜗牛ATM系统')
    while True:
        un = input('请输入注册用户名：')
        if un in username:
            print('用户已注册！')
        else:
            break
    while True:
         pw = input('请输入注册密码')
         if len(pw)<6:
             print('密码长度小于6位，请检查')
         else:
             break

    username.append(un)
    password.append(pw)
    print('注册成功！')
    return True

def login():
    while True:
        un = input('请输入登录用户名：')
        pw = input('请输入登录密码：')
        if un in username:
            i = username.index(un)
            if pw == password[i]:
                print('登录成功！')
                return True
            else:
                print('用户名或密码错误！')
        else:
            print('用户名或密码错误！')


if reg():
login()

P74

用二位列表存储并添加菜单

users = [['zhangsan','111111'], [['lisi', '222222']], ['wangwu', '333333']]

def reg():
    print('欢迎使用蜗牛ATM系统')
    while True:
        un = input('请输入注册用户名：')
        for item in users:
            if un == item[0]:
                print('用户已注册！')
                break
        else: # python 中存在一个for else 语句，如果for中能遍历完，则接着执行else语句中的内容；如果for 未能便利完，则else 也不执行
         pw = input('请输入注册密码')
         if len(pw)<6:
             print('密码长度小于6位，请检查')
         else:
            users.append([un, pw])
            print('注册成功！')
            return True

def login():
    while True:
        un = input('请输入登录用户名：')
        pw = input('请输入登录密码：')
        for item in users:
            if un == item[0] and pw == item[1]:
                print('登录成功！')
                return True
        else:
            print('用户名或密码错误！')

def mymenu():
    menu = '''
    **************welcome to woniu ATM****************
    ***********please choose below potions************
    **********1. register 2. login 3. exit************
    '''
    while True:
        print(menu)
        option = input('请输入您要操作的菜单：')
        if option == '1':
            reg()
        elif option == '2':
            login()
        elif option == '3':
            print('感谢使用，欢饮下次再来！')
            break
        else:
            print('菜单不存在，请重新输入！')

mymenu()