欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

三次登入冻结操作

程序员文章站 2022-04-28 14:38:34
要求: 1.将用户的密码以密文的形式存储 2.用户输错三次之后,再次输入此账户,显示'此账户已被冻结,无法登入' 思路: 1.先在运行脚本的路径下创建两个文件:一个存在用户信息的文件,一个存储冻结用户的文件 2.用户先注册:需要判断用户名是否存在 3.用户登入: - 先判断用户输入的用户是否在冻结文 ......

要求:

1.将用户的密码以密文的形式存储

2.用户输错三次之后,再次输入此账户,显示'此账户已被冻结,无法登入'

思路:

1.先在运行脚本的路径下创建两个文件:一个存在用户信息的文件,一个存储冻结用户的文件

2.用户先注册:需要判断用户名是否存在

3.用户登入:

  - 先判断用户输入的用户是否在冻结文件中

  - 如果不在,再继续匹配用户名和密码

  - 当用户输错三次账户或密码的时候,将此用户名写入到冻结文件中

 

代码如下:

import hashlib

def get_md5(user,pwd):
    md5 = hashlib.md5(user.encode('utf-8'))  # md5 算法加密
    md5.update(pwd.encode('utf-8'))
    return md5.hexdigest()

def register():
    '''
    用户注册
    :return:
    '''
    while true:
        print('*****用户注册*****')
        user = input('请输入账户(输入n返回上一级):')
        if user.upper() == 'n': return
        pwd = input('请输入密码:')
        flag = false
        with open('userinfo',mode='r',encoding='utf-8') as f:  #  打开用户文件,判断用户名是否存在
            for line in f:
                if user in line.strip().split('|'):
                    flag = true
        if flag:
            print('用户名已存在,请重新输入')
        else:
            with open('userinfo',mode='a',encoding='utf-8') as f:  # 将用户注册的信息写入到用户文件中
                f.write('%s|%s\n' %(user,get_md5(user,pwd)))  # 将用户的密码以密文的形式存储在文件中
                print('\033[1;31m注册成功\033[0m')

def login():
    '''
    用户登入
    :return:
    '''
    count = 1
    while true:
        print('*****用户登入*****')
        user = input('请输入账户(输入n返回上一级):')
        if user.upper() == 'n': return
        pwd = input('请输入密码:')
        with open('freeze',mode='r',encoding='utf-8') as f:   # 打开冻结文件,判断用户输入的用户名是否在冻结文件中
            content = f.read().strip().split('|')
            if user in content:
                print('此用户已冻结,请重新输入账户')
                continue
        flag = false
        new_name = false
        with open('userinfo',mode='r',encoding='utf-8') as f:
            for line in f:
                a,b  = line.strip().split('|')
                if user == a:      # 判断用户名是否存在
                    new_name = true
                if a == user and b == get_md5(user,pwd):  # 判断账户密码是否正确
                    flag = true
                    print('\033[1;34m登入成功\033[0m')
                    exit()
        if not new_name:
            print('账户不存在请重新输入')
            continue
        if not flag:
            print('账号或密码错误,请重新输入')
            count += 1
            if count == 4:
                print('输错三次,已被冻结')
                with open('freeze',mode='w',encoding='utf-8') as f:  # 将输错三次的用户名写入到冻结文件中
                    f.write(user+ '\n')
                    return

def run():
    '''
    程序入口
    :return:
    '''
    info = {'1':register,'2':login}
    while true:
        print('''管理界面
        1.用户注册
        2.用户登入''')
        choice = input('请选择服务(输入n退出):')
        if choice.upper() == 'n': break
        data = info.get(choice)
        if not data:
            print('输入有误,请重新输入')
            continue
        data()


if __name__ == '__main__':
    run()