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',"")
                             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 = ""
         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\n"%(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的文件，到此即可正常运行