#!/usr/bin/env python

 # -*- coding: utf-8 -*-

 # author : Wang Yue

 import sys,hashlib,getpass,time

 #定义用户类

 class real_user(object):

     real_name=''#用户名

     real_pwd=''#密码

     login_count=0 #登录次数

     lock_time=0 #锁定时间

     #重写构造函数，用于初始化用户类实例

     def __init__(self,real_name,real_pwd):

         self.real_name=real_name

         self.real_pwd=real_pwd

     #属性设定方法

     def set_lock(self,lt):

         self.lock_time=lt

     def get_login_count(self):

         return self.login_count

     #定义业务方法，用于锁定自身用户实例的方法

     def lock_me(self,seconds):

         self.lock_time=seconds

         lock_info=self.real_name + "," + str(self.lock_time) + "\n"

         return lock_info

     #定义业务方法，用于解锁自身用户实例

     def unlock_me(self,u_name):#lock_time float

         if u_name == self.real_name and time.time() >= self.lock_time:

             self.lock_time = 0

             return True

         else:

             return False

 #单独封装的函数，主函数中会使用到的方法，公共性的，对于主函数透明

 #从文件（数据源）中，将已经存在的用户信息，写为字符串数组，返回该数组

 def get_file_line(file,mode_file):

     userlist = []

     with open(file,mode=mode_file,encoding='utf-8') as filedb:

         filedb.flush()

         filedb.seek(0)

         fileline=filedb.readline()

         while fileline:

            userlist.append(fileline)

            fileline=filedb.readline()

     return userlist

 #将信息回写到文件，主函数中用于写黑名单

 def input_line_file(file,info):#info is a list

     with open(file,mode="a+",encoding="utf-8") as filedb:

         for info_line in info:

             filedb.write(info_line + "\n")

 #将信息从文件中删除，含有信息的行会被删除

 def delete_line_file(file,info):

     with open(file,mode="w+",encoding="utf-8") as filedb:

         filedb.flush()

         filedb.seek(0)

         for line_one in filedb.readlines():

             if info in line_one:

                 line_info=line_one.split()

                 if line_info[0] == info:

                     continue

             filedb.write(line_one)

 #校验登录，参数：登录的用户名，密码，真是存在的用户实例list

 #用户名密码，检查锁定状态

 #返回值：登录成功，当前锁定。（用户名压根不存在的话，在主函数就过滤掉了）

 def verify_me(u_name,u_pwd,user):

     if user.lock_time != 0:

         return "fuck you,fuck off"

     if time.time()>=user.lock_time:

         user.lock_time=0

     if user.real_name == u_name:

         user.login_count +=1

     if user.real_name == u_name and user.real_pwd == u_pwd:

         user.login_count = 0

         user.lock_time = 0

         return "sucess"

     else:

         return "retry once....."

 #获得MD5，密码在文件中存放时是MD5加密的

 def get_md5_enc(need_md5_char):

     utf_kkk = need_md5_char.encode(encoding="utf-8")

     m = hashlib.md5()

     m.update(utf_kkk)

     md5_char = m.hexdigest()

     return md5_char

 #主函数

 if __name__ == '__main__':

     userlist=get_file_line("user_db.doo","r") #获取用户信息，存成信息list

     locklist=get_file_line("lock_db.doo","r") #获取黑名单信息，存成信息list

     user=[]  #定义用户实例数组

     #初始化全部用户实例，使用用户类的构造函数

     for user_line in userlist:

         name_pwd=user_line.split("::++++::")

         name=name_pwd[0].strip()

         pwd=name_pwd[1].strip()

         user.append(real_user(name,pwd))

     #遍历用户实例数组，将锁定属性赋值

     for lock_inf in locklist:

         lock_info=lock_inf.strip()

         if lock_info =='':

             continue

         lockif=lock_info.split(',')

         for user_ok in user:

             if user_ok.real_name==lockif[0]:

                 user_ok.set_lock(float(lockif[1].strip()))

     #以下开始正式业务

     count = 0 #用于记录登录次数

     while True:

         #输入用户名密码，并将密码MD5

         print("hello let try login")

         username=input("username:")

         passinput=input("password:")

         passwd = get_md5_enc(passinput)

         #从用户类实例list中，确定用户行为：用户名是否正确？密码是否正确？

         for user_one in user:

             #用户名不正确，3次，提示需要注册，程序结束

             if user_one.real_name !=username:

                 if count >2:

                     print("you need reg.....")

                     sys.exit(0)

                 continue

             #用户名正确，密码情况：

             check_login=verify_me(username,passwd,user_one)

             #密码正确，登录成功

             if check_login =="sucess":

                 print("hello:{_name},login success!!!!".format(_name=username))

                 delete_line_file("lock_db.doo",username)

                 sys.exit(0)

             #密码不正确，该用户实例的登录次数+1，避免仅提供三次输入用户名密码就结束的情况，不然中途换了用户名，就出bug了

             #用户名正确时，用户登录次数超过3次，将被锁定5分钟。

             elif user_one.get_login_count() > 2 and user_one.real_name == username:

                 lock_time=time.time()+300

                 lock_info=user_one.lock_me(lock_time)

                 lockjj=[]

                 lockjj.append(lock_info)

                 input_line_file("lock_db.doo",lockjj)

                 print("fuck you,lock you 5m!!!!")

                 sys.exit(0)

             #登录状态为锁定的话，就结束程序

             elif check_login =="fuck you,fuck off":

                 print("fuck you,fuck off:{_name}.you were locked".format(_name=username))

                 sys.exit(0)

             #解锁用户：锁定时间算法为（锁定发生时，增加300秒），当前时间>=锁定的时间，即可解锁

             elif user_one.lock_time <= time.time() and user_one.real_name==username:

                 user_one.unlock_me(username)

                 delete_line_file("lock_db.doo",username)

                 continue

             #判定用户名不正确的使用登录次数，超过3次就结束

             elif count <2:

                 count +=1

                 break

         count +=1

#!/usr/bin/env python

# -*- coding: utf-8 -*-

# author : Wang Yue

import hashlib,getpass,sys

#MD5加密算法

def get_md5_enc(need_md5_char):

    utf_kkk = need_md5_char.encode(encoding="utf-8")

    m = hashlib.md5()

    m.update(utf_kkk)

    md5_char = m.hexdigest()

    return md5_char

print("input username,and passwd")

#以两种方式打开文件的流，增加模式，只读模式

user_db=open("user_db.doo",mode="a+",encoding="utf-8")

check_db=open("user_db.doo",mode="r",encoding="utf-8")

#主业务程序：

while True:

    #获取输入的用户名，密码

    username = input("username:")

    passwd = getpass.getpass("password:")

    passwd2 = getpass.getpass("password confirmation:")

    #判断确认密码是否一致。

    if passwd == passwd2:

        cc_status=0

        check_line=check_db.readline()

        while check_line: #在密码文件中，检查是否已经存在了需要添加的用户名

            cl=check_line.split("::++++::")

            if cl[0] == username:

                print("repetition username !!! add user faild,username:{_username}".format(_username=username))

                cc_status=1

            check_line=check_db.readline()

        if cc_status == 1: #确定是否要退出程序

            quit_con = input("quit me (y/n):::")

            if quit_con == "y":

                user_db.close()

                check_db.close()

                sys.exit()

            else:

                continue

        #如果能走到这一步，说明，就是要添加一个用户了

        en_pass = get_md5_enc(passwd) #MD5密码。

        var_char = username + "::++++::" + en_pass + "\n" #设定好约定的用户名密码之间的分隔，及回行符

        user_db.write(var_char) #写入用户数据文件

        print("succed username:{_username}".format(_username=username))

    else:

        print("The input is inconsistent(password)")

    quit_con = input("quit me (y/n):::")

    if quit_con == "y":

        user_db.close()

        check_db.close()

        sys.exit(0)