我正在创建一个装饰器,允许我做类似的事情:
@cooldownf(lambda self, **eargs: 30 - self.level)
def method(self, **eargs):
...
这将简单地修饰方法,使其具有冷却时间。
这一切都很好,现在可以每隔30 - self.level
秒执行一次方法。
但是,如果方法仍处于冷却状态,我想添加一条消息。
我为message
装饰器添加了cooldownf
参数,但我在装饰器代码中的UnboundLocalError: local variable 'message' referenced before assignment
行中收到错误ìf message:
:
def cooldownf(fn, message=None):
"""Decorates a method to have a dynamic cooldown.
Decorator function for easily adding cooldown as a dynamic time
(function) into skill's methods. The function gets called when the
cooldown is needed, and the skill is passed to the function.
Args:
fn: Function to determine the cooldown of the method
message: Optional message sent if there's still cooldown left
Returns:
Decorated method with a dynamic cooldown
"""
# Create a decorator using the function and message provided
def method_decorator(method):
# Create a wrapper method
@wraps(method, assigned=WRAPPER_ASSIGNMENTS+('__dict__',), updated=())
def method_wrapper(self, **eargs):
# If the method's cooldown is over
if method_wrapper.cooldown.remaining <= 0:
# Restart the cooldown
method_wrapper.cooldown.start(1, fn(self, **eargs))
# And call the function
return method(self, **eargs)
# If there was cooldown remaining and a message is provided
if message:
# Format the provided message
message = message.format(
name=self.name,
cd=method_wrapper.cooldown.remaining,
max_cd=method_wrapper.cooldown.limit
)
# Send it to the player
SayText2(message=message).send(eargs['player'].index)
# And exit with code 3
return 3
# Create the cooldown object for the wrapper
method_wrapper.cooldown = TickRepeat(lambda: None)
# And return the wrapper
return method_wrapper
# Return the decorator
return method_decorator
造成这种情况的原因是什么?
我可以在message
或cooldownf
内打印method_decorator
,但在method_wrapper
中添加打印件会导致错误提升。
这个确切的代码我无法在IDLE中使用函数复制它,是否与我使用方法特别相关?
答案 0 :(得分:2)
你在最里面的函数中将分配给method
:
message = message.format(
name=self.name,
cd=method_wrapper.cooldown.remaining,
max_cd=method_wrapper.cooldown.limit
)
该分配使其成为局部变量,但如果没有首先访问message
,则无法进行该分配。你不能用本地人做到这一点。
由于您不想修改已关闭的参数,因此您希望在此处使用 new 本地名称:
formatted_message = message.format(
name=self.name,
cd=method_wrapper.cooldown.remaining,
max_cd=method_wrapper.cooldown.limit
)
SayText2(message=formatted_message).send(eargs['player'].index)