Início / coding / Perguntas / 77902593
Accepted
Pablo
Asked: 2024-01-30 04:58:52 +0800 CST

Defina o tipo de retorno de uma função dependendo do valor real de um argumento

  • 772

Eu sei que com TypeVarvocê você pode definir o tipo de retorno de uma função dependendo do tipo de um de seus argumentos. No entanto, quero o tipo baseado no conteúdo de um argumento.

Por exemplo, a função openretorna tipos diferentes dependendo do conteúdo do modeargumento:

a = open("/etc/fstab", "r")
b = open("/etc/fstab", "rb")

Neste caso, my pyrightdiz que aé a TextIOWrappere bé a BufferedReader.

Também percebi que meu direito autoral infere o tipo correto com este código:

mode = "rb"
a = open("/etc/fstab", mode)

mas se eu fizer:

def funny(mode: str):
    a = open("/ect/fstab", mode)

então o direito autoral infere IO[Any]. Portanto, claramente os direitos autorais levam em consideração mais do que o valor literal de mode. Então, como o direito autoral está inferindo o tipo corretamente? Isso ocorre porque o tipo sugere openou é algum tipo de "codificação" de direitos autorais para a função open?

python

1 respostas

  1. Best Answer

    Você pode fazer isso com o typing.overloaddecorador e o Literaltipo.

    type TextReadModes = Literal['r', 'r+']
type BinaryReadModes = Literal['rb', 'rb+']

@overload
def open(filename, mode: TextReadModes, ...) -> TextIOWrapper:
    ...

@overload
def open(filename, mode: BinaryReadModes, ...) -> BufferedReader:
    ...
    • 4

relate perguntas