Décorateurs et métadonnées

Écrire des fonctions en Python

Shayne Miel

Software Architect @ Duo Security

def sleep_n_seconds(n=10):
  """Met en pause le traitement pendant n secondes.

  Args:
    n (int): Nombre de secondes de pause.
  """
  time.sleep(n)

print(sleep_n_seconds.__doc__)

Met en pause le traitement pendant n secondes.

  Args:
    n (int): Nombre de secondes de pause.

def sleep_n_seconds(n=10):
  """Met en pause le traitement pendant n secondes.

  Args:
    n (int): Nombre de secondes de pause.
  """
  time.sleep(n)

print(sleep_n_seconds.__name__)

sleep_n_seconds

print(sleep_n_seconds.__defaults__)

(10,)

@timer
def sleep_n_seconds(n=10):
  """Met en pause le traitement pendant n secondes.

  Args:
    n (int): Nombre de secondes de pause.
  """
  time.sleep(n)

print(sleep_n_seconds.__doc__)

print(sleep_n_seconds.__name__)

wrapper

Le décorateur timer

def timer(func):
  """Un décorateur qui affiche le temps d'exécution d'une fonction."""

  def wrapper(*args, **kwargs):
    t_start = time.time()

    result = func(*args, **kwargs)

    t_total = time.time() - t_start
    print('{} a pris {}s'.format(func.__name__, t_total))

    return result

  return wrapper

from functools import wraps

def timer(func):
  """Un décorateur qui affiche le temps d'exécution d'une fonction."""

  @wraps(func)
  def wrapper(*args, **kwargs):

    t_start = time.time()

    result = func(*args, **kwargs)

    t_total = time.time() - t_start
    print('{} a pris {}s'.format(func.__name__, t_total))

    return result

  return wrapper

@timer
def sleep_n_seconds(n=10):
  """Met en pause le traitement pendant n secondes.

  Args:
    n (int): Nombre de secondes de pause.
  """
  time.sleep(n)

print(sleep_n_seconds.__doc__)

Met en pause le traitement pendant n secondes.

  Args:
    n (int): Nombre de secondes de pause.

@timer
def sleep_n_seconds(n=10):
  """Met en pause le traitement pendant n secondes.

  Args:
    n (int): Nombre de secondes de pause.
  """
  time.sleep(n)

print(sleep_n_seconds.__name__)

sleep_n_seconds

Accéder à la fonction d'origine

@timer
def sleep_n_seconds(n=10):
  """Met en pause le traitement pendant n secondes.

  Args:
    n (int): Nombre de secondes de pause.
  """
  time.sleep(n)

sleep_n_seconds.__wrapped__

<function sleep_n_seconds at 0x7f52cab44ae8>

Passons à la pratique !

Écrire des fonctions en Python