¿Cómo sobrecargar el método __init__ basado en el tipo de argumento?

Question

Digamos que tengo una clase que tiene un miembro llamado datos que es una lista.

Quiero poder inicializar la clase con, por ejemplo, un nombre de archivo (que contiene datos para inicializar la lista) o con una lista real.

¿Cuál es tu técnica para hacer esto?

¿Simplemente verifica el tipo mirando __class__ ?

¿Hay algún truco que pueda faltar?

Estoy acostumbrado a C ++ donde la sobrecarga por tipo de argumento es fácil.

Answer 1

Una forma mucho mejor de obtener 'constructores alternativos' es usar métodos de clase. Por ejemplo:

>>> class MyData:
...     def __init__(self, data):
...         "Initialize MyData from a sequence"
...         self.data = data
...     
...     @classmethod
...     def fromfilename(cls, filename):
...         "Initialize MyData from a file"
...         data = open(filename).readlines()
...         return cls(data)
...     
...     @classmethod
...     def fromdict(cls, datadict):
...         "Initialize MyData from a dict's items"
...         return cls(datadict.items())
... 
>>> MyData([1, 2, 3]).data
[1, 2, 3]
>>> MyData.fromfilename("/tmp/foobar").data
['foo\n', 'bar\n', 'baz\n']
>>> MyData.fromdict({"spam": "ham"}).data
[('spam', 'ham')]

La razón por la que está más ordenada es que no hay dudas sobre qué tipo de fuente se espera, y no estás obligado a adivinar qué pretendía hacer la persona que llama con el tipo de datos que te proporcionó. El problema con isinstance (x, basestring) es que no hay forma de que la persona que llama le diga, por ejemplo, que aunque el tipo no sea una cadena de base, debe tratarlo como una cadena ( y no otra secuencia.) Y quizás la persona que llama quisiera usar el mismo tipo para diferentes propósitos, algunas veces como un solo elemento y otras como una secuencia de elementos. Ser explícito elimina todas las dudas y conduce a un código más sólido y claro.

Answer 2

Excelente pregunta. También he abordado este problema, y ??aunque estoy de acuerdo con que " fábricas " (los constructores de métodos de clase) son un buen método, me gustaría sugerir otro, que también me parece muy útil:

Aquí hay una muestra (este es un método de read y no un constructor, pero la idea es la misma):

def read(self, str=None, filename=None, addr=0):
    """ Read binary data and return a store object. The data
        store is also saved in the interal 'data' attribute.

        The data can either be taken from a string (str 
        argument) or a file (provide a filename, which will 
        be read in binary mode). If both are provided, the str 
        will be used. If neither is provided, an ArgumentError 
        is raised.
    """
    if str is None:
        if filename is None:
            raise ArgumentError('Please supply a string or a filename')

        file = open(filename, 'rb')
        str = file.read()
        file.close()
    ...
    ... # rest of code

La idea clave aquí es usar el excelente soporte de Python para los argumentos con nombre para implementar esto. Ahora, si quiero leer los datos de un archivo, digo:

obj.read(filename="blob.txt")

Y para leerlo de una cadena, digo:

obj.read(str="\x34\x55")

De esta manera, el usuario tiene un solo método para llamar. Manejarlo dentro, como viste, no es demasiado complejo

Answer 3

Solución rápida y sucia

class MyData:
    def __init__(string=None,list=None):
        if string is not None:
            #do stuff
        elif list is not None:
            #do other stuff
        else:
            #make data empty

Entonces puedes llamarlo con

MyData(astring)
MyData(None, alist)
MyData()

Answer 4

Una mejor manera sería usar isinstance y la conversión de tipos. Si te estoy entendiendo bien, quieres esto:

def __init__ (self, filename):
    if isinstance (filename, basestring):
        # filename is a string
    else:
        # try to convert to a list
        self.path = list (filename)

Answer 5

con python3, puede usar Implementing Multiple Enviar con anotaciones de funciones como escribió Python Cookbook:

import time


class Date(metaclass=MultipleMeta):
    def __init__(self, year:int, month:int, day:int):
        self.year = year
        self.month = month
        self.day = day

    def __init__(self):
        t = time.localtime()
        self.__init__(t.tm_year, t.tm_mon, t.tm_mday)

y funciona como:

>>> d = Date(2012, 12, 21)
>>> d.year
2012
>>> e = Date()
>>> e.year
2018

Answer 6

Debes usar isinstance

isinstance(...)
    isinstance(object, class-or-type-or-tuple) -> bool

    Return whether an object is an instance of a class or of a subclass thereof.
    With a type as second argument, return whether that is the object's type.
    The form using a tuple, isinstance(x, (A, B, ...)), is a shortcut for
    isinstance(x, A) or isinstance(x, B) or ... (etc.).

Answer 7

Probablemente desee la función incorporada isinstance :

self.data = data if isinstance(data, list) else self.parse(data)

Answer 8

Mi solución preferida es:

class MyClass:
    _data = []
    __init__(self,data=None):
        # do init stuff
        if not data: return
        self._data = list(data) # list() copies the list, instead of pointing to it.

Luego invócalo con MyClass () o MyClass ([1,2,3]) .

Espero que ayude. ¡Feliz codificación!

Answer 9

OK, genial. Acabo de juntar este ejemplo con una tupla, no con un nombre de archivo, pero eso es fácil. Gracias a todos.

class MyData:
    def __init__(self, data):
        self.myList = []
        if isinstance(data, tuple):
            for i in data:
                self.myList.append(i)
        else:
            self.myList = data

    def GetData(self):
        print self.myList

a = [1,2]

b = (2,3)

c = MyData (a)

d = MyData (b)

c.GetData ()

d.GetData ()

[1, 2]

[2, 3]

Answer 10

¿Por qué no vas más pitón?

class AutoList:
def __init__(self, inp):
    try:                        ## Assume an opened-file...
        self.data = inp.read()
    except AttributeError:
        try:                    ## Assume an existent filename...
            with open(inp, 'r') as fd:
                self.data = fd.read()
        except:
            self.data = inp     ## Who cares what that might be?