目录

一、概述

1、定义

2、作用

二、主要应用场景

1、构造和析构

2、操作符重载

3、字符串和表示

4、容器管理

5、可调用对象

6、上下文管理

7、属性访问和描述符

8、迭代器和生成器

9、数值类型

10、复制和序列化

11、自定义元类行为

12、自定义类行为

13、类型检查和转换

14、自定义异常

三、学习方法

1、理解基础

2、查阅文档

3、编写示例

4、实践应用

5、阅读他人代码

6、参加社区讨论

7、持续学习

8、练习与总结

9、注意兼容性

10、避免过度使用

四、魔法方法

14、__enter__方法

14-1、语法

14-2、参数

14-3、功能

14-4、返回值

14-5、说明

14-6、用法

15、__eq__方法

15-1、语法

15-2、参数

15-3、功能

15-4、返回值

15-5、说明

15-6、用法

16、__exit__方法

16-1、语法

16-2、参数

16-3、功能

16-4、返回值

16-5、说明

16-6、用法

五、推荐阅读

1、Python筑基之旅

2、Python函数之旅

3、Python算法之旅

4、博客个人主页

一、概述

1、定义

        魔法方法(Magic Methods/Special Methods，也称特殊方法或双下划线方法)是Python中一类具有特殊命名规则的方法，它们的名称通常以双下划线(`__`)开头和结尾。

        魔法方法用于在特定情况下自动被Python解释器调用，而不需要显式地调用它们，它们提供了一种机制，让你可以定义自定义类时具有与内置类型相似的行为。

2、作用

        魔法方法允许开发者重载Python中的一些内置操作或函数的行为，从而为自定义的类添加特殊的功能。

二、主要应用场景

1、构造和析构

1-1、__init__(self, [args...])：在创建对象时初始化属性。

1-2、__new__(cls, [args...])：在创建对象时控制实例的创建过程(通常与元类一起使用)。

1-3、__del__(self)：在对象被销毁前执行清理操作，如关闭文件或释放资源。

2、操作符重载

2-1、__add__(self, other)、__sub__(self, other)、__mul__(self, other)等：自定义对象之间的算术运算。

2-2、__eq__(self, other)、__ne__(self, other)、__lt__(self, other)等：定义对象之间的比较操作。

3、字符串和表示

3-1、__str__(self)：定义对象的字符串表示，常用于print()函数。

3-2、__repr__(self)：定义对象的官方字符串表示，用于repr()函数和交互式解释器。

4、容器管理

4-1、__getitem__(self, key)、__setitem__(self, key, value)、__delitem__(self, key)：用于实现类似列表或字典的索引访问、设置和删除操作。

4-2、__len__(self)：返回对象的长度或元素个数。

5、可调用对象

5-1、__call__(self, [args...])：允许对象像函数一样被调用。

6、上下文管理

6-1、__enter__(self)、__exit__(self, exc_type, exc_val, exc_tb)：用于实现上下文管理器，如with语句中的对象。

7、属性访问和描述符

7-1、__getattr__, __setattr__, __delattr__：这些方法允许对象在访问或修改不存在的属性时执行自定义操作。

7-2、描述符(Descriptors)是实现了__get__, __set__, 和__delete__方法的对象，它们可以控制对另一个对象属性的访问。

8、迭代器和生成器

8-1、__iter__和__next__：这些方法允许对象支持迭代操作，如使用for循环遍历对象。

8-2、__aiter__, __anext__：这些是异步迭代器的魔法方法，用于支持异步迭代。

9、数值类型

9-1、__int__(self)、__float__(self)、__complex__(self)：定义对象到数值类型的转换。

9-2、__index__(self)：定义对象用于切片时的整数转换。

10、复制和序列化

10-1、__copy__和__deepcopy__：允许对象支持浅复制和深复制操作。

10-2、__getstate__和__setstate__：用于自定义对象的序列化和反序列化过程。

11、自定义元类行为

11-1、__metaclass__(Python 2)或元类本身(Python 3)：允许自定义类的创建过程，如动态创建类、修改类的定义等。

12、自定义类行为

12-1、__init__和__new__：用于初始化对象或控制对象的创建过程。

12-2、__init_subclass__：在子类被创建时调用，允许在子类中执行一些额外的操作。

13、类型检查和转换

13-1、__instancecheck__和__subclasscheck__：用于自定义isinstance()和issubclass()函数的行为。

14、自定义异常

14-1、你可以通过继承内置的Exception类来创建自定义的异常类，并定义其特定的行为。

三、学习方法

        要学好Python的魔法方法，你可以遵循以下方法及步骤：

1、理解基础

        首先确保你对Python的基本语法、数据类型、类和对象等概念有深入的理解，这些是理解魔法方法的基础。

2、查阅文档

        仔细阅读Python官方文档中关于魔法方法的部分，文档会详细解释每个魔法方法的作用、参数和返回值。你可以通过访问Python的官方网站或使用help()函数在Python解释器中查看文档。

3、编写示例

        为每个魔法方法编写简单的示例代码，以便更好地理解其用法和效果，通过实际编写和运行代码，你可以更直观地感受到魔法方法如何改变对象的行为。

4、实践应用

        在实际项目中尝试使用魔法方法。如，你可以创建一个自定义的集合类，使用__getitem__、__setitem__和__delitem__方法来实现索引操作。只有通过实践应用，你才能更深入地理解魔法方法的用途和重要性。

5、阅读他人代码

        阅读开源项目或他人编写的代码，特别是那些使用了魔法方法的代码，这可以帮助你学习如何在实际项目中使用魔法方法。通过分析他人代码中的魔法方法使用方式，你可以学习到一些新的技巧和最佳实践。

6、参加社区讨论

        参与Python社区的讨论，与其他开发者交流关于魔法方法的使用经验和技巧，在社区中提问或回答关于魔法方法的问题，这可以帮助你更深入地理解魔法方法并发现新的应用场景。

7、持续学习

        Python语言和其生态系统不断发展，新的魔法方法和功能可能会不断被引入，保持对Python社区的关注，及时学习新的魔法方法和最佳实践。

8、练习与总结

        多做练习，通过编写各种使用魔法方法的代码来巩固你的理解，定期总结你学到的知识和经验，形成自己的知识体系。

9、注意兼容性

        在使用魔法方法时，要注意不同Python版本之间的兼容性差异，确保你的代码在不同版本的Python中都能正常工作。

10、避免过度使用

        虽然魔法方法非常强大，但过度使用可能会导致代码难以理解和维护，在编写代码时，要权衡使用魔法方法的利弊，避免滥用。

        总之，学好Python的魔法方法需要不断地学习、实践和总结，只有通过不断地练习和积累经验，你才能更好地掌握这些强大的工具，并在实际项目中灵活运用它们。

四、魔法方法

14、__enter__方法

14-1、语法

__enter__(self) return self # 可以返回任何对象，或者简单地返回self

14-2、参数

14-2-1、self(必须)：一个对实例对象本身的引用，在类的所有方法中都会自动传递。

14-3、功能

        用于在进入一个由with语句控制的代码块时执行特定的操作。

14-4、返回值

        在with语句块开始执行时调用，并通常返回一个对象。

14-5、说明

        在Python的官方文档中，__enter__ 方法是不接受任何参数的。但是，如果你真的看到了带有参数的 __enter__ 方法，那可能是某个特定库或框架的扩展用法，或者可能是对该方法的误用。

14-6、用法

# 014、__enter__方法：# 1、文件操作# 定义一个名为 FileContextManager 的类，用于管理文件的上下文class FileContextManager: # 初始化方法，接收一个文件名作为参数 def __init__(self, filename): # 将传入的文件名保存到实例变量 filename 中 self.filename = filename # 定义 __enter__ 方法，用于 with 语句执行时进入上下文时调用 def __enter__(self): # 使用实例变量 filename 打开文件，并设置文件模式为只读 ('r') self.file = open(self.filename, 'r') # 返回打开的文件对象，以便在 with 语句的 as 子句中引用 return self.file # 定义 __exit__ 方法，用于 with 语句执行完毕后退出上下文时调用 def __exit__(self, exc_type, exc_val, exc_tb): # 关闭之前打开的文件对象 self.file.close() # 使用 with 语句和 FileContextManager 类来管理文件操作# 'test.txt' 是要打开的文件名，'file' 是打开文件后在 with 语句内部使用的引用名with FileContextManager('test.txt') as file: # 读取文件内容并打印，假设文件内容是 "Hello, World!" print(file.read()) # 输出：Hello, World!# 2、数据库连接# 假设有一个数据库连接库（但在这里我们只是模拟它）class DatabaseContext: # 定义一个上下文管理器的进入方法 def __enter__(self): # 当使用 with 语句进入上下文时，该方法会被调用 # 这里只是一个示例，实际中会使用数据库连接库来建立真实的数据库连接 print("Connecting to database...") # 假设我们有一个模拟的数据库连接字符串 self.connection = "Mock Database Connection" # 返回连接对象（或连接字符串），以便在 with 语句的 as 子句中引用 return self.connection # 定义一个上下文管理器的退出方法 def __exit__(self, exc_type, exc_val, exc_tb): # 当 with 语句块结束时，无论是否发生异常，该方法都会被调用 print("Closing database connection...") # 在这里，我们只是打印了一条消息，但在真实情况下，你会关闭数据库连接# 使用 with 语句和 DatabaseContext 类来管理数据库连接的上下文with DatabaseContext() as db: # 在这里使用数据库连接（这里只是打印了一条消息） # 假设 db 是一个数据库连接对象，但在本例中它只是一个字符串 print("Using database:", db) # 当 with 语句块结束时，__exit__ 方法会被自动调用，打印 "Closing database connection..."# 3、临时目录# 导入tempfile模块，用于创建临时文件和目录import tempfile# 导入shutil模块，用于删除目录及其内容import shutil# 定义一个名为TempDirContext的上下文管理类class TempDirContext: # 定义__enter__方法，当使用with语句进入上下文时调用 def __enter__(self): # 使用tempfile.mkdtemp()方法创建一个临时目录 self.temp_dir = tempfile.mkdtemp() # 打印创建的临时目录的路径 print(f"Created temp directory: {self.temp_dir}") # 返回临时目录的路径，以便在with语句的as子句中引用 return self.temp_dir # 定义__exit__方法，当with语句块结束时调用 def __exit__(self, exc_type, exc_val, exc_tb): # 打印将要删除的临时目录的路径 print(f"Deleting temp directory: {self.temp_dir}") # 使用shutil.rmtree()方法删除临时目录及其所有内容 shutil.rmtree(self.temp_dir)# 使用with语句和TempDirContext类来管理临时目录的上下文with TempDirContext() as temp_dir: # 打印正在使用的临时目录的路径 print(f"Using temp directory: {temp_dir}") # 在这里可以使用temp_dir变量来引用临时目录，并执行相关操作 # 例如：在临时目录中创建文件、写入数据等 # 注意：由于此代码片段是一个注释示例，所以没有包含实际的文件操作# 4、锁机制# 导入threading模块，用于多线程操作import threading# 定义一个名为ThreadLock的类，用于管理线程锁class ThreadLock: # 初始化方法，创建一个threading.Lock对象 def __init__(self): self.lock = threading.Lock() # 定义__enter__方法，当使用with语句进入上下文时调用 def __enter__(self): # 尝试获取锁 self.lock.acquire() # 打印锁已被获取的消息 print("Acquired lock") # 返回锁对象（虽然通常不需要返回，但这里为了示例还是返回了） return self.lock # 定义__exit__方法，当with语句块结束时调用 def __exit__(self, exc_type, exc_val, exc_tb): # 释放锁 self.lock.release() # 打印锁已被释放的消息 print("Released lock")# 使用with语句和ThreadLock类来管理线程锁with ThreadLock() as lock: # 在这里进行线程安全的操作 # 注意：因为with语句已经处理了锁的获取和释放，所以这里的lock变量其实不需要使用 # 除非你想在with块内部检查锁的状态或做其他与锁相关的操作 pass# 5、设置和恢复环境变量# 定义一个名为EnvVarContext的上下文管理器类，用于临时修改环境变量class EnvVarContext: def __init__(self, var_name, new_value): # 初始化时接收环境变量名和新值 self.var_name = var_name # 存储要修改的环境变量名 self.new_value = new_value # 存储新的环境变量值 self.old_value = None # 用于存储原始的环境变量值 def __enter__(self): # 当使用with语句进入上下文时调用 # 获取环境变量原来的值 self.old_value = os.environ.get(self.var_name) # 设置新的环境变量值 os.environ[self.var_name] = self.new_value # 打印已设置新的环境变量值 print(f"Set {self.var_name} to {self.new_value}") # 因为with语句的as子句没有使用变量来接收返回值，所以这里不需要返回具体的值 def __exit__(self, exc_type, exc_val, exc_tb): # 当with语句块结束时调用 # 恢复环境变量为原来的值 if self.old_value is not None: # 如果原值存在，则恢复为原值 os.environ[self.var_name] = self.old_value else: # 如果原值不存在，则从环境变量中删除该变量 del os.environ[self.var_name] # 打印环境变量已恢复为原始值或已删除 print(f"Restored {self.var_name} to its original value")# 导入os模块，以便操作环境变量import os# 使用with语句和EnvVarContext类来修改环境变量with EnvVarContext('MY_VAR', 'new_value') as _: # 在with块内部，MY_VAR的值已被临时修改为'new_value' print(os.environ['MY_VAR']) # 输出: new_value# 退出with块后，MY_VAR被恢复为原来的值或删除# 之后的代码可以确认MY_VAR的值是否已被恢复# 6、模拟上下文状态管理class MyContextManager: def __enter__(self): print("Entering the context") # 在这里，你可以设置任何进入上下文时需要的状态或资源 # 例如，打开一个文件、获取一个锁等 self.resource = "some resource" # 假设这是一个需要管理的资源 return self # 通常，__enter__ 应该返回一个对象，以便在 with 块中使用 def __exit__(self, exc_type, exc_value, traceback): print("Exiting the context") # 在这里，你可以执行任何清理工作，例如关闭文件、释放锁等 # 你可以检查 exc_type, exc_value, traceback 来确定是否发生了异常 # 如果发生了异常并且你想重新抛出它，则不应该在这里进行任何操作 # 如果你想忽略异常，你可以在这里返回 True # 如果你想在退出时总是执行一些清理代码，无论是否发生异常，你可以将这里留空# 使用 with 语句和上下文管理器with MyContextManager() as manager: print("Inside the with block, accessing resource:", manager.resource) # 这里是代码块，当执行到这里时，__enter__ 已经被调用 # 当这个代码块执行完毕（或者发生异常）时，__exit__ 将会被调用

15、__eq__方法

15-1、语法

__eq__(self, other, /) Return self==other

15-2、参数

15-2-1、self(必须)：表示调用该方法的对象本身。

15-2-2、other(必须)：表示与self进行相等性比较操作的对象。

15-2-3、/(可选)：这是从Python 3.8开始引入的参数注解语法，它表示这个方法不接受任何位置参数(positional-only parameters)之后的关键字参数(keyword arguments)。

15-3、功能

        用于定义对象之间的相等性比较。

15-4、返回值

        返回一个布尔值(True或False)，表示两个对象是否相等。

15-5、说明

        当你使用==运算符来比较两个对象是否相等时，Python会尝试调用这两个对象的__eq__方法(如果它们定义了的话)。

15-6、用法

# 015、__eq__方法：# 1、简单的数值类# 定义一个名为Number的类，用于封装数值class Number: # 类的初始化方法，当创建Number类的实例时调用 def __init__(self, value): # 给实例对象添加一个属性value，并赋值为传入的参数value self.value = value # 类的相等性比较方法，用于比较两个Number对象是否相等 def __eq__(self, other): # 检查other是否是Number类的实例 if isinstance(other, Number): # 如果是，则比较两个Number对象的value属性是否相等 return self.value == other.value # 如果other不是Number类的实例，则直接返回False return False# 当这个脚本作为主程序运行时，以下代码会被执行if __name__ == '__main__': # 创建两个Number对象，它们的value属性都是5 num1 = Number(5) num2 = Number(5) # 创建一个Number对象，它的value属性是10 num3 = Number(10) # 使用==操作符比较num1和num2是否相等，并打印结果（注释说明预期输出为True） print(num1 == num2) # 输出: True # 使用==操作符比较num1和num3是否相等，并打印结果（注释说明预期输出为False） print(num1 == num3) # 输出: False# 2、点类(二维坐标)# 定义一个名为Point的类，用于表示二维平面上的点class Point: # 初始化方法，当创建Point类的实例时调用 def __init__(self, x, y): # 设置点的x坐标 self.x = x # 设置点的y坐标 self.y = y # 相等性比较方法，用于比较两个Point对象是否相等 def __eq__(self, other): # 检查other是否也是Point类的实例 if isinstance(other, Point): # 如果是，则比较两个点的x和y坐标是否都相等 return self.x == other.x and self.y == other.y # 如果other不是Point类的实例，则返回False return False# 当这个脚本作为主程序运行时，以下代码会被执行if __name__ == '__main__': # 创建一个Point对象p1，其x坐标为1，y坐标为2 p1 = Point(1, 2) # 创建一个Point对象p2，其x坐标为1，y坐标为2（与p1相同） p2 = Point(1, 2) # 创建一个Point对象p3，其x坐标为2，y坐标为2（与p1不同） p3 = Point(2, 2) # 使用==操作符比较p1和p2是否相等，并打印结果（注释说明预期输出为True） print(p1 == p2) # 输出: True # 使用==操作符比较p1和p3是否相等，并打印结果（注释说明预期输出为False） print(p1 == p3) # 输出: False# 3、字符串封装类# 定义一个名为StringWrapper的类，用于封装字符串 class StringWrapper: # 初始化方法，当创建StringWrapper类的实例时调用 def __init__(self, text): # 给实例对象添加一个属性text，并赋值为传入的参数text self.text = text # 相等性比较方法，用于比较两个StringWrapper对象是否相等 def __eq__(self, other): # 检查other是否也是StringWrapper类的实例 if isinstance(other, StringWrapper): # 如果是，则比较两个StringWrapper对象的text属性是否相等 return self.text == other.text # 如果other不是StringWrapper类的实例，则返回False return False# 当这个脚本作为主程序运行时，以下代码会被执行 if __name__ == '__main__': # 创建一个StringWrapper对象s1，封装字符串"hello" s1 = StringWrapper("hello") # 创建一个StringWrapper对象s2，封装字符串"hello"（与s1相同） s2 = StringWrapper("hello") # 创建一个StringWrapper对象s3，封装字符串"world"（与s1不同） s3 = StringWrapper("world") # 使用==操作符比较s1和s2是否相等，并打印结果（注释说明预期输出为True） print(s1 == s2) # 输出: True # 使用==操作符比较s1和s3是否相等，并打印结果（注释说明预期输出为False） print(s1 == s3) # 输出: False# 4、用户类(只比较用户名)# 定义一个名为 User 的类，表示用户信息class User: # 初始化方法，当创建 User 类的实例时会被调用 def __init__(self, username, age): # 设置实例属性 username，存储用户名 self.username = username # 设置实例属性 age，存储年龄 self.age = age # 重写 __eq__ 方法，用于比较两个 User 实例是否相等 # 这里的相等定义为：如果两个 User 实例的 username 属性相等，则它们相等 def __eq__(self, other): # 检查 other 是否是 User 类的实例 if isinstance(other, User): # 如果 other 是 User 类的实例，则比较两者的 username 属性 return self.username == other.username # 如果 other 不是 User 类的实例，则返回 False return False# 如果当前脚本作为主程序运行（而不是被导入为模块）if __name__ == '__main__': # 创建一个 User 实例，用户名为 "Myelsa"，年龄为 18 user1 = User("Myelsa", 18) # 创建一个 User 实例，用户名为 "Myelsa"，年龄为 28 user2 = User("Myelsa", 28) # 创建一个 User 实例，用户名为 "Jimmy"，年龄为 15 user3 = User("Jimmy", 15) # 使用 __eq__ 方法比较 user1 和 user2 是否相等，并打印结果 # 因为它们的 username 相等，所以输出为 True print(user1 == user2) # 输出: True # 使用 __eq__ 方法比较 user1 和 user3 是否相等，并打印结果 # 因为它们的 username 不相等，所以输出为 False print(user1 == user3) # 输出: False# 5、列表包装类(考虑顺序)# 定义一个名为ListWrapper的类，用于包装列表class ListWrapper: # 类的初始化方法，用于设置items属性 def __init__(self, items): # items属性存储传入的列表 self.items = items # 类的相等性比较方法，用于判断两个ListWrapper对象是否相等 def __eq__(self, other): # 检查other是否是ListWrapper的实例 if isinstance(other, ListWrapper): # 如果是，则比较两个ListWrapper对象的items属性是否相等 # 这里使用了Python列表的相等性比较，即元素相同且顺序也相同 return self.items == other.items # 如果other不是ListWrapper的实例，则返回False return False# 如果当前脚本作为主程序运行（而不是被导入）if __name__ == '__main__': # 创建三个ListWrapper对象，分别包装不同的列表 lst1 = ListWrapper([1, 2, 3]) # 第一个ListWrapper对象，包装了列表[1, 2, 3] lst2 = ListWrapper([2, 1, 3]) # 第二个ListWrapper对象，包装了列表[2, 1, 3]，与lst1元素相同但顺序不同 lst3 = ListWrapper([1, 2, 3]) # 第三个ListWrapper对象，包装了列表[1, 2, 3]，与lst1完全相同 # 使用Python内置的相等性比较来比较lst1和lst2 # 因为lst1和lst2的items属性（即包装的列表）元素相同但顺序不同，所以不相等 print(lst1 == lst2) # 输出：False # 使用Python内置的相等性比较来比较lst1和lst3 # 因为lst1和lst3的items属性（即包装的列表）完全相同，所以相等 print(lst1 == lst3) # 输出：True# 6、日期类(考虑年、月、日)# 导入Python内置的date类，该类用于处理日期from datetime import date# 定义一个名为CustomDate的类，用于封装日期功能class CustomDate: # 类的初始化方法，用于创建CustomDate对象时设置日期 def __init__(self, year, month, day): # 使用Python内置的date类来创建一个日期对象，并将其存储在self.date属性中 self.date = date(year, month, day) # 类的相等性比较方法，用于判断两个CustomDate对象是否相等 def __eq__(self, other): # 检查other是否是CustomDate的实例 if isinstance(other, CustomDate): # 如果是，则比较两个CustomDate对象的date属性是否相等 # 这里使用了Python内置的date类的相等性比较 return self.date == other.date # 如果other不是CustomDate的实例，则返回False return False# 如果当前脚本作为主程序运行（而不是被导入）if __name__ == '__main__': # 创建三个CustomDate对象，分别表示不同的日期 date_1 = CustomDate(2024, 10, 24) # 第一个CustomDate对象，表示2024年10月24日 date_2 = CustomDate(2024, 10, 24) # 第二个CustomDate对象，也表示2024年10月24日 date_3 = CustomDate(2024, 10, 8) # 第三个CustomDate对象，表示2024年10月8日 # 使用Python内置的相等性比较来比较date_1和date_2 # 因为date_1和date_2的date属性相同（都是2024年10月24日），所以相等 print(date_1 == date_2) # 输出：True # 使用Python内置的相等性比较来比较date_1和date_3 # 因为date_1和date_3的date属性不同（分别是2024年10月24日和2024年10月8日），所以不相等 print(date_1 == date_3) # 输出：False

16、__exit__方法

16-1、语法

__exit__(self, exc_type, exc_value, exc_tb) return True # 或者False或不返回任何值(即None) # 如果要忽略异常，则返回True

16-2、参数

16-2-1、self(必须)：一个对实例对象本身的引用，在类的所有方法中都会自动传递。

16-2-2、exc_type(可选)：异常类型，如果在with块中引发了异常，则此参数是异常的类型；否则

为None。

16-2-3、exc_value(可选)：异常值，即with块中引发的异常的实例；如果未引发异常则为None。

16-2-4、exc_tb(可选)：追踪信息(traceback object)，它是一个指向引发异常位置的堆栈跟

踪；如果未引发异常则为None。

16-3、功能

        用于在退出一个由with语句控制的代码块时执行特定的操作。

16-4、返回值

16-4-1、返回True，那么它会抑制(忽略)异常，也就是说，即使with块中的代码引发了异常，这个异常也不会被传播到with语句之外。

16-4-2、返回False或不返回任何值(即None)，那么with块中引发的异常将会按照正常的异常处理机制来处理。

16-5、说明

        无

16-6、用法

# 016、__exit__方法：# 1、文件操作# 定义一个名为 FileContextManager 的类，用于管理文件的上下文class FileContextManager: # 初始化方法，接收一个文件名作为参数 def __init__(self, filename): # 将传入的文件名保存到实例变量 filename 中 self.filename = filename # 定义 __enter__ 方法，用于 with 语句执行时进入上下文时调用 def __enter__(self): # 使用实例变量 filename 打开文件，并设置文件模式为只读 ('r') self.file = open(self.filename, 'r') # 返回打开的文件对象，以便在 with 语句的 as 子句中引用 return self.file # 定义 __exit__ 方法，用于 with 语句执行完毕后退出上下文时调用 def __exit__(self, exc_type, exc_val, exc_tb): # 关闭之前打开的文件对象 self.file.close() # 使用 with 语句和 FileContextManager 类来管理文件操作# 'test.txt' 是要打开的文件名，'file' 是打开文件后在 with 语句内部使用的引用名with FileContextManager('test.txt') as file: # 读取文件内容并打印，假设文件内容是 "Hello, World!" print(file.read()) # 输出：Hello, World!# 2、数据库连接# 假设有一个数据库连接库（但在这里我们只是模拟它）class DatabaseContext: # 定义一个上下文管理器的进入方法 def __enter__(self): # 当使用 with 语句进入上下文时，该方法会被调用 # 这里只是一个示例，实际中会使用数据库连接库来建立真实的数据库连接 print("Connecting to database...") # 假设我们有一个模拟的数据库连接字符串 self.connection = "Mock Database Connection" # 返回连接对象（或连接字符串），以便在 with 语句的 as 子句中引用 return self.connection # 定义一个上下文管理器的退出方法 def __exit__(self, exc_type, exc_val, exc_tb): # 当 with 语句块结束时，无论是否发生异常，该方法都会被调用 print("Closing database connection...") # 在这里，我们只是打印了一条消息，但在真实情况下，你会关闭数据库连接# 使用 with 语句和 DatabaseContext 类来管理数据库连接的上下文with DatabaseContext() as db: # 在这里使用数据库连接（这里只是打印了一条消息） # 假设 db 是一个数据库连接对象，但在本例中它只是一个字符串 print("Using database:", db) # 当 with 语句块结束时，__exit__ 方法会被自动调用，打印 "Closing database connection..."# 3、临时目录# 导入tempfile模块，用于创建临时文件和目录import tempfile# 导入shutil模块，用于删除目录及其内容import shutil# 定义一个名为TempDirContext的上下文管理类class TempDirContext: # 定义__enter__方法，当使用with语句进入上下文时调用 def __enter__(self): # 使用tempfile.mkdtemp()方法创建一个临时目录 self.temp_dir = tempfile.mkdtemp() # 打印创建的临时目录的路径 print(f"Created temp directory: {self.temp_dir}") # 返回临时目录的路径，以便在with语句的as子句中引用 return self.temp_dir # 定义__exit__方法，当with语句块结束时调用 def __exit__(self, exc_type, exc_val, exc_tb): # 打印将要删除的临时目录的路径 print(f"Deleting temp directory: {self.temp_dir}") # 使用shutil.rmtree()方法删除临时目录及其所有内容 shutil.rmtree(self.temp_dir)# 使用with语句和TempDirContext类来管理临时目录的上下文with TempDirContext() as temp_dir: # 打印正在使用的临时目录的路径 print(f"Using temp directory: {temp_dir}") # 在这里可以使用temp_dir变量来引用临时目录，并执行相关操作 # 例如：在临时目录中创建文件、写入数据等 # 注意：由于此代码片段是一个注释示例，所以没有包含实际的文件操作# 4、锁机制# 导入threading模块，用于多线程操作import threading# 定义一个名为ThreadLock的类，用于管理线程锁class ThreadLock: # 初始化方法，创建一个threading.Lock对象 def __init__(self): self.lock = threading.Lock() # 定义__enter__方法，当使用with语句进入上下文时调用 def __enter__(self): # 尝试获取锁 self.lock.acquire() # 打印锁已被获取的消息 print("Acquired lock") # 返回锁对象（虽然通常不需要返回，但这里为了示例还是返回了） return self.lock # 定义__exit__方法，当with语句块结束时调用 def __exit__(self, exc_type, exc_val, exc_tb): # 释放锁 self.lock.release() # 打印锁已被释放的消息 print("Released lock")# 使用with语句和ThreadLock类来管理线程锁with ThreadLock() as lock: # 在这里进行线程安全的操作 # 注意：因为with语句已经处理了锁的获取和释放，所以这里的lock变量其实不需要使用 # 除非你想在with块内部检查锁的状态或做其他与锁相关的操作 pass# 5、设置和恢复环境变量# 定义一个名为EnvVarContext的上下文管理器类，用于临时修改环境变量class EnvVarContext: def __init__(self, var_name, new_value): # 初始化时接收环境变量名和新值 self.var_name = var_name # 存储要修改的环境变量名 self.new_value = new_value # 存储新的环境变量值 self.old_value = None # 用于存储原始的环境变量值 def __enter__(self): # 当使用with语句进入上下文时调用 # 获取环境变量原来的值 self.old_value = os.environ.get(self.var_name) # 设置新的环境变量值 os.environ[self.var_name] = self.new_value # 打印已设置新的环境变量值 print(f"Set {self.var_name} to {self.new_value}") # 因为with语句的as子句没有使用变量来接收返回值，所以这里不需要返回具体的值 def __exit__(self, exc_type, exc_val, exc_tb): # 当with语句块结束时调用 # 恢复环境变量为原来的值 if self.old_value is not None: # 如果原值存在，则恢复为原值 os.environ[self.var_name] = self.old_value else: # 如果原值不存在，则从环境变量中删除该变量 del os.environ[self.var_name] # 打印环境变量已恢复为原始值或已删除 print(f"Restored {self.var_name} to its original value")# 导入os模块，以便操作环境变量import os# 使用with语句和EnvVarContext类来修改环境变量with EnvVarContext('MY_VAR', 'new_value') as _: # 在with块内部，MY_VAR的值已被临时修改为'new_value' print(os.environ['MY_VAR']) # 输出: new_value# 退出with块后，MY_VAR被恢复为原来的值或删除# 之后的代码可以确认MY_VAR的值是否已被恢复# 6、模拟上下文状态管理class MyContextManager: def __enter__(self): print("Entering the context") # 在这里，你可以设置任何进入上下文时需要的状态或资源 # 例如，打开一个文件、获取一个锁等 self.resource = "some resource" # 假设这是一个需要管理的资源 return self # 通常，__enter__ 应该返回一个对象，以便在 with 块中使用 def __exit__(self, exc_type, exc_value, traceback): print("Exiting the context") # 在这里，你可以执行任何清理工作，例如关闭文件、释放锁等 # 你可以检查 exc_type, exc_value, traceback 来确定是否发生了异常 # 如果发生了异常并且你想重新抛出它，则不应该在这里进行任何操作 # 如果你想忽略异常，你可以在这里返回 True # 如果你想在退出时总是执行一些清理代码，无论是否发生异常，你可以将这里留空# 使用 with 语句和上下文管理器with MyContextManager() as manager: print("Inside the with block, accessing resource:", manager.resource) # 这里是代码块，当执行到这里时，__enter__ 已经被调用 # 当这个代码块执行完毕（或者发生异常）时，__exit__ 将会被调用

五、推荐阅读

1、Python筑基之旅

2、Python函数之旅

3、Python算法之旅

4、博客个人主页