用函数写注册功能

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

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

 1 def get_userinfo():
 2     '''
 3     读取用户信息
 4     :return:
 5     '''
 6     user_info_dict={}
 7     with open('db','r',encoding='utf-8') as f:
 8         for line in f:
 9             line = line.strip().split(',')
10             user_info_dict[line[0]] ={
11                 'password':line[1],
12                 'balance':line[2],
13             }
14         return user_info_dict
15 def get_name():
16     '''
17     输入用户名
18     :return:
19     '''
20     while True:
21         name = input('用户名>>:').strip()
22         if name.isalpha():
23             if name not in get_userinfo():
24                 return name
25             else:
26                 print('用户名已存在！')
27         else:
28             print('用户名输入不合法！')
29 def get_pwd():
30     '''
31     输入密码
32     :return:
33     '''
34     while True:
35         pwd1 = input('请输入密码>>:').strip()
36         pwd2 = input('请再次输入密码>>:').strip()
37         if pwd1 == pwd2:
38             return pwd1
39         else:
40             print('两次输入密码不一致，请重新输入！')
41 def get_bal():
42     '''
43     输入金额
44     :return:
45     '''
46     bal = input('请输入金额>>:').strip()
47     if bal.isdigit():
48         return bal
49     else:
50         print('金额必须是数字')
51 def register():
52     '''
53     注册功能
54     :return:
55     '''
56     get_userinfo()
57     username = get_name()
58     password = get_pwd()
59     salary = get_bal()
60     user_info = [username,password,salary]
61     with open('db','a',encoding='utf-8')as f:
62         f.write(','.join(user_info)+'
')
63         print('用户注册成功！！！')
64 
65 register()

View Code

下面是文本db

1 alex,abc,1500
2 egon,asd,5200
3 jack,123,2300