Python __init__ setattr для аргументов?
-
07-07-2019 - |
Вопрос
Кажется , что часто __init__
методы аналогичны этому:
def __init__(self, ivar1, ivar2, ivar3):
self.ivar1 = ivar1
self.ivar2 = ivar2
self.ivar3 = ivar3
Есть ли какой-нибудь способ превратить аргументы в список (не прибегая к *args
или **kwargs
) , а затем с помощью setattr
установить переменные экземпляра с переданным именем параметра и аргументом?И, возможно, нарезать список, напримервам нужно было бы, по крайней мере, нарезать его на [1:]
потому что ты не хочешь self.self
.
(на самом деле, я предполагаю, что это должен быть словарь для хранения имени и значения)
вот так:
def __init__(self, ivar1, ivar2, ivar3, optional=False):
for k, v in makedict(self.__class__.__init__.__args__): # made up __args__
setattr(self, k, v)
Спасибо!
Отвечая на ответ Неизвестного, я обнаружил, что это работает:
Class A(object):
def __init__(self, length, width, x):
self.__dict__.update(dict([(k, v) for k, v in locals().iteritems() if k != 'self']))
или
Class A(object):
def __init__(self, length, width, x):
self.__dict__.update(locals())
del self.__dict__['self']
Не так уж плохо..
Решение
Держи.Да, это уродливый злой взлом.Да, объекту нужна переменная __dict__ .Но эй, это аккуратный маленький лайнер one liner!
def __init__(self):
self.__dict__.update(locals())
Конструктор может принимать аргументы любого типа.
class test(object):
def __init__(self, a, b, foo, bar=5)...
a = test(1,2,3)
dir(a)
['__class__', '__delattr__', '__dict__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__module__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', '__weakref__', 'a', 'b', 'foo', 'bar', 'self']
Он также будет включать self, но вы можете легко удалить это или создать свою собственную функцию обновления, которая игнорирует self.
Другие советы
Вы могли бы использовать inspect.getargspec и инкапсулировать его как декоратор.Поиск необязательных аргументов и ключевых слов немного сложен, но это должно сработать:
def inits_args(func):
"""Initializes object attributes by the initializer signature"""
argspec = inspect.getargspec(func)
argnames = argspec.args[1:]
defaults = dict(zip(argnames[-len(argspec.defaults):], argspec.defaults))
@functools.wraps(func)
def __init__(self, *args, **kwargs):
args_it = iter(args)
for key in argnames:
if key in kwargs:
value = kwargs[key]
else:
try:
value = args_it.next()
except StopIteration:
value = defaults[key]
setattr(self, key, value)
func(self, *args, **kwargs)
return __init__
Затем вы можете использовать его следующим образом:
class Foo(object):
@inits_args
def __init__(self, spam, eggs=4, ham=5):
print "Foo(%r, %r, %r)" % (self.spam, self.eggs, self.ham)
Нет хорошего способа получить аргументы в виде списка, если они указаны по отдельности в сигнатуре функции.Вероятно, вы можете что-то сделать с помощью inspect или frame hacks, но это будет более уродливо, чем просто изложить это так, как вы это сделали.
Попробуй инспектировать.getargspec:
In [31]: inspect.getargspec(C.__init__)
Out[31]: ArgSpec(args=['self', 'ivar1', 'ivar2', 'ivar3', 'optional'],
varargs=None, keywords=None, defaults=(False,))
Посмотрите, может ли новый namedtuple (новый в Python 2.6) из модуля collections работать для вас.
Вы можете сделать это, используя самоанализ аргументов, но код будет длиннее, чем код, который вы пытаетесь заменить.Особенно, если вы работаете с квт, что вам, возможно, придется сделать.
Этот короткий код работает в большинстве случаев (улучшен по сравнению с Неизвестным примером):
>>> class Foo:
... def __init__(self, labamba, **kw):
... params = locals().copy()
... del params['self']
... if 'kw' in params:
... params.update(params['kw'])
... del params['kw']
... self.__dict__.update(params)
Но это уродливый взлом, делающий код менее читабельным без какой-либо конкретной причины, кроме лени, так что не делайте этого.А также, как часто вы действительно есть классы, которые имеют более 5-6 параметров инициализации?
Мне больше всего нравится эта форма, не слишком длинная и одновременно копируемая и подклассируемая:
class DynamicInitClass(object):
__init_defargs=('x',)
def __init__(self,*args,**attrs):
for idx,val in enumerate(args): attrs[self.__init_defargs[idx]]=val
for key,val in attrs.iteritems(): setattr(self,key,val)
Как насчет получения из специального класса?Я думаю, что это более явный и более гибкий вот так:
class InitMe:
def __init__(self, data):
if 'self' in data:
data = data.copy()
del data['self']
self.__dict__.update(data)
class MyClassA(InitMe):
def __init__(self, ivar1, ivar2, ivar3 = 'default value'):
super().__init__(locals())
class MyClassB(InitMe):
def __init__(self, foo):
super().__init__({'xxx': foo, 'yyy': foo, 'zzz': None})
# or super().__init__(dict(xxx=foo, yyy=foo, zzz=None))
class MyClassC(InitMe):
def __init__(self, foo, **keywords):
super().__init__(keywords)