模拟登陆

练习题2 —— 模拟登陆：
用户输入帐号密码进行登陆
用户信息加注册时间保存在文件内
用户密码输入错误三次后锁定用户，下次再登录，检测到是这个用户也登录不了
储存账号的信息文件，涉及到文件名、和格式：
db-login.txt 

dadada1|dadada1|2019-06-10 15:50:50
dadada2|dadada2|2019-06-10 16:02:03
dadada3|dadada3|2019-06-10 16:02:14
dadada|dadada|2019-06-11 18:29:03
dadada4|dadada4|2019-06-11 18:30:32
dadada5|dadada5|2019-06-11 18:34:43
dadada6|dadada6|2019-06-11 18:36:08
dadada100|dadada100|2019-06-11 18:43:40
dadada7|dadada7|2019-06-11 18:48:02
dadada8|dadada8|2019-06-11 19:16:50

db-ban.txt

dadada3|3 

代码：

# @Time     :2019/6/7 23:01
# -*- encoding:utf-8 -*-

from _datetime import datetime

def register():
    print("register".center(50,'*'))
    user_list = []
    # 把db信息读取到列表中,检测注册的账号是否已经存在
    with open('db-login.txt', mode='r+', encoding='utf-8') as f:
        for line in f:
            # print(line,type(line)) #asdd|asdd|2019-06-10 14:06:41
            user_list.append(line.split('|')[0])  # 存放db中的账号，检测注册账号是否重复

        username = input("username : ")
        pwd = input("pwd:")
        ctime = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
        if username in user_list:
            print("账号已存在，请重新注册")
            return register() #调用该函数，重新注册  ***出了问题，如何调用自己继续注册？
        else:
            line = '%s|%s|%s
' % (username, pwd, ctime)
            f.seek(0, 2)
            f.write(line)
            return True



def login():
    sign = True #登录成功时，sign为False，否则将一直登录
    count = 0  # 计次，控制账号是否锁定
    while sign:
        db_login = {}#允许登录的信息
        db_ban = {} #被锁定账号的信息
        #把db-login.txt中的信息 存到 db_login字典中
        with open('db-login.txt',mode='r',encoding='utf-8') as f :
            for line in f :
                line = line.split('|')
                k,v =line[0], line[1]
                db_login.setdefault(k,v)

        #把db_ban.txt中的信息 存到 db_ban字典中
        with open('db_ban.txt',mode='r',encoding='utf-8') as f :
            for line in f:
                line = line.split('|')
                db_ban.setdefault(line[0],line[1])

        print("login".center(50,'*'))
        username = input("username : ")
        pwd = input("pwd:")

        '''
        输入的username有以下几种情况：
            1.账号已锁定，无法登录
            2.count 等于 3  账号讲被锁定
            3.输入的账号不存在db_login字典中
            4.输入的账号存在db_login字典中：
                1）.账号密码正确 ==》登陆成功
                1）.账号密码不正确 ==》登陆失败 count+=1
        '''
        if db_ban.__contains__(username) :
            print("账号已锁定，无法登录")

        elif count == 2:
            print("错了3次，账号：%s 已经被锁定"%username)
            db_ban.setdefault(username,pwd)
            for k ,v in db_ban.items():
                with open('db_ban.txt',mode='r+',encoding='utf-8') as f:
                    line = "%s|%s "%(k,v)
                    f.seek(0,2)
                    f.write(line)

        elif not db_login.__contains__(username):
            print("账号不存在")

        else:
            for k ,v in db_login.items():
                if username == k and pwd == v :

                    sign = False


            if not sign :
                print("登录成功")

            else:
                print("账号密码不匹配")
                count += 1
                print("三次错误登录，账号讲锁定")
                print("这是第%d登录" % count)


def run():
    while True:
        print("注册输1
登陆输2
退出输q")
        choice = input(">>> ")
        if choice == '1':
            if register() :
                print("注册成功")
            else:
                print(" 注册失败")
        elif choice == '2':
            login()
        elif choice == 'q':
            exit()
        else:
            print("Wrong inputing")

if __name__ == '__main__':
    run()