Python基础之函数定义及文件修改

函数

函数的定义

#登录函数和注册函数

def register():
	"""注册函数"""
	username = input('请输入你的用户名：')
	password = input('请输入你的密码：')

	#按照“用户名：密码|”的格式存储
	with open('test.txt', 'w', encoding='utf8') as fw:
		fw.write(f'{username}:{password}|')

def login():
	"""登录函数"""
	username = input('请输入你的用户名：')
	password = input('请输入你的密码：')

	#从文件中读数据进行匹配
	with open('test.txt', 'r', encoding='utf8') as fr:
		data = fr.read()
		info_list = data.split('|')
		info_user = f'{username}:{password}'
		if info_user in info_list:
			print('登录成功')
		else:
			print('登陆失败')

#直接用以下语句进行调用
#register()
#login()

#注释写在函数名的下一行未来忘记函数功能时，可用print(login.__doc__)进行查询

三种定义方式及其调用

#无参函数
def add():
	"""无参函数"""
	y = input('num1:')
	x = input('num2:')

	return int(x) + int(y)
res = add(10, 30)
print(res * 12)

#有参函数
def add(x, y)
	"""有参函数"""
	return int(x)+int(y)

print(add(10, 20))

#空函数
def func():
	pass

##函数返回值
def add(x, y)
	return x, y, x + y #return可以返回任意数据类型，不加括号默认返回元组类型

x,y,z = add(1, 2) # 可以用解压缩
print(x, y, z)

文件修改

思路：对一个文件中的数据进行修改，通常是先对该文件进行读操作，然后修改文件内容，把修改的内容写入新的文件中，在把旧文件remove掉并重命名新文件。

	import os #导入os库，用于对文件进行修改和删除操作
	with open('test.py', 'r', encoding='utf8') as fr, \
			open('test_swap.py', 'w', encoding='utf8') as fw:
		for i in fr:
			data = i.replace('123', '456) #将文件中的123修改为456
			fw.write(data)
	os.remove('test.py') #移除原文件
	os.rename('test_swap.py', 'test.py') #重命名新文件