Python装饰器

1. 函数

def foo():
    return 1

foo()

2. 作用域

全局作用域与本地作用域

a = "a global variable"

def foo():
    print locals()

print globals()
# {'a': 'a global variable', ...}

foo() # 2
# {}

内置函数 globals 返回一个包含所有 Python 解释器知道的变量名称的字典。

在#2调用了函数并把内部本地作用域里面的内容打印出来。函数foo有自己独立的命名空间。

3. 变量解析规则

在 Python 的作用域规则里,创建变量一定会在当前作用域里创建一个变量,但是访问或者修改变量时会先在当前作用域查找变量,如果没有找到匹配的变量会依次向上在闭合的作用域里进行查找。

a = "a global variable"

def foo():
    print a # 1

foo()
# a global variable

在#1处,Python 解释器会尝试查找变量 a,在函数本地作用域是找不到的,所以接着会去上层作用域里查找。

如果在函数foo里对全局变量赋值,结果却与想的不一样:

a = "a global variable"

def foo():
    a = "test" # 1
    print locals()

foo()
# {'a': 'test'}
a
# 'a global variable'

全局变量可以在函数内部被访问到,但赋值不行。在#1处,实际创建了一个局部变量,同名的全局变量无法被访问到。

4. 变量生存周期

变量不仅生存在一个个命名空间内,他们都有自己的生存周期。

def foo():
    x = 1

foo()
print x #1
# Traceback (most recent call last):
#  File "<stdin>", line 1, in <module>
# NameError: name 'x' is not defined

1处发生的错误不仅仅是因为作用域规则导致的。它还和Python以及其他很多编程语言中函数调用实现的机制有关。在这个地方这个执行时间点变量x不存在。函数foo的命名空间随着函数调用开始而开始,结束而销毁。

5. 函数参数

参数会变成本地变量存在于函数内部。

def foo(x):
    print locals()

foo(1)
# {'x': 1}
def foo(x, y=0): # 1
    return x - y

foo(3, 1) # 2

2
foo(3) # 3
3

foo() # 4
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: foo() takes at least 1 argument (0 given)

foo(y=1, x=3) # 5
2

6. 嵌套函数

Pyhton 允许创建嵌套函数。作用域和变量生存周期依旧适用。

def outer():
    x = 1
    print locals()
    def inner():
        print x # 1
        print locals()
    inner() # 2

outer()
1

7. 函数是 Python 里的一级类对象

Python 里的函数与其他类型一样都是对象。

issubclass(int, object) # Python 中所有对象均继承自 object
True

def foo():
    pass

foo.__class__
<type 'function'>

issubclass(foo.__class__, object)
True

将函数作为参数传递:

def add(x, y):
    return x + y

def sub(x, y):
    return x - y

def apply(func, x, y): # 1
    return func(x, y)  # 2

apply(add, 2, 1) # 3
3

apply(sub, 2, 1)
1

将函数作为返回值传递:

def outer():
    def inner():
        print "Inside inner"
    return inner

foo = outer()
foo
<function inner at 0x>
foo()
Inside inner

8. 闭包

def outer():
    x = 1
    def inner():
        print x # 1
    return inner

foo = outer()
foo.func_closure
(<cell at 0x1007397f8: int object at 0x...)

Python支持一个叫做函数闭包的特性,用人话来讲就是,嵌套定义在非全局作用域里面的函数能够记住它在被定义的时候它所处的封闭命名空间。这能够通过查看函数的func_closure属性得出结论,这个属性里面包含封闭作用域里面的值(只会包含被捕捉到的值,比如x,如果在outer里面还定义了其他的值,封闭作用域里面是不会有的)

9. 装饰器

装饰器其实就是一个闭包。

def outer(some_func):
    def inner():
        print "before some_func"
        ret = some_func() # 1
        return ret + 1
    return inner
def foo():
    return 1
decorated = outer(foo) # 2
decorated()
before some_func
2

接下来写一个有用的装饰器:

class Coordinate(object):
     def __init__(self, x, y):
         self.x = x
         self.y = y
     def __repr__(self):
         return "Coord: " + str(self.__dict__)
def add(a, b):
     return Coordinate(a.x + b.x, a.y + b.y)
def sub(a, b):
     return Coordinate(a.x - b.x, a.y - b.y)
one = Coordinate(100, 200)
two = Coordinate(300, 200)
add(one, two)
Coord: {'y': 400, 'x': 400}

如果加减函数需要一些边界检查的行为怎么办?

现在期望是这样:

one = Coordinate(100, 200)
two = Coordinate(300, 200)
three = Coordinate(-100, -100)
sub(one, two)
Coord: {'y': 0, 'x': -200}
add(one, three)
Coord: {'y': 100, 'x': 0}

边界检查的装饰器:

def wrapper(func):
     def checker(a, b): # 1
         if a.x < 0 or a.y < 0:
             a = Coordinate(a.x if a.x > 0 else 0, a.y if a.y > 0 else 0)
         if b.x < 0 or b.y < 0:
             b = Coordinate(b.x if b.x > 0 else 0, b.y if b.y > 0 else 0)
         ret = func(a, b)
         if ret.x < 0 or ret.y < 0:
             ret = Coordinate(ret.x if ret.x > 0 else 0, ret.y if ret.y > 0 else 0)
         return ret
     return checker
add = wrapper(add)
sub = wrapper(sub)
sub(one, two)
Coord: {'y': 0, 'x': 0}
add(one, three)
Coord: {'y': 200, 'x': 100}

10. 使用 @ 标识符将装饰器应用到函数

Python 2.4 支持使用标识符 @ 将装饰器应用在函数上。

以前的写法:

add = wrapper(add)

使用@:

@wrapper
def add(a, b):
    return Coordinate(a.x + b.x, a.y + b.y)

使用 @ 与先前简单的用包装方法替代原有方法是一模一样的。只是用了一个语法糖让装饰的行为更优雅。

11. *args**kwargs

arg 和 kwargs 并不是 Python 语法的一部分,但在定义函数的时候,使用这样的变量名算是一个不成文的约定。

def one(*args):
    print args # 1

one()
()

one(1, 2, 3)
(1, 2, 3)

def two(x, y, *args): # 2
    print x, y, args

two('a', 'b', 'c')
a b ('c',)
def foo(**kwargs):
     print kwargs
foo()
{}
foo(x=1, y=2)
{'y': 2, 'x': 1}
dct = {'x': 1, 'y': 2}
def bar(x, y):
     return x + y
bar(**dct)
3

12. 更通用的装饰器

def logger(func):
     def inner(*args, **kwargs): #1
         print "Arguments were: %s, %s" % (args, kwargs)
         return func(*args, **kwargs) #2
     return inner

@logger
def foo1(x, y=1):
    return x * y

@logger
def foo2():
    return 2

foo1(5, 4)
Arguments were: (5, 4), {}
20

foo1(1)
Arguments were: (1,), {}
1

foo2()
Arguments were: (), {}
2
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,530评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 86,403评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,120评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,770评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,758评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,649评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,021评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,675评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,931评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,659评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,751评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,410评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,004评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,969评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,042评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,493评论 2 343

推荐阅读更多精彩内容