【python】方法/函数重载

原创已于 2023-03-23 09:27:55 修改 · 215 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

于 2023-03-22 18:01:34 首次发布

文章展示了如何在Python中使用`singledispatch`和`overload`来实现多态功能。通过定义不同类型的参数处理函数，可以创建一个可以根据参数类型动态调用不同实现的通用函数。同时，文章还演示了如何在类的方法中使用`singledispatchmethod`实现类似的功能。

# -*- encoding: utf-8 -*-
"""
@File: 123.py
@Time: 2021/3/22 14:54
@Author: cyl
"""
from typing import Any, overload
from functools import singledispatch, singledispatchmethod


@singledispatch
def fun(page: int) -> int:
    return page + 100


@fun.register
def _(page: str) -> str:
    return page + ", overload str"


@fun.register
def _(page: tuple) -> int:
    return page[0] + page[1]


class Run:
    def __init__(self):
        self.name = ", cyl"
        self.age = 0
        self.width = 100

    # def area(self, num: Any) -> Any:
    #     if isinstance(num, int):
    #         return num + self.width + 1
    #     if isinstance(num, str):
    #         return num + self.name + ", handsome"
    #     if isinstance(num, tuple):
    #         return num

    @singledispatchmethod
    def area(self, num: int) -> int:
        return num + self.width

    @area.register
    def _(self, num: str, path: str) -> str:
        return num + self.name + path

    @area.register
    def _(self, num: tuple) -> tuple:
        return num


if __name__ == "__main__":
    print(fun("123"))
    print(fun(123))
    print(fun((123, 456)))

    print("*" * 30)

    run: Run = Run()
    print(run.area(100))
    print(run.area("100", "hhh"))
    print(run.area((100, 101)))

输出：

123, overload str
223
579
******************************
200
100, cylhhh
(100, 101)