登陆注册优化版

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:Er fei


def lan_ding(username,password):#登陆函数
    '''
    这是一个登陆函数
    :param username:用户名
    :param password: 密码
    :return:登陆成功与否
    '''
    with open("456","r")as user :
        for i in user :
            k=i.split("|")
            if k[0]==username and k[1]==password:
                return True
            return False
def registered(username,password):
    """
    这是一个注册函数。。。
    :param username: 用户名
    :param password: 密码
    :return: 成功与否
    """
    with open("456","r")as user ,open("456","a")as us :
        for i in user:
            k=i.split("|")
            if k[0]!=username:
                us.write("
"+username+"|"+password)
                return True
            else:
                print("用户名已存在")
                return False
def main():
    '''
    这是一个主函数
    :return:
    '''
    user_input=input("如果注册输入1：如果登陆输入2：")
    if user_input=="1":
        username=input("请输入用户名：")
        password=input("  请输入密码：")
        s=lan_ding(username,password)
        if s :
            print("登陆成功")
        else:
            print("用户名或密码错误")
    elif user_input=="2":
        username=input("请输入用户名：")
        password=input("  请输入密码：")
        k=registered(username,password)
        if k :
            print("注册成功")
        else:
            print("注册失败")

main()