Python—函数式编程实现登录注册
代码:
def login(username, password):
"""
用于用户登录
:param username:用户输入的用户名
:param password: 用户输入的密码
:return: true表示登录成功,false登录失败
"""
f = open('db', 'r')
for line in f:
line_list = line.strip().split('|')
if line_list[0] == username and line_list[1] == password:
return True
return False
def register(username, password):
"""
用户注册功能
:param username: 用户名
:param password: 密码
:return: 默认返回null
"""
f = open('db', 'a')
temp = '\n' + username + '|' + password
f.write(temp)
f.close()
def main():
t = input('1:登录 2:注册')
if t == '1':
user = input('请输入用户名:')
pwd = input('请输入密码:')
r = login(user, pwd)
if r:
print('登录成功')
else:
print('登录失败')
elif t == '2':
user = input('请输入用户名:')
pwd = input('请输入密码:')
print('注册成功。')
register(user, pwd)
main()
创建一个db.txt(存储登录信息)的文件,内容格式如下:
效果:
登录
注册