自学Python之简易用户登陆

一、利用字典保存用户数据

　　本人小白，之前在学习了爬虫之后，对Pyhton有了一个深刻的印象，觉得这是一个相比其他常见编程语言而言较为简洁的编程语言。于是便决定利用自己空余的时间自学一下Pyhton。

 　　在经过了对其简单的了解之后，决定编写一些小代码来练习一下。于是便决定使用字典来实现用户的登陆。代码如下：

 

#info_users = {'M':'111','S':'222'} 
info_users = {} #保存用户账号及密码
info_users_locked ={} #保存被锁定的用户账号及密码
def user_login():
    count = {} #输入错误密码的计数
    while True:
        username = input('username:')
        password = input('password:')   #输入用户账号及密码
        for info_user_locked in info_users_locked:
            if username == info_user_locked:    #检测给用户是否被锁定
                print('this user was locked!')
        for info_user in info_users:        #根据账号检测密码是否正确
            if info_user == username:
                if info_users[info_user] == password:
                    print('Welcome user {} login....'.format(info_user))
                    return
                else:
                    print('invalid username or password')
                    count[username] += 1   #密码错误，该用户计数+1
                    #print(count)
                    break
        else:   #所输入的不存在，保存在字典中，并增加进计数
            info_users[username] = password
            count[username] = 0
            #print(info_users)
        if count[username] == 3:  #用户锁定条件
            print('user {} was locked'.format(info_user))
            info_users_locked[username] = info_users[username]





if __name__ == '__main__':
    user = user_login()

 

　代码较为简单，主要是利用字典保存用户的账号及密码，利用键值的关系，能够进行简单的登陆信息检测。

二、利用文件保存数据

　　在了解了文件的机制后，决定使用文件来保存数据，以此继续补充。用户数据为提前输入，代码如下：

 

import pickle

#获取被锁用户的数据
def get_locked_info():
    locked_users =[]
    try:                                 #此处不加入异常处理的话会由于文件为空导致报错
         with open('locked_users.txt','rb') as locked_users_file:
            locked_users = pickle.load(locked_users_file)
            #print(locked_users)
            locked_users_file.close()
            return locked_users
    except EOFError:
        return {}

#获取用户数据
def get_info():
    info_users = {}
    users_file = open('info_users.txt','rb')
    info_users = pickle.load(users_file)
    users_file.close()
    #print(info_users.keys())
    return info_users

#检测用户名及密码，并将输入密码错误3次的用户锁定保存在锁定文件中
def check(user,pwd,info_users,count,locked_user):
    for info_user in info_users:
        if user == info_user:
            if pwd == info_users[info_user]:
                print('Welcom to login!')
                break
            else:
                print('Error of username or password!')
                count[user] += 1 #密码错误计数
                if count[user] == 3:   #错误为3次，保存入另一个文件
                    locked_user.append(user)
                    #print(locked_user)
                    locked_users_file = open('locked_users.txt','wb')
                    pickle.dump(locked_user,locked_users_file)
                    locked_users_file.close()
                print(count)
                break
    else:
        print('Not find this id!')
    return locked_user

#判断该用户是否被锁
def isLocked(user,locked_users):
    for i in range(len(locked_users)):
        if user == locked_users[i]:
            #print(locked_users[i])
            print('This username was locked!')
            return 0
    return 1

#初始化错误计数
def init_count(infos):
    count = {}#保存各个用户所对应的错误次数
    for info in infos:
        count[info] = 0
    return count

if __name__ == '__main__':
    locked_user = []  # 用来保存将被锁定的用户名
    info_users = get_info() #获取用户数据
    count = init_count(info_users) #初始化计数
    while True :
        locked_users = get_locked_info()#获取被锁用户
        username = input('username:')
        password = input('password:')  #输入用户名及密码
        judge = isLocked(username,locked_users)
        if judge == 0:     #当所输入的用户名被锁，退出循环
            break
        check(username,password,info_users,count,locked_user) #检测密码是否正确

 

　　大致就是利用pickle来读取和写入用户的数据，然后在利用字典进行比较以此实现密码检测。