- c - 在位数组中找到第一个零
- linux - Unix 显示有关匹配两种模式之一的文件的信息
- 正则表达式替换多个文件
- linux - 隐藏来自 xtrace 的命令
我想为一些将文件作为第一个参数的函数编写一个装饰器。装饰器必须实现 context manager protocol (即将包装函数变成上下文管理器),所以我想我需要用类包装函数。
我对装饰器模式并没有真正的经验,之前从未实现过上下文管理器,但我写的东西在 Python 2.7 中有效,如果我注释掉 wraps
,它在 Python 3.3 中也有效行。
from functools import wraps
def _file_reader(func):
"""A decorator implementing the context manager protocol for functions
that read files."""
# @wraps(func)
class CManager:
def __init__(self, source, *args, **kwargs):
self.source = source
self.args = args
self.kwargs = kwargs
self.close = kwargs.get('close', True)
def __enter__(self):
# _file_obj is a little helper that opens the file for reading
self.fsource = _file_obj(self.source, 'r')
return func(self.fsource, *self.args, **self.kwargs)
def __exit__(self, exc_type, exc_value, traceback):
if self.close:
self.fsource.close()
return False
return CManager
取消注释 wraps
行时出现的错误发生在 update_wrapper
中:
/usr/lib/python3.3/functools.py in update_wrapper(wrapper, wrapped, assigned, updated)
54 setattr(wrapper, attr, value)
55 for attr in updated:
---> 56 getattr(wrapper, attr).update(getattr(wrapped, attr, {}))
57 # Return the wrapper so this can be used as a decorator via partial()
58 return wrapper
AttributeError: 'mappingproxy' object has no attribute 'update'
我知道文档没有说我什至可以使用 functools.wraps
用这样的类包装一个函数,但话又说回来,它只适用于 Python 2。有人可以解释一下这个回溯到底告诉我什么以及我应该怎么做才能实现 wraps
的效果在两个版本的 Python 上?
编辑:我错了。上面的代码没有做我想做的事。我希望能够同时使用 with 和 without with
功能,就像内置的 open
一样。
上面的代码将装饰函数变成了上下文管理器。我希望能够做到:
reader = func('source.txt', arg)
for item in reader:
pass
还有
with func('source.txt', arg) as reader:
for item in reader:
pass
所以我的代码版本应该大概如下所示:
def _file_reader(func):
"""A decorator implementing the context manager protocol for functions
that read files."""
@wraps(func)
class CManager:
def __init__(self, source, *args, **kwargs):
self.close = kwargs.get('close', True)
self.fsource = _file_obj(source, 'r')
self.reader = func(self.fsource, *args, **kwargs)
def __enter__(self):
return self.reader
def __iter__(self):
return self.reader
def __next__(self):
return next(self.reader)
def __exit__(self, exc_type, exc_value, traceback):
if self.close and not self.fsource.closed:
self.fsource.close()
return False
return CManager
请随时评论我忽略的任何内容。
注意:J.F. Sebastian 的类版本似乎可以工作:
我基本上从类中删除了 wraps
并将 return CManager
更改为:
@wraps(func)
def helper(*args, **kwargs):
return CManager(*args, **kwargs)
return helper
最佳答案
functools.wraps()
用于包装函数:
import contextlib
import functools
def file_reader(func):
@functools.wraps(func)
@contextlib.contextmanager
def wrapper(file, *args, **kwargs):
close = kwargs.pop('close', True) # remove `close` argument if present
f = open(file)
try:
yield func(f, *args, **kwargs)
finally:
if close:
f.close()
return wrapper
@file_reader
def f(file):
print(repr(file.read(10)))
return file
with f('prog.py') as file:
print(repr(file.read(10)))
如果您想使用基于类的上下文管理器,那么解决方法是:
def file_reader(func):
@functools.wraps(func)
def helper(*args, **kwds):
return File(func, *args, **kwds)
return helper
无论是直接使用装饰函数还是作为上下文管理器,要使其表现相同,您应该在 __enter__()
中返回 self
:
import sys
class File(object):
def __init__(self, file, func, *args, **kwargs):
self.close_file = kwargs.pop('close', True)
# accept either filename or file-like object
self.file = file if hasattr(file, 'read') else open(file)
try:
# func is responsible for self.file if it doesn't return it
self.file = func(self.file, *args, **kwargs)
except: # clean up on any error
self.__exit__(*sys.exc_info())
raise
# context manager support
def __enter__(self):
return self
def __exit__(self, *args, **kwargs):
if not self.close_file:
return # do nothing
# clean up
exit = getattr(self.file, '__exit__', None)
if exit is not None:
return exit(*args, **kwargs)
else:
exit = getattr(self.file, 'close', None)
if exit is not None:
exit()
# iterator support
def __iter__(self):
return self
def __next__(self):
return next(self.file)
next = __next__ # Python 2 support
# delegate everything else to file object
def __getattr__(self, attr):
return getattr(self.file, attr)
file = f('prog.py') # use as ordinary function
print(repr(file.read(20)))
file.seek(0)
for line in file:
print(repr(line))
break
file.close()
关于python - functools.wraps 不允许我用 Python 3 中的类包装函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/14095025/
我有一个 ASP.NET 网站,我希望只允许 AD 组中的用户访问该网站。我正在使用如下的 web.config 片段,但这似乎不起作用:
仅当选中所有框时才应禁用“允许”按钮。我该怎么做?我已经完成了 HTML 部分,如下所示。如何执行其中的逻辑部分?即使未选中一个复选框,也应禁用“允许”按钮
当前有一个Navigator.push(context,route),但是上下文部分返回了错误,在尝试调试后,我发现问题是因为我在调用一个函数而不是直接将home设置为widget树。但是现在我不确定
这是我的邮政编码正则表达式 ^[a-zA-Z0-9]{1,9}$ 但不允许 A-12345。如何更改 - 也将被允许的正则表达式? 最佳答案 在字符集的开头或结尾添加-([...]): ^[-a-zA
我目前正在建立我的网站,但遇到了一个问题 JavaScript 中的混合内容阻止 当我尝试加载和显示来自 的图像和页面时,Chrome、Mozilla 和 Explorer 会发生这种情况http 我
我见过使用: [mysqld] bind-address = 255.112.324.12 允许远程访问单个 IP。我如何允许从 mysql 远程访问所有 IP? 最佳答案 如果你想允许它用于所
我想知道是否可以使用模板实现某些功能。我想要做的是允许特定的“复制构造函数和赋值运算符”从一个模板到另一个模板并禁用其他模板。 我想我只完成了一件我想要的事情,所以我提供了下面的类(class)。对于
这个问题在这里已经有了答案: How to validate an email address in PHP (15 个答案) 关闭 2 年前。 正则表达式让我大吃一惊。我如何更改此设置以验证带有加
解析可以采用以下格式之一的日期的最佳方法是什么 "dd-MM-yyyy HH:mm" "dd/MM/yyyy HH:mm" "dd.MM.yyyy HH:mm" 无需创建 3 个 SimpleD
我们知道,下面的代码格式不正确,因为成员 x 在依赖的基类中。但是,将指定行上的 x 更改为 this->x 将修复错误。 template struct B { int x; }; tem
如果能帮助我理解“Java 并发实践”中的以下内容,我将不胜感激: Calling an overrideable instance method(one that is neither privat
此时如果上传一个不在预定义的安全扩展名列表,如.lrc,会报错: File type does not meet security guidelines. Try another. 解决此问题有
我有一个运行韵律,可以为我的几个域和一个 friend 域处理 XMPP。我 friend 域中的一位用户(他的妻子)想更改她的密码(实际上她忘记了她,所以我会用 prosodyctl 设置一个,然后
使用 nginx,您可以允许和拒绝范围和 ips (https://www.nginx.com/resources/admin-guide/restricting-access/)。使用realip模
什么是一些好的克里金法/插值想法/选项,可以让重度权重的点在绘制的 R map 上的轻权重点上流血? 康涅狄格州有八个县。我找到了质心并想绘制这八个县中每个县的贫困率。其中三个县人口稠密(约 100
我正在使用 virtualbox + ubuntu + vagrant . 但是我不能ping或 wget任何网址。请指导我如何允许虚拟机访问我的主机的互联网? 最佳答案 这对我有用。 使用此配置 V
标题可能有点令人困惑,所以让我向您解释一下。 在 Swift 中,我们可以拥有带有默认参数值的函数,例如: func foo(value: Int = 32) { } 我们也可以有 In-Out 参数
有TextView1 和TextView2。 TextView2 应该 float 在 TextView1 的右侧。只要两个 TextView 的总宽度不使 TextView2 与右侧的框重叠,Tex
使用 Magento 收集方法 addFieldToFilter 时是否可以允许按 NULL 值进行过滤?我想选择集合中具有自定义属性的所有产品,即使没有为该属性分配任何值。 最佳答案 您不需要使用
我正试图从 .htaccess 文件中的规则中“排除”一个目录(及其所有文件夹)... 不确定这是否可能? .htaccess 文件是这样的: Order Allow,Deny Deny from a
我是一名优秀的程序员,十分优秀!