裝飾器

Python 函式寫作

Shayne Miel

Software Architect @ Duo Security

函式

function

裝飾器

function with decorator

修改輸入

modify inputs

修改輸出

modify outputs

修改函式

modify function

裝飾器長什麼樣？

@double_args
def multiply(a, b):
  return a * b

multiply(1, 5)

double_args 裝飾器

def multiply(a, b):
  return a * b

def double_args(func):
  return func

new_multiply = double_args(multiply)

new_multiply(1, 5)

multiply(1, 5)

double_args 裝飾器

def multiply(a, b):
  return a * b

def double_args(func):

  # Define a new function that we can modify
  def wrapper(a, b):

    # For now, just call the unmodified function
    return func(a, b)

  # Return the new function
  return wrapper

new_multiply = double_args(multiply)

new_multiply(1, 5)

double_args 裝飾器

def multiply(a, b):
  return a * b

def double_args(func):
  def wrapper(a, b):

    # Call the passed in function, but double each argument
    return func(a * 2, b * 2)

  return wrapper

new_multiply = double_args(multiply)

new_multiply(1, 5)

double_args 裝飾器

def multiply(a, b):
  return a * b

def double_args(func):
  def wrapper(a, b):
    return func(a * 2, b * 2)
  return wrapper

multiply = double_args(multiply)

multiply(1, 5)

multiply.__closure__[0].cell_contents

<function multiply at 0x7f0060c9e620>

裝飾器語法

def double_args(func):
  def wrapper(a, b):
    return func(a * 2, b * 2)
  return wrapper



def multiply(a, b):
  return a * b

multiply = double_args(multiply)

multiply(1, 5)

def double_args(func):
  def wrapper(a, b):
    return func(a * 2, b * 2)
  return wrapper


@double_args
def multiply(a, b):
  return a * b



multiply(1, 5)

一起來練習吧！

Python 函式寫作