Как обрабатывать конструкторы или методы с другим набором (или типом) аргументов в Python?

Question

Есть ли способ в Python иметь более одного конструктора или более одного метода с одинаковым именем , которые отличаются количеством аргументов , которые они принимают или тип (ы) одного или нескольких аргументов ?

Если нет, то как лучше всего справляться с такими ситуациями?

Для примера я составил цветовой класс. Этот класс должен работать только как базовый пример для обсуждения этого , там много ненужных и / или избыточных вещей.

Было бы неплохо, если бы я мог вызывать конструктор с другими объектами (списком, другим объектом цвета или тремя целыми числами ...), и конструктор обрабатывает их соответствующим образом. В этом базовом примере в некоторых случаях он работает с * args и * * kwargs, но использование методов класса - это единственный общий способ, который я придумал. Каким было бы « передовой опыт » как решение для этого?

За исключением конструктора, если я хотел бы реализовать метод _ _ add _ _, как я могу заставить этот метод принимать все это: простое целое число (которое добавляется ко всем значениям ), три целых числа (где первое добавляется к красному значению и т. д.) или другой цветовой объект (где оба красных значения добавляются вместе и т. д.)?

EDIT

• Я добавил альтернативный конструктор (инициализатор, _ _ init _ _), который в основном выполняет все, что я хотел.

• Но я придерживаюсь первого и заводских методов. Кажется, яснее.

• Я также добавил _ _ add _ _, который выполняет все вышеперечисленное, но я не уверен, что это хороший стиль . Я пытаюсь использовать протокол итерации и возвращаюсь к «режиму с одним значением» вместо проверки конкретных типов. Может быть, все еще уродливый.

• Я взглянул на _ _ new _ _, спасибо за ссылки.

• Моя первая быстрая попытка с ним не удалась: я отфильтровываю значения rgb из * args и * * kwargs (это класс, список и т. Д.), Затем вызываю суперкласс _ _ new _ _ с помощью правые аргументы (просто r, g, b), чтобы передать его init. Вызов 'Super (cls, self) ._ _ new _ _ (....)' работает, но поскольку я генерирую и возвращаю тот же объект, что и тот, из которого я вызываю (как и предполагалось), все исходные аргументы получить _ _ init _ _ (работает, как задумано), поэтому он освобождает от обязательств.

• Я мог бы полностью избавиться от _ _ init _ _ и установить значения в _ _ new _ _, но я не знаю ... мне кажется, что я здесь злоупотребляю ;-) Я Я должен хорошо взглянуть на метаклассы и новые сначала, я думаю.

Источник:

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

class Color (object):

  # It's 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's iterable.
  # Maybe that's 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)>

Answer 1

У вас могут быть заводские методы, это нормально. Но почему бы просто не назвать это как есть?

Color(r, g, b)
Color(*[r, g, b])
Color(**{'r': r, 'g': g, 'b': b})

Это путь на питоне. Что касается конструктора объекта from, я бы предпочел что-то вроде:

Color(*Color2.as_list())

Явное лучше, чем неявное - Python Zen

Answer 2

В общем, используйте фабричные методы, помеченные как @classmethod с. Они также будут корректно работать с подклассами. С точки зрения дизайна они более явные, особенно когда им дается хорошее имя.

В этом случае, вероятно, удобнее смешивать все вместе, но это также усложняет контракт для вашего конструктора.

Answer 3

Python не принимает несколько методов с одним и тем же именем, точка. Один метод делает одно.

Я видел разные подходы, рекомендованные для того, как обрабатывать эти ... методы класса (как вы обрисовали в общих чертах выше) или фабричные функции. Мне больше всего нравятся аргументы ключевых слов.

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, а также позволяет легко добавлять новые аргументы, если это необходимо.

РЕДАКТИРОВАТЬ: Плюс мне не нужно смотреть на фактический код конструктора, чтобы понять аргументы. Объяснение снабжено ключевым словом.

Answer 4

По вопросу __add__:

Во-первых, вы не можете получить "три целых числа", я предполагаю, что вы имеете в виду три кортежа целых чисел?

В этом случае вы не сможете обойти некоторые 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__ для поддержки +=.

Answer 5

Мой первый совет - использовать заводские методы.

Хотя, если вы действительно хотите использовать единственный метод, дайте ему что-то для обработка параметров.

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)

Хотя при этом добавляется еще один параметр, он все же намного лучше, чем тесты типов, и сохраняет данные в явном виде. Он предоставляет приятные исключения из коробки при незаконных вызовах и легко расширяется даже в подклассах.

Answer 6

Python всегда полностью заменяет методы с одинаковыми именами. В отличие от C # это, если я правильно помню, сделает методы с одинаковыми именами для ввода разных аргументов.

Если есть только один вариант в ключевых словах, например, 3 или 4 аргумента одного и того же типа, я бы сказал, что можно использовать предварительную настройку последнего аргумента или все из них.

Однако, если вам нужны списки, кортежи и другие типы, вам, вероятно, следует перейти к списку произвольных аргументов и проверить его содержимое в функции

def function(*args):
    if type(args[0]) is int:
        dothis()
    #and so on

Answer 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

Может быть, это поможет.