Как обрабатывать конструкторы или методы с другим набором (или типом) аргументов в Python?
Есть ли способ в Python, иметь более одного конструктора или более одного метода с тем же именем, которые отличаются количеством аргументов, которые они принимают, или типа (ов) одного или нескольких аргументов?
Если нет, как лучше всего справиться с такими ситуациями?
В качестве примера я составил класс цветов. Этот класс должен работать только в качестве базового примера, чтобы обсудить это, там много ненужного и/или избыточного материала.
Было бы неплохо, если бы я мог вызвать конструктор с разными объектами (список, другой цветной объект или три целых числа...), и конструктор обрабатывает их соответственно. В этом базовом примере он работает в некоторых случаях с * args и * * kwargs, но использование методов класса - единственный общий способ, с которым я столкнулся. Что было бы " лучшей практикой" как решение для этого?
Конструктор в стороне, если я хотел бы реализовать метод _ _ добавить _ _ тоже, как я могу заставить этот метод принять все это: простое целое (которое добавляется ко всем значения), три целых числа (где первый добавляется к красному значению и т.д.) или другому цветному объекту (где оба красных значения добавляются вместе и т.д.)?
ИЗМЕНИТЬ
-
Я добавил альтернативный конструктор (initializer, _ _ init _ _), который в основном делает все, что я хотел.
-
Но я придерживаюсь первого и factory методов. Кажется яснее.
-
Я также добавил _ _ добавить _ _, что делает все, о чем говорилось выше, но я не уверен, что это хороший стиль. Я пытаюсь использовать протокол итерации и возвращаться к "режиму одиночного значения" вместо проверки определенных типов. Может быть, все еще уродливый.
-
Я взглянул на _ _ новый _ _, спасибо за ссылки.
-
Моя первая попытка скомпрометировать: я фильтрую значения rgb из * args и * * kwargs (это класс, список и т.д.), затем вызывайте суперкласс _ _ new _ _ с помощью right args (просто r, g, b), чтобы передать его в init. Вызов "Супер (cls, self)._ _ new _ _ (....)" работает, но так как я генерирую и возвращаю тот же объект, что и тот, с которого я звоню (по желанию), все исходные аргументы переходите к _ _ init _ _ (работайте по назначению), поэтому он освобождается.
-
Я мог бы полностью избавиться от _ _ init _ _ и установить значения в _ _ new _ _, но я не знаю... чувствует, что я злоупотребляю здесь;-) я должен хорошо взглянуть на метаклассы и новые, как я предполагаю.
Источник:
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
class Color (object):
# It strict on what to accept, but I kinda like it that way.
def __init__(self, r=0, g=0, b=0):
self.r = r
self.g = g
self.b = b
# Maybe this would be a better __init__?
# The first may be more clear but this could handle way more cases...
# I like the first more though. What do you think?
#
#def __init__(self, obj):
# self.r, self.g, self.b = list(obj)[:3]
# This methods allows to use lists longer than 3 items (eg. rgba), where
# 'Color(*alist)' would bail
@classmethod
def from_List(cls, alist):
r, g, b = alist[:3]
return cls(r, g, b)
# So we could use dicts with more keys than rgb keys, where
# 'Color(**adict)' would bail
@classmethod
def from_Dict(cls, adict):
return cls(adict['r'], adict['g'], adict['b'])
# This should theoreticaly work with every object that iterable.
# Maybe that more intuitive duck typing than to rely on an object
# to have an as_List() methode or similar.
@classmethod
def from_Object(cls, obj):
return cls.from_List(list(obj))
def __str__(self):
return "<Color(%s, %s, %s)>" % (self.r, self.g, self.b)
def _set_rgb(self, r, g, b):
self.r = r
self.g = g
self.b = b
def _get_rgb(self):
return (self.r, self.g, self.b)
rgb = property(_get_rgb, _set_rgb)
def as_List(self):
return [self.r, self.g, self.b]
def __iter__(self):
return (c for c in (self.r, self.g, self.b))
# We could add a single value (to all colorvalues) or a list of three
# (or more) values (from any object supporting the iterator protocoll)
# one for each colorvalue
def __add__(self, obj):
r, g, b = self.r, self.g, self.b
try:
ra, ga, ba = list(obj)[:3]
except TypeError:
ra = ga = ba = obj
r += ra
g += ga
b += ba
return Color(*Color.check_rgb(r, g, b))
@staticmethod
def check_rgb(*vals):
ret = []
for c in vals:
c = int(c)
c = min(c, 255)
c = max(c, 0)
ret.append(c)
return ret
class ColorAlpha(Color):
def __init__(self, r=0, g=0, b=0, alpha=255):
Color.__init__(self, r, g, b)
self.alpha = alpha
def __str__(self):
return "<Color(%s, %s, %s, %s)>" % (self.r, self.g, self.b, self.alpha)
# ...
if __name__ == '__main__':
l = (220, 0, 70)
la = (57, 58, 61, 255)
d = {'r': 220, 'g': 0, 'b':70}
da = {'r': 57, 'g': 58, 'b':61, 'a':255}
c = Color(); print c # <Color(0, 0, 0)>
ca = ColorAlpha(*la); print ca # <Color(57, 58, 61, 255)>
print '---'
c = Color(220, 0, 70); print c # <Color(220, 0, 70)>
c = Color(*l); print c # <Color(220, 0, 70)>
#c = Color(*la); print c # -> Fail
c = Color(**d); print c # <Color(220, 0, 70)>
#c = Color(**da); print c # -> Fail
print '---'
c = Color.from_Object(c); print c # <Color(220, 0, 70)>
c = Color.from_Object(ca); print c # <Color(57, 58, 61, 255)>
c = Color.from_List(l); print c # <Color(220, 0, 70)>
c = Color.from_List(la); print c # <Color(57, 58, 61, 255)>
c = Color.from_Dict(d); print c # <Color(220, 0, 70)>
c = Color.from_Dict(da); print c # <Color(57, 58, 61, 255)>
print '---'
print 'Check =', Color.check_rgb('1', 0x29a, -23, 40)
# Check = [1, 255, 0, 40]
print '%s + %s = %s' % (c, 10, c + 10)
# <Color(57, 58, 61)> + 10 = <Color(67, 68, 71)>
print '%s + %s = %s' % (c, ca, c + ca)
# <Color(57, 58, 61)> + <Color(57, 58, 61, 255)> = <Color(114, 116, 122)>
Ответы
Ответ 1
В общем, используйте методы factory, помеченные как @classmethod
s. Они также будут корректно работать в подклассах. С точки зрения дизайна они более ясны, особенно когда им дается доброе имя.
В этом случае смешивание всего вместе, вероятно, более удобно, но это также затрудняет контракт для вашего конструктора.
Ответ 2
У вас могут быть методы factory, это нормально. Но почему бы просто не назвать это так, как есть?
Color(r, g, b)
Color(*[r, g, b])
Color(**{'r': r, 'g': g, 'b': b})
Это путь python. Что касается конструктора от объекта, я бы предпочел бы что-то вроде:
Color(*Color2.as_list())
Явный лучше, чем неявный - Python Zen
Ответ 3
Python не принимает несколько методов с тем же именем, периодом. Один метод делает одно.
Я видел разные подходы, рекомендованные для того, как обращаться с этими... classmethods (как вы указали выше) или factory. Я больше всего использую ключевые слова.
class Color (object):
def __init__(self, **parms):
if parms.get('list'):
self.r, self.g, self.b = parms['list']
elif parms.get('color'):
color = parms['color']
self.r = color.r
self.g = color.g
self.b = color.b
else:
self.r = parms['red']
self.g = parms['green']
self.b = parms['blue']
c1 = Color(red=220, green=0, blue=270)
c2 = Color(list=[220, 0, 70])
c3 = Color(color=c1)
Это соответствует Python-способу быть явным и читаемым, а также легко позволяет добавлять новые аргументы, если это необходимо.
EDIT: Плюс я не должен смотреть на фактический код конструктора, чтобы понять аргументы. Объяснение предоставляется ключевым словом.
Ответ 4
В __add__
проблема:
Во-первых, вы не можете получить "три целых числа", я предполагаю, что вы имеете в виду 3-кортежи целых чисел?
В этом случае вы не сможете обойти несколько вызовов isinstance
:
def __add__(self, other):
if isinstance(other, Color):
...
elif isinstance(other, (int, long)):
...
elif len(other) == 3 and all(isinstance(e, (int, long)) for e in other):
...
else:
raise TypeError("Can only add Color to Color, int or three-tuple")
Вы также можете добавить реализации __radd__
, чтобы вы могли обрабатывать
1 + Color(1, 2, 3)
но это просто
def __radd__(self, other):
return self.__add__(other)
хотя строго, он никогда не будет вызываться, когда type(other) is Color
.
Кроме того, не забудьте __iadd__
поддержать +=
.
Ответ 5
Вначале советую использовать методы factory.
Если вам действительно нужен один метод, дайте ему что-то для отправки
обработки параметров.
def __init__(self, method, *args, **kw):
getattr(self, '_init_'+method)(*args, **kw)
def _init_coponents(self, r, g, b):
...
def _init_fromColor(self, color):
...
И используйте как:
c1 = Color('components', 0, 0, 0,)
c2 = Color('fromColor', c1)
Пока этот параметр добавляет еще один параметр, он по-прежнему лучше, чем тесты типа, и сохраняет явные данные. Он предоставляет прекрасные исключения из коробки при незаконных вызовах и легко расширяется даже в подклассах.
Ответ 6
Python всегда полностью заменяет методы с тем же именем. В отличие от С#, который, если я правильно помню, создаст методы с одинаковыми параметрами имени для ввода разных аргументов.
Если в ключевых словах есть только один вариант, например 3 или 4 аргумента одного и того же типа, я бы сказал, что использование пресета последнего аргумента или всех из них будет способом.
Однако, если вам нужны списки, кортежи и другие типы, вам, вероятно, следует перейти к списку произвольных аргументов и проверить содержимое этого в функции
def function(*args):
if type(args[0]) is int:
dothis()
#and so on
Ответ 7
Вы можете проверить тип аргумента, переданного внутри вашего конструктора:
def __init__(self, r = 0, g = 0, b = 0):
# if r is a list
if (type(r) == type([1,2,3])):
r, g, b = r[0], r[1], r[2]
# if r is a color
if (type(r) == type(self)):
r, g, b = r.r, r.g, r.b
self.r = r
self.g = g
self.b = b
Может быть, это поможет.