python实现简单的用户密码登录控制(输入三次就锁定用户)

问题描述

我们经常在登录一些网站的时候,发现我们如果连续的输错好几次密码,我们的帐号就被锁定起来了,那这个过程是如何实现的呢?本程序主要就是解决以下3件事情
1.输入用户名密码
2.认证成功并显示欢迎信息
3.输错三次后锁定

解决办法

# !/usr/bin/env python
# -*- coding: utf-8 -*-
# Time: 17-11-18 上午8:45
# Author: sty
# File: login_judge.py

# 1.输入用户名密码
# 2.认证成功并显示欢迎信息
# 3.输错三次后锁定
import json


# 初始化json文件
def utilize():
    data = [
        {'usr': 'sty', 'pwd': '123', 'lock': 0, 'cnt': 0},
        {'usr': 'bat', 'pwd': 'alibaba', 'lock': 0, 'cnt': 0}
    ]

    with open('data.json', 'w') as f:
        json.dump(data, f)

# login判断登录
def login():
    with open('data.json', 'r') as f:
        res = json.load(f)
    while True:
        new = []
        user_name = input('user_name:')
        user_pwd = input('user_pwd:')
        flag = 0
        # flag: 0表示该用户不存在 1表示用户存在密码正确 2表示用户名存在,但是被锁定或密码错误
        for user in res:
            if user['usr'] == user_name:
                flag = 2
                if user['pwd'] == user_pwd:
                    if user['lock'] == 1:
                        print('you have been locked')
                    else:
                        flag = 1
                        user['cnt'] = 0
                        print("welcome to file")
                else:
                    if user['lock'] == 1:
                        print('you have been locked')
                    else:
                        user['cnt'] = user['cnt'] + 1
                        # 如果user输入密码三次 锁定用户
                        if user['cnt'] == 3:
                            user['lock'] = 1
            new.append(user)
        # 文件中更新帐户数据
        with open('data.json', 'w') as f:
            json.dump(new, f)
        if flag == 1:
            break
        if flag == 0:
            print("this user is not exist")

if __name__ == "__main__":
    # 第一次使用初始化json文件
    # utilize()
    login()

增加一些改进

貌似很多网站还设置了一个时间记录,如果最近的一次登录时间和此次登录时间的间隔大于24小时了,你将继续获得三次登录机会,这个我们仅需要在上述的代码中添加一个处理时间的判断和相应的在json文件中添加time字段用来存储时间。需要注意的是在文件中使用到了一个格林威治时间与本地时间的装换。

# !/usr/bin/env python
# -*- coding: utf-8 -*-
# Time: 17-11-18 上午8:45
# Author: sty
# File: login_judge.py

# 1.输入用户名密码
# 2.认证成功并显示欢迎信息
# 3.输错三次后锁定
import json
import time
from datetime import datetime

# 初始化json文件
def utilize():
    data = [
        {'usr': 'sty', 'pwd': '123', 'lock': 0, 'cnt': 0, 'time': '2017-01-01 0:0:0'},
        {'usr': 'bat', 'pwd': 'alibaba', 'lock': 0, 'cnt': 0, 'time': '2017-01-01 0:0:0'}
    ]

    with open('data.json', 'w') as f:
        json.dump(data, f)

#判断间隔时间
def time_judge(start_time, end_time):
    day = (end_time - start_time).days
    return day

# UTC时间转本地时间(+8:00)
def utc2local(utc_st):
    now_stamp = time.time()
    local_time = datetime.fromtimestamp(now_stamp)
    utc_time = datetime.utcfromtimestamp(now_stamp)
    offset = local_time - utc_time
    local_st = utc_st + offset
    return local_st

# 本地时间转UTC时间(-8:00)
def local2utc(local_st):
    time_struct = time.mktime(local_st.timetuple())
    utc_st = datetime.utcfromtimestamp(time_struct)
    return utc_st

# login判断登录
def login():
    with open('data.json', 'r') as f:
        res = json.load(f)
    while True:
        new = []
        user_name = input('user_name:')
        user_pwd = input('user_pwd:')
        flag = 0
        # flag: 0表示该用户不存在 1表示用户存在密码正确 2表示用户名存在,但是被锁定或密码错误
        for user in res:
            if user['usr'] == user_name:
                flag = 2
                last_time = datetime.strptime(user['time'],"%Y-%m-%d %H:%M:%S")
                now_time = utc2local(datetime.utcnow())
                # 判断时间间隔
                if time_judge(last_time, now_time) > 1:
                    user['lock'] = 0
                    user['cnt'] = 0
                if user['pwd'] == user_pwd:
                    if user['lock'] == 1:
                        print('you have been locked')
                    else:
                        flag = 1
                        user['cnt'] = 0
                        print("welcome to file")
                else:
                    if user['lock'] == 1:
                        print('you have been locked')
                    else:
                        user['cnt'] = user['cnt'] + 1
                        # 如果user输入密码三次 锁定用户
                        if user['cnt'] == 3:
                            user['lock'] = 1
                user['time'] = now_time.strftime("%Y-%m-%d %H:%M:%S") #更新登录时间
            new.append(user)
        # 文件中更新帐户数据
        #print(new)
        with open('data.json', 'w') as f:
            json.dump(new, f)
        if flag == 1:
            break
        if flag == 0:
            print("this user is not exist")


if __name__ == "__main__":
    # 第一次使用初始化json文件
    #utilize()
    login()

转载请注明出处:
CSDN:楼上小宇__home:http://blog.csdn.net/u014265347

你可能感兴趣的:(Python)