浅谈Python中的with，可能有你不知道的

Python中的with，没那么简单，虽然也不难

https://docs.python.org/zh-cn/3.9/reference/compound_stmts.html#the-with-statement 8.3 with 语句

https://docs.python.org/zh-cn/3.9/reference/datamodel.html#context-managers 3.3.9 with语句上下文管理器

with 语句

• 语法

  with_stmt ::=  "with" with_item ("," with_item)* ":" suite
  with_item ::=  expression ["as" target]
  

• 执行过程

  1. 对上下文表达式 (在 with_item 中给出的表达式) 求值以获得一个上下文管理器。

  2. 载入上下文管理器的 __enter__() 以便后续使用。

  3. 载入上下文管理器的 __exit__() 以便后续使用。

  4. 发起调用上下文管理器的 __enter__() 方法。

  5. 如果 with 语句中包含一个目标，来自 __enter__() 的返回值将被赋值给它。

     with语句会保证如果 __enter__()方法返回时未发生错误，则 __exit__()将总是被调用。 因此，如果在对目标列表赋值期间发生错误，则会将其视为在语句体内部发生的错误。 参见下面的第 6 步。
     

  6. 执行语句体。

  7. 发起调用上下文管理器的 __exit__() 方法。 如果语句体的退出是由异常导致的，则其类型、值和回溯信息将被作为参

• 说明

  • 如果语句体的退出是由异常导致的，并且来自 __exit__() 方法的返回值为假，则该异常会被重新引发。 如果返回值为真，则该异常会被抑制，并会继续执行 with 语句之后的语句。
  • 如果语句体由于异常以外的任何原因退出，则来自 __exit__() 的返回值会被忽略，并会在该类退出正常的发生位置继续执行

示例1：打开文件的with

• 最常见的就是with打开文件

  with open(r'd:\with.txt','w') as f:
      f.write('hello')
  

  • 这样我们就不用去f.close()
  • 在这里你并不能从源码上看到enter和exit的影子
  • 底层是c代码的实现

  示例2：selenium中的with

  • SELENIUM中的WebDriver就是一个典型的实现了enter和exit的

  • 所以我们可以这么用（在selenium官网也有类似的DEMO）

    from selenium import webdriver
    from time import sleep
    
    with webdriver.Chrome() as driver:
        driver.get('https://www.baidu.com')
        sleep(2)
    

    • 上面的代码你运行后可以看到打开了一个chrome浏览器，输入了百度的网址，等待2s后自动关闭了

  • 我们来追溯下源码

    with webdriver.Chrome() as driver:   # ctrl+点击Chrome()
    
    class WebDriver(ChromiumDriver): # ctrl+点击ChromiumDriver
    
    class ChromiumDriver(RemoteWebDriver):  #  ctrl+点击RemoteWebDriver
    
    class WebDriver(BaseWebDriver):  #
        ...
    
    	def __enter__(self):
            return self
    
        def __exit__(self,
                     exc_type: typing.Optional[typing.Type[BaseException]],
                     exc: typing.Optional[BaseException],
                     traceback: typing.Optional[types.TracebackType]):
            self.quit()
    

    • 可以看到继承的WebDriver中实现了enter和exit，其实也没做啥，exit主要是退出了。
    • 这样你再来看下面的enter和exit的官方说明似乎更好理解一些

关于enter

• object.__enter__(self)

  进入与此对象相关的运行时上下文。 with 语句将会绑定这个方法的返回值到 as 子句中指定的目标，如果有的话。

关于exit

• object.__exit__(self, exc_type, exc_value, traceback)

  退出关联到此对象的运行时上下文。 各个参数描述了导致上下文退出的异常。 如果上下文是无异常地退出的，三个参数都将为 None。

  如果提供了异常，并且希望方法屏蔽此异常（即避免其被传播），则应当返回真值。 否则的话，异常将在退出此方法时按正常流程处理。

  请注意 __exit__() 方法不应该重新引发被传入的异常，这是调用者的责任

其他示例

• 复制一个文件的内容到另外一个文件中，不考虑异常，只演示用法

  with open(r'd:\old.txt','r') as f1,open(r'd:\new.txt','w') as f2:
      f2.write(f1.read())
  

• 打开多个网页，串行的。

  from selenium import webdriver
  from time import sleep
  with webdriver.Chrome() as driver1,webdriver.Chrome() as driver2:
      driver1.get('https://www.baidu.com')
      sleep(2)
      driver2.get('https://cn.bing.com')
      sleep(2)
  

• with的另外一种写法：肯定是不建议的，但可以理解一下。

  with open(r'd:\old.txt','r') as f:   # 原始的写法，打印文件的内容
      print(f.read())
  

  import sys
  
  manager = open(r'd:\old.txt','r')
  enter = type(manager).__enter__
  exit = type(manager).__exit__
  value = enter(manager)
  
  hit_expect = False
  
  try:
      TARGET = value
      print(TARGET.read())
  except:
      hit_expect = True
      if not exit(manager,*sys.exc_info()):
          raise 'hehe'
  finally:
      if not hit_expect:
          exit(manager, None,None,None)
  

1. 数传递给 __exit__()。 否则的话，将提供三个 None 参数