Как избежать шаблона "self.x = x; self.y = y; self.z = z" в __init__?
Я вижу такие шаблоны, как
def __init__(self, x, y, z):
...
self.x = x
self.y = y
self.z = z
...
довольно часто, часто с гораздо большим количеством параметров. Есть ли хороший способ избежать такого утомительного повторения? Должен ли класс наследовать от namedtuple
вместо?
11 ответов
Решение для декоратора, которое сохраняет подпись:
import decorator
import inspect
import sys
@decorator.decorator
def simple_init(func, self, *args, **kws):
"""
@simple_init
def __init__(self,a,b,...,z)
dosomething()
behaves like
def __init__(self,a,b,...,z)
self.a = a
self.b = b
...
self.z = z
dosomething()
"""
#init_argumentnames_without_self = ['a','b',...,'z']
if sys.version_info.major == 2:
init_argumentnames_without_self = inspect.getargspec(func).args[1:]
else:
init_argumentnames_without_self = tuple(inspect.signature(func).parameters.keys())[1:]
positional_values = args
keyword_values_in_correct_order = tuple(kws[key] for key in init_argumentnames_without_self if key in kws)
attribute_values = positional_values + keyword_values_in_correct_order
for attribute_name,attribute_value in zip(init_argumentnames_without_self,attribute_values):
setattr(self,attribute_name,attribute_value)
# call the original __init__
func(self, *args, **kws)
class Test():
@simple_init
def __init__(self,a,b,c,d=4):
print(self.a,self.b,self.c,self.d)
#prints 1 3 2 4
t = Test(1,c=2,b=3)
#keeps signature
#prints ['self', 'a', 'b', 'c', 'd']
if sys.version_info.major == 2:
print(inspect.getargspec(Test.__init__).args)
else:
print(inspect.signature(Test.__init__))
Отказ от ответственности: Кажется, что несколько человек обеспокоены представлением этого решения, поэтому я предоставлю очень четкую оговорку. Вы не должны использовать это решение. Я предоставляю это только в качестве информации, так что вы знаете, что язык способен на это. Остальная часть ответа просто показывает языковые возможности, а не одобряет их использование таким образом.
Нет ничего плохого в том, чтобы явно копировать параметры в атрибуты. Если в ctor слишком много параметров, это иногда считается запахом кода, и, возможно, вам следует сгруппировать эти параметры в меньшее количество объектов. В других случаях это необходимо, и в этом нет ничего плохого. В любом случае, делать это явно - путь.
Однако, поскольку вы спрашиваете, КАК это можно сделать (а не нужно ли это делать), то одно из решений заключается в следующем:
class A:
def __init__(self, **kwargs):
for key in kwargs:
setattr(self, key, kwargs[key])
a = A(l=1, d=2)
a.l # will return 1
a.d # will return 2
Явное лучше, чем неявное... так что вы можете сделать его более кратким:
def __init__(self,a,b,c):
for k,v in locals().items():
if k != "self":
setattr(self,k,v)
Лучший вопрос, должен ли ты?
... при этом если вы хотите именованный кортеж, я бы порекомендовал использовать именованный кортеж (помните, что у кортежей есть определенные условия): возможно, вы хотите упорядоченный приговор или даже просто диктовку...
Как уже упоминалось, повторение неплохое, но в некоторых случаях именованный кортеж отлично подходит для такого типа проблем. Это позволяет избежать использования locals() или kwargs, что обычно является плохой идеей.
from collections import namedtuple
# declare a new object type with three properties; x y z
# the first arg of namedtuple is a typename
# the second arg is comma-separated or space-separated property names
XYZ = namedtuple("XYZ", "x, y, z")
# create an object of type XYZ. properties are in order
abc = XYZ("one", "two", 3)
print abc.x
print abc.y
print abc.z
Я нашел ограниченное использование для него, но вы можете наследовать именованный кортеж как с любым другим объектом (пример продолжается):
class MySuperXYZ(XYZ):
""" I add a helper function which returns the original properties """
def properties(self):
return self.x, self.y, self.z
abc2 = MySuperXYZ(4, "five", "six")
print abc2.x
print abc2.y
print abc2.z
print abc2.properties()
Расширить на gruszczy
s ответ, я использовал шаблон как:
class X:
x = None
y = None
z = None
def __init__(self, **kwargs):
for (k, v) in kwargs.items():
if hasattr(self, k):
setattr(self, k, v)
else:
raise TypeError('Unknown keyword argument: {:s}'.format(k))
Мне нравится этот метод, потому что это:
- избегает повторения
- устойчив к опечаткам при строительстве объекта
- хорошо работает с подклассами (может просто
super().__init(...)
) - позволяет документировать атрибуты на уровне класса (где они принадлежат), а не в
X.__init__
До Python 3.6 это не давало контроля над порядком, в котором установлены атрибуты, что может быть проблемой, если некоторые атрибуты являются свойствами с установщиками, которые обращаются к другим атрибутам.
Возможно, это можно немного улучшить, но я единственный пользователь своего собственного кода, поэтому я не беспокоюсь о какой-либо форме санитарии ввода. Возможно AttributeError
было бы более уместным.
Вы также можете сделать:
locs = locals()
for arg in inspect.getargspec(self.__init__)[0][1:]:
setattr(self, arg, locs[arg])
Конечно, вам придется импортировать inspect
модуль.
Это решение без какого-либо дополнительного импорта.
Вспомогательная функция
Небольшая вспомогательная функция делает ее более удобной и многократно используемой:
def auto_init(local_name_space):
"""Set instance attributes from arguments.
"""
self = local_name_space.pop('self')
for name, value in local_name_space.items():
setattr(self, name, value)
заявка
Вам нужно позвонить с locals()
:
class A:
def __init__(self, x, y, z):
auto_init(locals())
Тестовое задание
a = A(1, 2, 3)
print(a.__dict__)
Выход:
{'y': 2, 'z': 3, 'x': 1}
Без изменения locals()
Если вы не любите менять locals()
используйте эту версию:
def auto_init(local_name_space):
"""Set instance attributes from arguments.
"""
for name, value in local_name_space.items():
if name != 'self':
setattr(local_name_space['self'], name, value)
Python 3.7 и выше
В Python 3.7 вы можете (ab) использовать dataclass
декоратор, доступный от dataclasses
модуль. Из документации:
Этот модуль предоставляет декоратор и функции для автоматического добавления сгенерированных специальных методов, таких как
__init__()
а также__repr__()
для пользовательских классов. Первоначально он был описан в PEP 557.Переменные-члены, используемые в этих сгенерированных методах, определяются с помощью аннотаций типа PEP 526. Например этот код:
@dataclass class InventoryItem: '''Class for keeping track of an item in inventory.''' name: str unit_price: float quantity_on_hand: int = 0 def total_cost(self) -> float: return self.unit_price * self.quantity_on_hand
Добавит, между прочим,
__init__()
это выглядит так:def __init__(self, name: str, unit_price: float, quantity_on_hand: int=0): self.name = name self.unit_price = unit_price self.quantity_on_hand = quantity_on_hand
Обратите внимание, что этот метод автоматически добавляется в класс: он не указан напрямую в приведенном выше определении InventoryItem.
Если ваш класс большой и сложный, может быть нецелесообразно использовать dataclass
, Я пишу это в день выпуска Python 3.7.0, поэтому шаблоны использования еще не установлены.
Интересная библиотека, которая обрабатывает это (и избегает многих других шаблонов), является attrs. Ваш пример, например, может быть сведен к этому (предположим, что класс называется MyClass
):
import attr
@attr.s
class MyClass:
x = attr.ib()
y = attr.ib()
z = attr.ib()
Вам даже не нужно __init__
метод больше, если он не делает другие вещи, а также. Вот хорошее введение от Глифа Лефковица.
Мой 0,02$. Это очень близко к ответу Джорана Бизли, но более элегантно:
def __init__(self, a, b, c, d, e, f):
vars(self).update((k, v) for k, v in locals().items() if v is not self)
Кроме того, ответ Майка Мюллера (лучший на мой вкус) можно уменьшить с помощью этой техники:
def auto_init(ns):
self = ns.pop('self')
vars(self).update(ns)
И просто позвони auto_init(locals())
от твоего __init__
Это естественный способ делать вещи в Python. Не пытайтесь придумать что-то более умное, это приведет к чрезмерно умному коду, который никто в вашей команде не поймет. Если вы хотите быть командным игроком, а затем продолжайте писать это так.