python接口自动化学习五之函数案例实战

#! /usr/bin/env python
# -*-coding:utf-8 -*-

#Author:liulingling

'''
要求：实现注册功能，登陆成功后 返回昵称
'''

def inOut():
	username = input("请输入账号")
	password = input("请输入密码")
	return username,password

def register():
	'''注册函数'''
	username,password=inOut()
	# username=input("请输入账号")
	# password=input("请输入密码")
	per=username+'|'+password
	with open('login.md','w') as f:
		f.write(per)

def login():
	'''登陆函数'''
	# username = input("请输入账号")
	# password = input("请输入密码")
	username,password=inOut()
	with open('login.md','r') as f:
		info=f.read()
	info=info.split('|')
	# print(info)
	if username==info[0] and password==info[1]:
		return True
	else:
		return False

def getNick(func):
	'''获取昵称'''
	with open('login.md','r') as f:
		info=f.read()
	info=info.split('|')
	if func:
		print('{0}您好，欢迎您访问无涯课堂'.format(info[0]))
	else:
		print('请登陆系统')

def main():
	while True:
		try:
			t=int(input('1，注册 2，登陆 3，退出系统'))
		except Exception as e:
			print(e.args)
		else:
			if t==1:
				register()
			elif t==2:
				getNick(login())
			elif t==3:
				import sys
				sys.exit(1)
			else:
				print('输入错误，请继续')
		finally:pass


main()