Как реализовать перегрузку методов в python?

Question

[mkone112]

Как в питоне 3.6 реализовать подобное поведение?

from datetime import date

class Datepicker:
    def __init__(self, date: date) -> None:
        print('date object')
    def __init__(self, year, month, day) -> None:
        print('args')


Datepicker(date.today())  # date object
Datepicker(2020, 1, 1)  # args
help(Datepicker.__init__)
Help on wrapper_descriptor:

__init__(self, date: date) -> None
__init__(self, year, month, day) -> None

Answer 1

[mkone112]

from functools import singledispatch


class Test:
    def __init__(self):
        self.render = singledispatch(self.render)
        self.render.register(str, self._render_str)
        self.render.register(int, self._render_int)

    def render(self, s):
        raise TypeError(f"{type(s)} type isn't supported:")

    def _render_str(self, s, b=None):
        print("render str")

    def _render_int(self, s):
        print("render int")

t = Test()
>> t.render('s')
render str
>> t.render(1)
render int

p.s. конечно это можно реализовать с помощью метакласса - но как по мне это перебор
https://github.com/dabeaz/python-cookbook/blob/mas...

Answer 2

[shurshur]

В python нет перегрузки методов, вместо этого там можно принимать произвольные параметеры через *args и **kwargs, например:

def __init__(self, *args):
    if isinstance(args[0], date):
        # это date
    elif all(isinstance(arg, int) for arg in args[0:3]):
        # это 3 int

Comments

[mkone112]

в 3.8 по идее есть благодаря singledispathmethod, вот я и хочу примерно тоже, но в 3.6

[shurshur]

mkone112, singledispatch больше похож на грязный хак :) Так как работает по типу первого аргумента. В принципе, ничто не мешает прямо бэкпортировать или написать что-то похожее по аналогии на декораторах.

[mkone112]

shurshur, вот хочется такого-же, но без бэкпортов.

[mkone112]

def __init__(self, *args):
    if isinstance(args[0], date):
        # это date
    elif all(isinstance(arg, int) for arg in args[0:3]):
        # это 3 int

В этом примере при вызове
help(cls.__init__)
сигнатура вызова будет выглядеть как
__init__(self, *args)
как сделать так чтобы метод имел как-бы несколько сигнатур?

[shurshur]

mkone112, ну так посмотреть в исходники Python нужной версии, как у них реализовано. Скорее всего, не очень сложно.

[mkone112]

shurshur, вообще не сложно, но я не хочу тащить реализацию такой простой вещи из python 3.8+. Хотелось сделать проще, и таки удалось.

Answer 3

[Сергей Карбивничий]

@functools.singledispatch

Comments

[mkone112]

Оно ведь вроде только с функциями работает, а singledispatchmethod появился только в 3.8