【Python】引数が異なる関数を定義する方法【オーバーロード】

この記事では、Pythonで関数のオーバーロードを実装する方法を解説します。

C++ や C# のようには定義できませんが functoolsモジュールを使うことで Python でもオーバーロードを実装することができます。

オーバーロードとは、引数が異なる同じ名前の関数を定義することです。

以下のコードは、異なる型の引数を受け取る同じ名前の関数を定義し、呼び出していますが、後に定義した関数しか呼び出せません。

def func(text: str):
    print(f'text: {text}')


def func(num: int):
    print(f'num: {num}')


func('abc')
# num: abc

func(10)
# num: 10

このようなコードに functoolsモジュールを用いることで、渡された引数の型によって呼び出す関数を変更させることができます。

それでは、オーバーロードする方法を見ていきましょう！

オーバーロードの実装方法
まとめ

オーバーロードの実装方法

Pythonでオーバーロードを実装するには、functoolsモジュールのsingledispatchデコレーターを使って、

from functools import singledispatch

以下の手順で定義します。

STEP1

基準となる関数の定義
基準となる関数に singledispatchデコレーターを付けます。関数名はここでは func にしておきます。
```
@singledispatch
def func(引数):
    # 何らかの処理
```
オーバーロードするには、引数を1つ以上定義する必要があります
STEP2

他の関数を登録
次に、オーバーロード実装を追加します。

この関数にもデコレーターを修飾させますが singledispatchデコレーターを付けた関数から呼び出した registerデコレーターを付けます。関数名はアンダースコアにしましたが func関数と同じでなければなんでも良いです。
```
@func.register
def _(引数):
    # 何らかの処理
```
singledispatchデコレーターは、１つ目の引数の型で呼び出す関数を決めています。なので、１つ目の引数の型がわかるように定義しなければなりません。

型アノテーションを定義してある場合は、自動的に推測されます。
```
@func.register
def _(num: int):
    # 何らかの処理
```
型アノテーションを定義してない場合は、registerデコレーターに型を指定する必要があります。
```
@func.register(int)
def _(num):
    # 何らかの処理
```

試しに最初のコードをオーバーロードしてみましょう！

from functools import singledispatch

@singledispatch
def func(text: str):
    print(f'text: {text}')


@func.register
def _(num: int):
    print(f'num: {num}')


func('abc')
# text: abc

func(10)
# num: 10

このように、引数に渡した値の型によって呼び出す関数を変更することができました。