Rédaction des gestionnaires de contexte

Écrire des fonctions en Python

Shayne Miel

Software Architect @ Duo Security

Deux méthodes pour définir un gestionnaire de contexte

  • Basé sur les classes
  • Basé sur les fonctions
Écrire des fonctions en Python

Deux méthodes pour définir un gestionnaire de contexte

  • Basé sur les classes
  • Basé sur les fonctions
Écrire des fonctions en Python

Comment créer un gestionnaire de contexte


def my_context():

  # Add any set up code you need

  yield

  # Add any teardown code you need
  1. Définir une fonction.
  2. (facultatif) Ajouter tout code de configuration requis par votre contexte.
  3. Utiliser le mot-clé « yield ».
  4. (facultatif) Ajouter tout code de démontage requis par votre contexte.
Écrire des fonctions en Python

Comment créer un gestionnaire de contexte

@contextlib.contextmanager
def my_context():

  # Add any set up code you need

  yield

  # Add any teardown code you need
  1. Définir une fonction.
  2. (facultatif) Ajouter tout code de configuration requis par votre contexte.
  3. Utiliser le mot-clé « yield ».
  4. (facultatif) Ajouter tout code de démontage requis par votre contexte.
  5. Ajouter le décorateur `@contextlib.contextmanager`.
Écrire des fonctions en Python

Le mot-clé « 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
Écrire des fonctions en Python

Configuration et démontage

@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'
  )

blankspace

arrow

Écrire des fonctions en Python

Configuration et démontage

@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'
  )

blankspace

blankspace

blankspace

arrow

Écrire des fonctions en Python

Configuration et démontage

@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'
  )

blankspace

blankspace

blankspace

blankspace

blankspace

blankspace

arrow

Écrire des fonctions en Python

Renvoyer une valeur ou None

@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()
Écrire des fonctions en Python

Passons à la pratique !

Écrire des fonctions en Python

Preparing Video For Download...