python 实现注册、登陆功能

class Student:
	def register(self):
		print("=============注册页面=============")
		n = 3
		user = input("请输入帐号：").strip()
		while n > 0:
			n = n -1
			pass_wd = input("请输入密码:").strip()
			pass_wd_sec = input("确认你的密码:").strip()
			if pass_wd != pass_wd_sec:
				print("两次输入不一致，请从新输入！\n")
				print("你还有" + str(n) + "次机会!")
				continue
			else:
				print("注册成功！\n")
				user_info[user] = pass_wd_sec
				break

	def login(self):
		n = 3
		while n > 0:
			n = n - 1
			pass_wd = input("请输入你的密码:").strip()
			if pass_wd != user_info[user]:
				print("密码错误，请从新输入！\n")
				print("你还有" + str(n) + "次机会！")
				continue
			else:
				print("登陆成功!\n")
				break

	def achievement(self):
		# 这里实现成绩录入与成绩查询
		print("请录入你的成绩！")
		while True:
			subject_name = input("课程名称：")
			score = int(input("课程分数"))
			student_score[subject_name] = score
			status = input("\n选择你的操做  【0:继续，1:查看，其余:退出】")
			if status == '0':
				continue
			elif status == '1':
				choice = input("请输入你要查看的课程名称：【1:所有，其余:当前】")
				if choice == '1':
					print(student_score)
				else:
					print(student_score[choice])
			else:
				break

	def course_selection(self):
		all_course = ['java', 'hadoop']
		my_course = []
		while True:
			print("\n你当前的课程有：" + str(my_course))
			print("\n当前可选的课程有：" + str(all_course))
			course_name = input("请输入你要选择的课程:")
			my_course.append(course_name)
			print("选课结果为：" + str(my_course))
			break


if __name__ == '__main__':
	user_info = {"hadoop": "111", "hive": "222", "spark": "333"}
	student_score = {"语文": 86, "数学": 96, "英语": 81, "化学": 91, "物理": 92}
	user = input("请输入帐号：").strip()
	s = Student()
	if user not in user_info.keys():
		print("帐号不存在，请返回注册\n")
		s.register()
		print("=============登陆界面=============")
		user = input("请输入帐号：").strip()
		s.login()
		s.achievement()
	else:
		s.login()
		s.achievement()
		s.course_selection()