Python 高手编程系列八:缓存

缓存装饰器与参数检查十分相似,不过它重点是关注那些内部状态不会影响输出的函数。

每组参数都可以链接到唯一的结果。这种编程风格是函数式编程(functional programming,参

https://en.wikipedia.org/wiki/Functional_programming)的特点,当输入值有限时可以使用。

因此,缓存装饰器可以将输出与计算它所需要的参数放在一起,并在后续的调用中直

接返回它。这种行为被称为 memoizing(参见 https://en.wikipedia.org/wiki/Memoization),

很容易被实现为一个装饰器:

import time

import hashlib

import pickle

cache = {}

def is_obsolete(entry, duration):

return time.time() - entry['time'] > duration

def compute_key(function, args, kw):

key = pickle.dumps((function.name , args, kw))

return hashlib.sha1(key).hexdigest()

def memoize(duration=10):

def _memoize(function):

def __memoize(*args, **kw):

key = compute_key(function, args, kw)

是否已经拥有它了?

if (key in cache and

not is_obsolete(cache[key], duration)):

print('we got a winner')

return cache[key]['value']

计算

result = function(*args, **kw)

保存结果

cache[key] = {

'value': result,

'time': time.time()

}

return result

return __memoize

return _memoize

利用已排序的参数值来构建 SHA 哈希键,并将结果保存在一个全局字典中。利用 pickle

来建立 hash,这是冻结所有作为参数传入的对象状态的快捷方式,以确保所有参数都满足

要求。举个例子,如果用一个线程或套接字作为参数,那么会引发 PicklingError(参

https://docs.python.org/3/library/pickle.html)。duration 参数的作用是,如果上一次函数

调用已经过去了太长时间,那么它会使缓存值无效。

下面是一个使用示例:

@memoize()

... def very_ very very complex stuff(a, b):
... # 如果在执行这个计算时计算机过热
... # 请考虑中止程序
... return a + b
...
very
very very complex stuff(2, 2)
4
very
very very complex stuff(2, 2)
we got a winner
4
@memoize(1) # 1 秒后令缓存失效
... def very
very very complex stuff(a, b):
... return a + b
...
very
very very complex stuff(2, 2)
4
very
very very complex stuff(2, 2)
we got a winner
4
cache
{'c2727f43c6e39b3694649ee0883234cf': {'value': 4, 'time':
1199734132.7102251)}
time.sleep(2)
very
very very complex _stuff(2, 2)

4

缓存代价高昂的函数可以显著提高程序的总体性能,但必须小心使用。缓存值还可以

与函数本身绑定,以管理其作用域和生命周期,代替集中化的字典。但在任何情况下,更

高效的装饰器会使用基于高级缓存算法的专用缓存库。

代理

代理装饰器使用全局机制来标记和注册函数。举个例子,一个根据当前用户来保护代

码访问的安全层可以使用集中式检查器和相关的可调用对象要求的权限来实现:

class User(object):

def init (self, roles):

self.roles = roles

class Unauthorized(Exception):

pass

def protect(role):

def _protect(function):

def __protect(*args, **kw):

user = globals().get('user')

if user is None or role not in user.roles:

raise Unauthorized("I won't tell you")

return function(*args, **kw)

return __protect

return protect
这一模型常用于 Python Web 框架中,用于定义可发布类的安全性。例如,Django 提供
装饰器来保护函数访问的安全。
下面是一个示例,当前用户被保存在一个全局变量中。在方法被访问时装饰器会检查
他/她的角色:
tarek = User(('admin', 'user'))
bill = User(('user',))
class MySecrets(object):
... @protect('admin')
... def waffle
recipe(self):
... print('use tons of butter!')
...
these
_are = MySecrets()

user = tarek

these_ are.waffle recipe()
use tons of butter!
user = bill
these
are.waffle _recipe()

Traceback (most recent call last):

File "", line 1, in

File "", line 7, in wrap

__main __.Unauthorized: I won't tell you

相关推荐
IVEN_7 小时前
只会Python皮毛?深入理解这几点,轻松进阶全栈开发
python·全栈
Ray Liang8 小时前
用六边形架构与整洁架构对比是伪命题?
java·python·c#·架构设计
AI攻城狮8 小时前
如何给 AI Agent 做"断舍离":OpenClaw Session 自动清理实践
python
千寻girling8 小时前
一份不可多得的 《 Python 》语言教程
人工智能·后端·python
AI攻城狮11 小时前
用 Playwright 实现博客一键发布到稀土掘金
python·自动化运维
曲幽12 小时前
FastAPI分布式系统实战:拆解分布式系统中常见问题及解决方案
redis·python·fastapi·web·httpx·lock·asyncio
孟健1 天前
Karpathy 用 200 行纯 Python 从零实现 GPT:代码逐行解析
python
码路飞1 天前
写了个 AI 聊天页面,被 5 种流式格式折腾了一整天 😭
javascript·python
曲幽1 天前
FastAPI压力测试实战:Locust模拟真实用户并发及优化建议
python·fastapi·web·locust·asyncio·test·uvicorn·workers