python练习_简单登录

python练习_简单登录

需求：

1. 做一个登录的小程序，验证用户与密码
2. 要求登录三次错误后锁定

以下代码实现的功能与思路：

功能:

    1、除admin以外的用户累计登录失败次数超过三次则锁定，此时需要管理员解锁。
    2、当前IP地址登录失败五次则锁定IP地址一分钟后解锁。
    3、admin管理员有以下管理功能 锁定用户、解锁用户、更改密码、添加用户、删除用户。
    4、普通用户拥有更改密码的功能。

流程图：

思路：

1、用户信息通过ConfigParser模块进行处理并保存到文件

2、通过保存文件中counter 判断登录失败次数

3、通过socket获取当前计算机IP，如果失败把IP记录到文件中，并加入通过time获取的当前时间戳

4、管理功能通过ConfigParser模块中的一些功能处理存储用户信息的文件即可

代码Python2.7执行环境：

使用方式：

在linux上直接运行main.py即可
windows运行请把 225行解除注释，把226行注释

测试账号：
用户admin 密码：admin.123
用户test 密码：admin.123
用户test1 密码：admin.123 test1 默认锁定状态

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import time
import ConfigParser
import socket
import webbrowser
import sys
import getpass

# 全局变量 localIP 获取本地ip用于锁定本地ip，user_config操作用户文件句柄
LOCAL_IP = socket.gethostbyname(socket.gethostname())
user_config = ConfigParser.ConfigParser()
user_config.read('userdb')

class Admin_function(object):
    '''
    admin管理员功能类
    '''
    def __init__(self):
        pass

    def Admin_function_main(self):
        '''
        调用函数，根据输入指令调用类中的相关功能
        '''
        while True:
            print('''
            -------------Operation menu------------
                      Change password：pas
                      Unlocked：lc
                      Lock the user：ul
                      Add user：in
                      Delete user:rm
                      Exit:exit
            ---------------------------------------
        ''')
            admin_operation = raw_input("Please enter an action:")
            if admin_operation == "pas":
                self.pas_reset()
            elif admin_operation == "lc":
                self.counter_tag = 0
                self.unlock_lock_user()
            elif admin_operation == "ul":
                self.counter_tag = 3
                self.unlock_lock_user()
            elif admin_operation == "in":
                self.user_app_del = "in"
                self.add_del_user()
            elif admin_operation == "rm":
                self.user_app_del = "rm"
                self.add_del_user()
            elif admin_operation == "exit":
                sys.exit()
            else:
                print("Input Error")

    def users(self):
        '''
        判断admin输入用户是否存在
        '''
        while True:
            self.userlist = user_config.sections()
            print("------------Current user-------------")
            for single_user in self.userlist:
                print("               %s"%single_user)
            print("-------------------------------------")
            self.admin_inuser = raw_input("Please enter a user：")
            if self.admin_inuser in self.userlist:
                break
            else:
                print("User does not exist")

    def unlock_lock_user(self):
        '''
        锁定与解锁，主要更改usedb中的counter项，如果用户输入admin则不执行操作
        '''
        self.users()
        if self.admin_inuser == "admin":
            print("This action is not valid for admin")
        else:
            user_config.set(self.admin_inuser,'counter',self.counter_tag)
            user_config.write(open('userdb', "w"))
            if self.counter_tag == 0 :
                print_str = "Unlock"
            else:
                print_str = "locking"
            print("User %s is %s"%(self.admin_inuser,print_str))

    def pas_reset(self):
        '''
        更改密码函数，用于更改用户和admin的密码，密码最小长度不能低于8位，此处可以增加
        复杂度判断，主要更改userdb中的password字段
        '''
        self.users()
        admin_inpas = raw_input("Please enter a reset password (no less than 8 digits)：")
        if len(admin_inpas) > 7:
            print("The password has been reset to:%s"%(admin_inpas))
            user_config.set(self.admin_inuser,'password',admin_inpas)
            user_config.write(open('userdb', "w"))
        else:
            print("Password length is less than 8 digits Please re-enter")

    def add_del_user(self):
        '''
        增删用户操作，增加用户时判断用户是否存在，不存在则添加，
        并输入设定用户密码，同样不能少于八位（可增加密码复杂度判断）对admin无效
        删除用户时，对用户进行判断，存在则删除，对admin无效
        '''
        userlists = user_config.sections()
        print("------------Current user-------------")
        for single_user in userlists:
            print("              %s"%single_user)
        print("-------------------------------------")
        add_name = raw_input("Please enter a user:")
        if add_name == 'admin':
            print("This action is not valid for admin")
        else:
            if self.user_app_del == "in":
                if add_name in userlists:
                    print("User already exists")
                else:
                    while True:
                        add_inpas = raw_input("Please enter a password(no less than 8 digits)：")
                        if len(add_inpas) > 7:
                            user_config.add_section(add_name)
                            user_config.set(add_name,'password',add_inpas)
                            user_config.set(add_name,'counter',"0")
                            user_config.write(open('userdb', "w"))
                            print("The[%s] user creation is complete"%add_name)
                            break
                        elif add_inpas == "exit":
                            break
                        else:
                            print("Password length is less than 8 digits Please re-enter")
            else:
                if add_name in userlists:
                    user_config.remove_section(add_name)
                    user_config.write(open('userdb', "w"))
                    print("[%s]User deleted"%add_name)
                else:
                    print("User does not exist")

class user_function(object):
    '''
    普通用户功能类
    self.user 封装传入的用户
    '''
    def __init__(self,inuser):
        self.user = inuser

    def user_main(self):
        '''
        调用功能的函数
        '''
        while True:
            print('''
            -----------Operation menu-----------
                    Change password：pas
                    Exit：exit
            ------------------------------------
            ''')
            user_main_in = raw_input("Please enter an action:")
            if user_main_in == "pas":
                self.pas_reset()
            elif user_main_in == "exit":
                sys.exit()
            else:
                print("Input Error")

    def pas_reset(self):
        user_inpas = raw_input("Please enter a reset password (no less than 8 digits)：")
        if len(user_inpas) > 7:
            print("The password has been reset to:%s"%user_inpas)
            user_config.set(self.user,'password',user_inpas)
            user_config.write(open('userdb', "w"))
        else:
            print("The password is less than 8 digits long")

def for_count():
    '''
    主调函数，在执行前调用ip_Locking_verify（）函数，用于IP锁定验证
    当用户登录错误次数超过五次后将锁定ip地址，一分钟后开放并获取当前时间的时间戳，
    传入in_lock函数中;当条件正常的时候，传入用户验证函数
    '''
    user_count = 0
    while True:
        lock_times = time.time()
        new_lock_times = str(lock_times).split('.')[0]
        ip_Locking_verify(new_lock_times)
        user_certification()
        user_count += 1
        if user_count == 5:
            ip_lock(new_lock_times)
            print("Maximum login limit, see a video, please log in again after 1 minute")
            time.sleep(1.5)
            webbrowser.open("https://v.qq.com/x/page/i0014jfvo9j.html",new=0,autoraise=True)
            sys.exit()

def user_certification():
    '''
    user certification(用户认证)
    用户验证成功后，进行密码验证，admin用户不计算在次数锁定条件中
    '''
    times = time.strftime("%H:%M:%S",time.localtime(time.time()))
    user_in = raw_input("login user [%s] :"%times)
    user_list = user_config.sections()
    if user_in in user_list:
        if user_in == "admin":
            user_state = "0"
        else:
            user_state = user_config.get(user_in,"counter")
        if int(user_state) < 3:
            password_ertification(user_in)
        else:
            print("User is locked Please contact the administrator to unlock")
    else:
        print("User name wrong")

def password_ertification(user):
    '''
    密码验证，失败则保持用户文件中登录次数增加1当到达三次后则锁定
    密码验证成功后进入用户菜单，并清除登录计数
    linux启用getpass输入密码不显示
    '''
    password_in = raw_input("login password :")
#    password_in = getpass.getpass("login password :")
    user_password = user_config.get(user,"password")
    if password_in == user_password:
        user_config.set(user,'counter',0)
        user_config.write(open('userdb', "w"))
        if user == 'admin':
            admin_class = Admin_function()
            admin_class.Admin_function_main()
        else:
            user_class = user_function(user)
            user_class.user_main()
    else:
        if user == 'admin':
            print("Password mistake")
        else:
            user_counter = user_config.get(user,"counter")
            counter_value = int(user_counter) + 1
            user_config.set(user,'counter',str(counter_value))
            user_config.write(open('userdb', "w"))
            print("Password mistake")

def ip_lock(in_lock_times):
    '''
    IP锁定，触发此函数，将把ip和当前你时间戳记录到iplist文件中
    '''
    with open('iplist','a') as ip_lock_in:
        lock_in_content = "%s:%s
"%(LOCAL_IP,in_lock_times)
        ip_lock_in.write(lock_in_content)
def ip_Locking_verify(time_stamp):
    '''
    验证当前ip是否被锁定，符合则解锁时间则解锁
    通过iplist文件进行判断，程序执行时会读取现在的时间戳
    然后减去记录在文件中的时间戳，如果大60则将条目删除，否则输出解锁时间
    '''
    ip_list = []
    with open('iplist','rU')as main_iplist :
        for lock_ip in main_iplist.readlines():
            ip_list.append(lock_ip)
            if LOCAL_IP in lock_ip:
                new_ip_Locking = str(lock_ip).split(':')[1]
                if int(time_stamp) - int(new_ip_Locking) > 60:
                    ip_list.remove(lock_ip)
                    print("The current IP[%s] is unlocked"%LOCAL_IP)
                else:
                    remaining_time = 60 - (int(time_stamp) - int(new_ip_Locking))
                    print("The current IP[%s] is locked，Remaining unlock time [%s] seconds"%(LOCAL_IP,remaining_time))
                    sys.exit()
        open_iplist = open("iplist", "w+")
        if ip_list:
            open_iplist.writelines(ip_list)
        else:
            open_iplist.writelines("")
        open_iplist.close()

if __name__ == '__main__':
    for_count()

user存储用户数据的格式

[admin]
password = admin.123
counter = 0

[test]
password = admin.123
counter = 0

[test1]
password = admin.123
counter = 3

创建一个叫iplist的文件，到此即可正常运行