Issue
How can I get the literal value out of a Literal[]
from typing
?
from typing import Literal, Union
Add = Literal['add']
Multiply = Literal['mul']
Action = Union[Add,Multiply]
def do(a: Action):
if a == Add:
print("Adding!")
elif a == Multiply:
print("Multiplying!")
else:
raise ValueError
do('add')
The code above type checks since 'add'
is of type Literal['add']
, but at runtime, it raises a ValueError since the string 'add'
is not the same as typing.Literal['add']
.
How can I, at runtime, reuse the literals that I defined at type level?
Solution
The typing
module provides a function get_args
which retrieves the arguments with which your Literal
was initialized.
>>> from typing import Literal, get_args
>>> l = Literal['add', 'mul']
>>> get_args(l)
('add', 'mul')
However, I don't think you gain anything by using a Literal
for what you propose.
Answered By - tripleee
0 comments:
Post a Comment
Note: Only a member of this blog may post a comment.