Kontextmanager selbst erstellen

Funktionen in Python schreiben

Shayne Miel

Software Architect @ Duo Security

Zwei Möglichkeiten, einen Kontextmanager zu definieren

  • Klassenbasiert
  • Funktionsbasiert
Funktionen in Python schreiben

Zwei Möglichkeiten, einen Kontextmanager zu definieren

  • Klassenbasiert
  • Funktionsbasiert
Funktionen in Python schreiben

Wie man einen Kontextmanager erstellt


def my_context():

  # Add any set up code you need

  yield

  # Add any teardown code you need
  1. Definiere die Funktion.
  2. Optional: Füge den Code hinzu, den dein Kontext braucht.
  3. Benutze das Schlüsselwort „yield“.
  4. Optional: Füge den Aufräum-Code hinzu, den dein Kontext braucht.
Funktionen in Python schreiben

Wie man einen Kontextmanager erstellt

@contextlib.contextmanager
def my_context():

  # Add any set up code you need

  yield

  # Add any teardown code you need
  1. Definiere die Funktion.
  2. Optional: Füge den Code hinzu, den dein Kontext braucht.
  3. Benutz das Schlüsselwort „yield“.
  4. Optional: Füge den Aufräum-Code hinzu, den dein Kontext braucht.
  5. Füge den Dekorator `@contextlib.contextmanager` hinzu.
Funktionen in Python schreiben

Das Schlüsselwort „yield“

@contextlib.contextmanager
def my_context():
  print('hello')
  yield 42
  print('goodbye')

with my_context() as foo:
  print('foo is {}'.format(foo))

hello
foo is 42
goodbye
Funktionen in Python schreiben

Setup und teardown (Erstellen und Entfernen)

@contextlib.contextmanager
def database(url):
  # set up database connection
  db = postgres.connect(url)

  yield db

  # tear down database connection
  db.disconnect()

url = 'http://datacamp.com/data'
with database(url) as my_db:
  course_list = my_db.execute(
    'SELECT * FROM courses'
  )

Leerzeichen

arrow

Funktionen in Python schreiben

Setup und teardown (Erstellen und Entfernen)

@contextlib.contextmanager
def database(url):
  # set up database connection
  db = postgres.connect(url)

  yield db

  # tear down database connection
  db.disconnect()

url = 'http://datacamp.com/data'
with database(url) as my_db:
  course_list = my_db.execute(
    'SELECT * FROM courses'
  )

Leerzeichen

Leerzeichen

Leerzeichen

arrow

Funktionen in Python schreiben

Setup und teardown (Erstellen und Aufräumen)

@contextlib.contextmanager
def database(url):
  # set up database connection
  db = postgres.connect(url)

  yield db

  # tear down database connection
  db.disconnect()

url = 'http://datacamp.com/data'
with database(url) as my_db:
  course_list = my_db.execute(
    'SELECT * FROM courses'
  )

Leerzeichen

Leerzeichen

Leerzeichen

Leerzeichen

Leerzeichen

Leerzeichen

arrow

Funktionen in Python schreiben

Einen Wert oder „None“ liefern

@contextlib.contextmanager
def database(url):
  # set up database connection
  db = postgres.connect(url)

  yield db

  # tear down database connection
  db.disconnect()

url = 'http://datacamp.com/data'
with database(url) as my_db:
  course_list = my_db.execute(
    'SELECT * FROM courses'
  )

@contextlib.contextmanager
def in_dir(path):
  # save current working directory
  old_dir = os.getcwd()

  # switch to new working directory
  os.chdir(path)

  yield

  # change back to previous
  # working directory
  os.chdir(old_dir)

with in_dir('/data/project_1/'):
  project_files = os.listdir()
Funktionen in Python schreiben

Lass uns üben!

Funktionen in Python schreiben

Preparing Video For Download...