一、 with as 功能介绍
在 Python 中,with as 语句可以让我们更方便地使用处理资源的对象,比如文件、网络连接等,同时还能在使用完毕后自动关闭这些资源。
with open('example.txt', 'r') as f:
print(f.readlines())
上述代码使用 with as 语句打开 example.txt 文件,并自动关闭文件对象 f。
其中,open() 函数返回的是一个文件对象,我们可以使用 as 关键字给这个对象起一个方便识别的名称,这个名称会作为 with as 语句块中变量的名称。
二、 with as 的基本语法
Python with as 语句的基本语法如下:
with context_expression [as target(s)]:
with-body
其中,context_expression 表示要处理的资源对象,as target(s) 表示我们可以给这个对象起一个别名(or 名称),with-body 表示要执行的操作。
三、简单例子
下面看一个简单的例子,使用 with as 语句实现文件读写操作:
with open('example.txt', 'r') as f:
print(f.read())
with open('example.txt', 'a') as f:
f.write('这是新的一行。\n')
上述代码中,第一个 with as 语句用于读取 example.txt 的内容并输出到屏幕上,第二个 with as 语句用于向 example.txt 文件中追加内容。
四、同时打开多个文件
with as 语句还可以同时打开多个文件,并在处理完毕后自动关闭它们。下面的代码演示了同时打开两个文件,并将它们的内容合并到一个新文件中:
with open('example1.txt', 'r') as f1, open('example2.txt', 'r') as f2:
content = f1.read() + f2.read()
with open('example_merge.txt', 'w') as f:
f.write(content)
上述代码中,我们使用逗号分隔符同时打开 example1.txt 和 example2.txt 文件对象,并读取它们的内容。然后将这些内容合并到 content 变量中,并将 content 内容写入 example_merge.txt 文件中。
五、自定义上下文管理器
除了文件读写操作外,我们还可以针对其他资源编写自定义上下文管理器。下面使用 socket 为例,演示如何创建一个自定义的上下文管理器:
import socket
class MySocket():
def __init__(self, host, port):
self.host = host
self.port = port
def __enter__(self):
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.connect((self.host, self.port))
return self.sock
def __exit__(self, exc_type, exc_val, exc_tb):
self.sock.close()
with MySocket('www.baidu.com', 80) as sock:
sock.sendall(b'HEAD / HTTP/1.1\r\nHost: www.baidu.com\r\n\r\n')
data = sock.recv(1024)
print(data.decode())
上述代码中,我们使用自定义的 MySocket 类来管理 socket 对象,在 with 语句块中打开连接,完成请求后自动关闭连接。
在自定义的上下文管理器类中,我们需要包含 __enter__() 和 __exit__() 两个魔法方法。__enter__() 用于执行资源对象的初始化操作,返回该资源对象。__exit__() 用于在 with 语句块执行完成后关闭资源对象。
六、上下文环境的嵌套使用
上下文管理器支持嵌套使用,即在一个 with 语句块中可以使用多个 with as 语句。下面是一个嵌套使用的例子:
with open('example1.txt', 'r') as f1:
with open('example2.txt', 'r') as f2:
with open('example3.txt', 'w') as f3:
content = f1.read() + f2.read()
f3.write(content)
上述代码中,我们在一个 with 语句块中打开三个文件,同时使用了多个 with as 语句。即在第三个 with 语句块中嵌套使用了第一个和第二个 with as 语句块。这种嵌套使用可以让我们更方便地管理多个资源对象。