【python之路21】用户登陆程序函数

一、用户登陆函数实例

1、注意：以后写函数时必须在函数第一行后按回车加入“”““”“””两对三引号后回车，此时会自动列出参数值，注释函数的作用、参数的用法和返回值

#!usr/bin/env python
# -*- coding:utf-8 -*-
def LogIn(username,password):
    """
    读取login文件的用户名和密码进行校验
    :param username: 传入用户名
    :param password: 传入密码
    :return: 如果用户名和密码匹配则返回True，否则返回False
    """
    with open('login','r',encoding='utf-8') as f:
        for line in f:
            line = line.strip()
            login = line.split('$')
            if username == login[0] and password == login[1]:
                return True
        return False


bol = LogIn('tqg','111111')
if bol:
    print('登陆成功')
else:
    print('登陆失败')

 2）用户登陆及注册功能

#!usr/bin/env python
# -*- coding:utf-8 -*-
def LogIn(username,password):
    """
    读取login文件的用户名和密码进行校验
    :param username: 传入用户名
    :param password: 传入密码
    :return: 如果用户名和密码匹配则返回True，否则返回False
    """
    with open('login','r',encoding='utf-8') as f:
        for line in f:
            line = line.strip()
            login = line.split('$')
            if username == login[0] and password == login[1]:
                return True
def register(username,password):
    """
    用户注册实现
    :param username: 用户名
    :param password: 密码
    :return: None，把用户名和密码按照  用户名$密码  格式写入login文件
    """
    with open('login','a',encoding='utf-8') as f:
        emp = '
' + username + '$' + password
        f.write(emp)
def user_exist(uname):
    """
    判断文件中是否存在用户名
    :param uname: 用户名
    :return: 如果存在则返回True,不存在则返回False
    """
    with open('login','r',encoding='utf-8') as f:
        for line in f:
            line = line.strip()
            line_list = line.split('$')
            if line_list[0] == uname:
                return True
    return False
def main():
    emp = input("登陆请输入1；注册请输入2：")
    uname = input("请输入用户名：")
    pwd = input("请输入密码：")
    if emp == "1":
        if LogIn(uname,pwd):
            print('登陆成功！')
        else:
            print('登陆失败！')
    elif emp == "2":
        if user_exist(uname):
            print('用户名已存在，不允许注册。')
        else:
            register(uname,pwd)
            print("注册成功！")
    else:
        print('您输入的命令不正确！')

main()