Подскажите пожалуйста, как конкретизировать тип из union если я какбы

Question

Подскажите пожалуйста, как конкретизировать тип из union если я какбы

заведомо знаю какой он

Типа есть функция

def func() -> Foo | Bar

И в другом месте я пишу

a = func()

Но точно знаю, что это Foo

Как мне это указать?

#backend #programming #python #russian

0

03.01.2022

12 ответов

28 просмотров

Tishka17

Убрать юнион, взять typing.overload. или вообще делать не так

0

03.01.2022

Максим Мартынов

Есть typing.cast, но это прям совсем крайний случай

0

03.01.2022

Źmićer Rubinštejn Автор вопроса

Tishka17
Убрать юнион, взять typing.overload. или вообще де...

Ну вот у меня идея (может быть дурная) - сделать __getitem__ и в зависимости от того что передаешь в value - он возвращает разные типы.

0

03.01.2022

Tishka17

Źmićer Rubinštejn
Ну вот у меня идея (может быть дурная) - сделать _...

Оверлоад

0

03.01.2022

Źmićer Rubinštejn Автор вопроса

Tishka17
Оверлоад

К сожалению arg в _getitem__ одного и того же типа

0

03.01.2022

decorator-factory

покажи больше кода

0

03.01.2022

decorator-factory

decorator-factory
покажи больше кода

Если ты в конкретном месте знаешь по какой-то причине, что там Foo: a) assert value is Foo — проверить в рантайме b) value = typing.cast(Foo, value) — без проверки в рантайме c) подумать, как сделать так, чтобы юниона всё-таки не было — зависит от кода

0

03.01.2022

Źmićer Rubinštejn Автор вопроса

decorator-factory
покажи больше кода

class A: pass class B: pass mapping = { “a”: A, “b”: B } class Foo: def __getitem__(self, arg): return mapping[arg]

0

03.01.2022

decorator-factory

Źmićer Rubinštejn
class A: pass class B: pass mapping = { “a...

class Foo: @overload def __getitem__(self, arg: Literal["a"]) -> A: ... @overload def __getitem__(self, arg: Literal["b"]) -> B: ... def __getitem__(self, arg: str) -> A | B: return mapping[arg]

0

03.01.2022

Źmićer Rubinštejn Автор вопроса

decorator-factory
class Foo: @overload def __getitem__(self,...

Теперь надо понять как сделать это автоматически чтобы был single source of truth - mapping

0

03.01.2022

Tishka17

Źmićer Rubinštejn
class A: pass class B: pass mapping = { “a...

А зачем так?

0

03.01.2022