Python自动化——Day1——登录接口

#! /usr/bin/env python
#coding=utf-8

#登录接口
#
#输入用户名密码--用户名密码存在本地txt文件里
#验证输入的用户名密码，都对才能欢迎语
#超过三次输入错误，锁定账号，创建lock.txt,加入该账号
#若账号已锁定，则返回账号已锁定

import sys
import getpass
import os

i = 0
while i < 3:
    user_input = raw_input("Please input your username:")

    lock_file = open("account_lock.txt","r+")
    lock_list = lock_file.readlines()

    for lock_line in lock_list:
        lock_line = lock_line.strip('\n')
        if user_input == lock_line:
            sys.exit("Your account %s is locked! Exit!" % user_input)

    acct_file = open("account.txt","r+")
    acct_list = acct_file.readlines()
    
    for acct_line in acct_list:
        (user,password) = acct_line.strip('\n').split(',')
        if user_input == user:
            j = 0
            while j < 3:
                pwd_input = getpass.getpass("Please input your password:")
                if pwd_input == password:
                    print("Welcome to login platform, user %s" % user_input)
                    sys.exit()
                else:
                    print("User %s password error, please re-input, only %s times left" %(user_input,2-j))
                j += 1
            else:
                lock_file.write(user_input + '\n')
                sys.exit("Login failed, account %s locked!" %user_input)
        else:
            pass
    else:
        if i < 3:
            print ("User %s not exists,please re-input, only %s times left!" %(user_input,2-i))   
    i += 1
else:
    sys.exit("Login failed, user not exists!")

lock_file.close()
acct_file.close()