用函数写注册功能

昨天看了函数的视屏并完成了作业，与大家分享一下。只想为自己记录学习的过程，也希望大家指出我的不足和错误，你们的意见是我学习动力！谢谢！

作业： 用函数写用户注册功能

def get_userinfo():
    '''
    读取用户信息
    :return:
    '''
    user_info_dict={}
    with open('db','r',encoding='utf-8') as f:
        for line in f:
            line = line.strip().split(',')
            user_info_dict[line[0]] ={
                'password':line[1],
                'balance':line[2],
            }
        return user_info_dict
def get_name():
    '''
    输入用户名
    :return:
    '''
    while True:
        name = input('用户名>>:').strip()
        if name.isalpha():
            if name not in get_userinfo():
                return name
            else:
                print('用户名已存在！')
        else:
            print('用户名输入不合法！')
def get_pwd():
    '''
    输入密码
    :return:
    '''
    while True:
        pwd1 = input('请输入密码>>:').strip()
        pwd2 = input('请再次输入密码>>:').strip()
        if pwd1 == pwd2:
            return pwd1
        else:
            print('两次输入密码不一致，请重新输入！')
def get_bal():
    '''
    输入金额
    :return:
    '''
    bal = input('请输入金额>>:').strip()
    if bal.isdigit():
        return bal
    else:
        print('金额必须是数字')
def register():
    '''
    注册功能
    :return:
    '''
    get_userinfo()
    username = get_name()
    password = get_pwd()
    salary = get_bal()
    user_info = [username,password,salary]
    with open('db','a',encoding='utf-8')as f:
        f.write(','.join(user_info)+'
')
        print('用户注册成功！！！')

register()

下面是文本db

alex,abc,1500
egon,asd,5200
jack,123,2300