>  기사  >  백엔드 개발  >  Python에서 with 키워드 사용에 대한 자세한 설명

Python에서 with 키워드 사용에 대한 자세한 설명

高洛峰
高洛峰원래의
2017-03-28 15:21:121590검색

이 글에서는 Python에서 with 키워드의 자세한 사용에 대한 관련 정보를 주로 소개합니다. Python에서 with 키워드는 필요한 컨텍스트 프로토콜 객체의 구현을 관리하는 데 좋습니다. it 다음을 참고하시면 됩니다

">

Python 2.5에서는 with 키워드가 추가되었습니다. 일반적으로 사용되는 try ... Except ... finally ... 패턴을 만듭니다. 매우 편리합니다. 가장 전형적인 예를 살펴보세요.

with open('file.txt') as f:
  content = f.read()

이 코드에서는 with의 코드 블록을 실행하는 동안 무슨 일이 일어나든 결국 프로세스 중에 예외가 발생합니다. 그런 다음 예외가 발생하기 전에 프로그램은 먼저 열려 있는 파일

을 닫은 다음 데이터베이스 트랜잭션 요청을 시작하기 전에 다른 예제

를 살펴봅니다. 이때 이와 같은 코드가 자주 사용됩니다. :

db.begin()
try:
  # do some actions
except:
  db.rollback()
  raise
finally:
  db.commit()

트랜잭션 요청을 시작하는 동작을 with 키워드를 지원하도록 변경한다면 다음과 같은 코드로 충분합니다.

with transaction(db):
  # do some actions

아래에서는 with의 실행 과정을 자세히 설명합니다. 위 코드는

with

기본 with.Expression의 일반적인 실행 과정은 다음과 같습니다.

with EXPR as VAR:
  BLOCK

여기서: EXPR은 임의의 표현식일 수 있습니다. VAR은 선택 사항이므로 일반적인 실행 프로세스는 다음과 같습니다.

  1. EXPR을 계산하고 컨텍스트 관리자를 얻습니다. .

  2. 컨텍스트 관리자의 exit() 메서드는 후속 호출을 위해 저장됩니다. >

  3. 컨텍스트 관리자의 enter() 메서드를 호출합니다. 🎜>

  4. with 표현식에 VAR이 포함되어 있으면 EXPR의 반환 값이
  5. BLOCK에서 표현식을 실행합니다. >
  6. BLOCK 실행 중 이벤트가 발생하면 컨텍스트 관리자의 exit() 메서드를 호출합니다. 예외로 인해 프로그램이 종료되는 경우 예외 유형, 값 및 역추적(예: 반환) sys.exc_info())의 값은 exit() 메서드에 매개 변수로 전달됩니다. 그렇지 않으면 세 개의 None이 전달됩니다. >
  7. 이 프로세스를 다음과 같이 코드로 표현합니다.
  8. mgr = (EXPR)
    exit = type(mgr).exit # 这里没有执行
    value = type(mgr).enter(mgr)
    exc = True
    try:
      try:
        VAR = value # 如果有 as VAR
        BLOCK
      except:
        exc = False
        if not exit(mgr, *sys.exc_info()):
          raise
    finally:
      if exc:
        exit(mgr, None, None, None)

    이 프로세스는 다음과 같습니다. 몇 가지 세부 정보가 있습니다:

  9. 컨텍스트 관리자에 enter() 또는 exit()가 없으면 인터프리터는 AttributeError를 발생시킵니다.
BLOCK에서 예외가 발생한 후, exit() 메서드가 True로 간주할 수 있는 값을 반환하면 해당 예외는 발생하지 않고 후속 코드가 계속 실행됩니다.

다음으로 위의 프로세스를 구현하기 위해 두 가지 방법을 사용하겠습니다.

컨텍스트 관리자 클래스 구현


첫 번째 방법은 인스턴스

속성

db와 컨텍스트 관리자 enter() 및 종료에 필요한 메서드를 포함하는 클래스를 구현하는 것입니다. () .

class transaction(object):
  def init(self, db):
    self.db = db
  def enter(self):
    self.db.begin()
  def exit(self, type, value, traceback):
    if type is None:
      db.commit()
    else:
      db.rollback()

with의 실행 과정을 이해하고 나면 이 구현은 이해하기 쉽습니다. 아래에 소개된 구현 방법은 이해하기가 훨씬 더 복잡합니다.

제너레이터

데코레이터 사용

파이썬의 표준 라이브러리에는 제너레이터를 통해 컨텍스트 관리자를 얻을 수 있는 데코레이터가 있습니다. 제너레이터 데코레이터를 사용한 구현 과정은 다음과 같습니다.

from contextlib import contextmanager
@contextmanager
def transaction(db):
  db.begin()
  try:
    yield db
  except:
    db.rollback()
    raise
  else:
    db.commit()
언뜻 보기에 이 구현은 더 간단하지만 메커니즘은 더 복잡합니다. 실행 프로세스를 살펴보겠습니다.

Python 인터프리터가 Yield 키워드를 인식한 후 def는 일반 함수를 대체하기 위해 생성기

함수

를 생성합니다( 클래스 정의 나는 메소드 대신 함수를 사용하는 것을 선호합니다).

  1. 데코레이터 contextmanager가 호출되고 호출된 후 GeneratorContextManager 인스턴스를 생성하는 도우미 메서드를 반환합니다. 마지막으로 with 표현식의 EXPR은 contentmanager 데코레이터가 반환한 도우미 함수를 호출합니다.

  2. with 표현식은 실제로 도우미 함수를 호출하는 transaction(db)을 호출합니다. 도우미 함수는 생성기를 생성하는 생성기 함수를 호출합니다.
  3. 도우미 함수는 이 생성기를 GeneratorContextManager에 전달하고 GeneratorContextManager의 인스턴스 객체를 컨텍스트 관리자로 생성합니다.
  4. with 表达式调用实例对象的上下文管理器的 enter() 方法。

  5. enter() 方法中会调用这个生成器的 next() 方法。这时候,生成器方法会执行到 yield db 处停止,并将 db 作为 next() 的返回值。如果有 as VAR ,那么它将会被赋值给 VAR 。

  6. with 中的 BLOCK 被执行。

  7. BLOCK 执行结束后,调用上下文管理器的 exit() 方法。 exit() 方法会再次调用生成器的 next() 方法。如果发生 StopIteration 异常,则 pass 。

  8. 如果没有发生异常生成器方法将会执行 db.commit() ,否则会执行 db.rollback() 。

再次看看上述过程的代码大致实现:

def contextmanager(func):
  def helper(*args, **kwargs):
    return GeneratorContextManager(func(*args, **kwargs))
  return helper
class GeneratorContextManager(object):
  def init(self, gen):
    self.gen = gen
  def enter(self):
    try:
      return self.gen.next()
    except StopIteration:
      raise RuntimeError("generator didn't yield")
  def exit(self, type, value, traceback):
    if type is None:
      try:
        self.gen.next()
      except StopIteration:
        pass
      else:
        raise RuntimeError("generator didn't stop")
    else:
      try:
        self.gen.throw(type, value, traceback)
        raise RuntimeError("generator didn't stop after throw()")
      except StopIteration:
        return True
      except:
        if sys.exc_info()[1] is not value:
          raise

总结

Python的 with 表达式包含了很多Python特性。花点时间吃透 with 是一件非常值得的事情。

一些其他的例子

锁机制

@contextmanager
def locked(lock):
  lock.acquired()
  try:
    yield
  finally:
    lock.release()

标准输出重定向

@contextmanager
def stdout_redirect(new_stdout):
  old_stdout = sys.stdout
  sys.stdout = new_stdout
  try:
    yield
  finally:
    sys.stdout = old_stdout
with open("file.txt", "w") as f:
  with stdout_redirect(f):
    print "hello world"

위 내용은 Python에서 with 키워드 사용에 대한 자세한 설명의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.