我有以下代码段:
FEED_TYPES = [
('fan_mail', 'Fan Mail'),
('review', 'Review'),
('tip', 'Tip'),
('fan_user', 'Fan User'),
('fan_song', 'Fan Song'),
('fan_album', 'Fan Album'),
('played_song', 'Played Song'),
('played_album', 'Played Album'),
('played_radio', 'Played Radio'),
('new_event', 'New Event'),
]
class Feed:
@classmethod
def do_create(cls, **kwargs):
print kwargs
@classmethod
def create(cls, type, **kwargs):
kwargs['feed_type'] = type
cls.do_create(**kwargs)
for type_tuple in FEED_TYPES:
type, name = type_tuple
def notify(self, **kwargs):
print "notifying %s" % type
self.create(type, **kwargs)
notify.__name__ = "notify_%s" % type
setattr(Feed, notify.__name__, classmethod(notify))
Feed.create("FanMail", to_profile="Gerson", from_profile="Felipe")
Feed.notify_fan_mail(to_profile="Gerson2", from_profile="Felipe2")
我们的想法是为每种Feed类型动态创建一个类方法(例如 notify_fan_mail )。它工作得非常好,唯一的问题是 print 语句总是打印“通知new_event”,无论我调用哪种方法(对于 notify_new_mail , notify_review <相同) / strong>等。)。
我意识到这是因为它使用分配给键入的最后一个值。我的问题是:如何动态创建使用类型的正确值的方法?
另外,如果我在Python文件中有这个确切的代码,这是将方法添加到Feed类的正确方法,还是有更优雅的方式?
答案 0 :(得分:5)
使用闭包来保留kind
:
for type_tuple in FEED_TYPES:
kind, name = type_tuple
def make_notify(kind):
def notify(self, **kwargs):
print "notifying %s" % kind
self.create(kind, **kwargs)
return notify
notify = make_notify(kind)
notify.__name__ = "notify_%s" % kind
setattr(cls, notify.__name__, classmethod(notify))
顺便说一下,不要使用type
作为变量名,因为它会影响同名内置词。
修改Feed
的一种更优雅的方法是创建一个类装饰器。这使您更清楚地了解了修改Feed
的原始定义的代码。
FEED_TYPES = [
('fan_mail', 'Fan Mail'),
('review', 'Review'),
('tip', 'Tip'),
('fan_user', 'Fan User'),
('fan_song', 'Fan Song'),
('fan_album', 'Fan Album'),
('played_song', 'Played Song'),
('played_album', 'Played Album'),
('played_radio', 'Played Radio'),
('new_event', 'New Event'),
]
def add_feed_types(cls):
for type_tuple in FEED_TYPES:
kind, name = type_tuple
def make_notify(kind):
def notify(self, **kwargs):
print "notifying %s" % kind
self.create(kind, **kwargs)
return notify
notify = make_notify(kind)
notify.__name__ = "notify_%s" % kind
setattr(cls, notify.__name__, classmethod(notify))
return cls
@add_feed_types
class Feed:
@classmethod
def do_create(cls, **kwargs):
print kwargs
@classmethod
def create(cls, kind, **kwargs):
kwargs['feed_type'] = kind
cls.do_create(**kwargs)
Feed.create("FanMail", to_profile="Gerson", from_profile="Felipe")
Feed.notify_fan_mail(to_profile="Gerson2", from_profile="Felipe2")
答案 1 :(得分:1)
该错误是由Python中的闭包性质引起的。通知函数中的名称type
绑定到封闭范围中的type
。当您更改type
的值时,它会针对引用它的所有闭包发生更改。
解决此问题的一种方法是使用函数工厂:
def make_notify_function(type):
def notify(self, **kwargs):
print "notifying %s" % type
self.create(type, **kwargs)
return notify
答案 2 :(得分:1)
您遇到的问题是您的notify
函数未封装值type
,只是它的名称。所以当你的for循环进入下一个元组时,旧元素就会丢失。
您可以通过使type
成为函数的默认参数来解决此问题:
for type, name in FEED_TYPES: # no need to unpack the tuple separately
def notify(cls, type=type, **kwargs): # type is an argument and default value
print "notyfying %s" % type
cls.create(type, **kwargs)
...
请注意,我已将self
参数更改为cls
,这可能更正确,因为您将其设为类方法。
我认为这是在运行时向类添加方法的合适方法。我不确定这是否是您需要做的事情,但没有关于您的任务的更多信息(例如,do_create
做了什么?)我没有看到任何其他明显的改进。