我有以下代码:
In [38]: %paste
def set_session_attribute(obj, attribute):
if attribute.endswith('()'):
attribute = attribute.replace('()', '')
return getattr(obj, attribute)()
else:
return getattr(obj, attribute)
class Thing(object):
def __init__(self):
self.legs = 4
def length(self):
return 6
## -- End pasted text --
In [39]: x = Thing()
In [40]: y = set_session_attribute(x, 'legs')
In [41]: y
Out[41]: 4
In [42]: z = set_session_attribute(x, 'length()')
In [43]: z
Out[43]: 6
这是因为用"length()"
调用没有成功(AttributeError, no attribute length()
)
是否有更短、更易于维护的方法来实现这样的功能?谢谢。
最佳答案
解决方案1
您可以将length
设为property
:
class Thing(object):
def __init__(self):
self.legs = 4
@property
def length(self):
return 6
>>> thing = Thing()
>>> thing.legs
4
>>> thing.length
6
如果你真的想使用你的函数:
def set_session_attribute(obj, attribute):
return getattr(obj, attribute)
>>> set_session_attribute(thing, 'legs')
4
>>> set_session_attribute(thing, 'length')
6
如果你不能直接改变thing
的来源,你可以在导入类之后做:
class Thing(object):
def __init__(self):
self.legs = 4
def length(self):
return 6
这里:
Thing.length2 = property(Thing.length)
>>> thing = Thing()
>>> thing.length2
6
解决方案2
或者,您可以检查属性是否可调用:
class Thing(object):
def __init__(self):
self.legs = 4
def length(self):
return 6
def set_session_attribute(obj, attribute):
attr = getattr(obj, attribute)
if hasattr(attr, '__call__'): # may use `callable()`
return attr()
return attr
>> thing = Thing()
>>> set_session_attribute(thing, 'legs')
4
>>> set_session_attribute(thing, 'length')
6
关于python - 使用一个函数来获取函数或常规属性的属性,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36411611/