本文实例为大家分享了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('|') # 去行(\n)拆分
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(存储登录信息)的文件,内容格式如下:
效果:
登录
注册