python的登录验证练习

#!/usr/bin/python
#coding=utf-8
#Author=wyl

def register():
uname = input("请注册用户：")
passwd = input("请输入密码：")
f = open("db",'a')
temp = "\n"+uname+","+passwd
f.write(temp)
f.close()
print("注册成功")

def login(name,pwd):
f = open("db", "r")
for line in f:
list = line.split(",")
username = list[0]
password = list[1]
if name == username and pwd == password:
return True
return False

def main():
t = input("1.登录, 2.注册")
if t == "1":
name = input("请输入用户名：")
pwd = input("请输入密码")
t = login(name,pwd)
if t:
print("登录成功")
else:
print("登录失败")

elif t == '2':
register()

if __name__ == '__main__':
main()