三次登入凍結操做

要求:

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()