Python阶段复习 - part 4 - 用户登录程序

#!/usr/bin/env python
# _*_ coding:UTF-8 _*_
# __auth__:Dahlhin

import sys


userinfo = r'userinfo.txt'
userlock = r'userlock.txt'


def user_exist_check(user):
    '''检查用户是否存在'''
    with open(userinfo) as fd:
        for info in fd:
            if user == info.strip().split()[0]:
                return True
        return False

def user_lock_check(user):
    '''检查用户是都被锁定'''
    with open(userlock) as fd:
        for lock in fd:
            if user == lock.strip():
                return True
        return False

def passwd_check(passwd):
    '''检查密码合法性'''
    with open(userinfo) as fd:
        for info in fd:
            if passwd == info.strip().split()[1]:
                return True
            return False

def lock_user(user):
    '''用户锁定'''
    with open(userlock,'a') as fd:
        fd.write('%s\n' % user)


def user_login():
    '''登陆主程序'''
    count = 0
    while count < 3:
        user = input('UserName:')
        passwd = input('PassWord:')
        if user_exist_check(user):
            if not user_lock_check(user):
                if passwd_check(passwd):
                    return True
                else:
                    print('密码错误')
            else:
                print('账号被锁定')
        else:
            print('账号不存在')
        count += 1

    else:
        if user_exist_check(user):
            if not user_lock_check(user):
                lock_user(user)
        return  False


if __name__ == '__main__':
    result = user_login()
    if result:
        print('Welcome login')
    else:
        print('bye bye')