fastapi / typer

Typer, build great CLIs. Easy to code. Based on Python type hints.
https://typer.tiangolo.com/
MIT License
15.76k stars 671 forks source link

Is support for Union types on the roadmap? #461

Open msarmi9 opened 2 years ago

msarmi9 commented 2 years ago

First Check

Commit to Help

Example Code

import typer
from typing import Union

def main(name: Union[str, int]):
    print(f"Hello {name}")

if __name__ == "__main__":
    typer.run(main)

Description

Typer does not yet support Union types. Running the above snippet yields:

AssertionError: Typer Currently doesn't support Union types

Not sure how much demand there is for this feature, but I'd personally love to have it.

Operating System

macOS

Operating System Details

No response

Typer Version

0.6.1

Python Version

3.10

Additional Context

Typer is amazing. Absolutely in love with it 💖.

EmmaJaneBonestell commented 1 year ago

https://typer.tiangolo.com/tutorial/commands/help/#help-panels-for-cli-parameters It might be nice to remove Unions from your examples if it's not on the horizon.

squarebridges commented 1 year ago

Just wanted to throw some more engagement on this issue. The number of issues around even the simple | None use case (Union syntax for Optional types) seems to be increasing. For my part, I have a growing number of cases in my CLI where I want to use a Union of types (particularly enums), and I've had to resort to str or Any and parse the type inside the command, because of this missing feature. I would really appreciate if this picked up steam soon.

A quick example of what I'm looking to do:

class EnumA(str, enum.Enum):
    A = "A"
    B = "B"

class EnumB(str, enum.Enum):
    C = "C"
    D = "D"

@cli.command()
def my_command(choice: EnumA | EnumB = typer.Argument(..., help="")) -> None:
    typer.echo(f"You chose {choice}")

Currently, I have to fall back on a str annotation and then parse the str into an enum as the first operation of the command. This also means the documentation for the command is lacking clarity on the options, unless I write it out myself in the help.

msarmi9 commented 1 year ago

@ntaylorwss excellent example. much better than the one i gave and, funnily enough, exactly what i'm after as well. cheers!

rtbs-dev commented 12 months ago

Just wanted to chime in here. The official Mypy docs lean toward using typing.Literal[] syntax for these kinds of cases, and you don't actually need to support typing.Union at all... literals of literals automatically merge the options (like a union).

See the example here from those docs (the AllowedColors example is particularly salient).

Note that supporting literals would solve #76, which has a lot of recent activity, despite a bot marking it closed :sweat_smile:... I think #669 is a pretty complete PR for this, though @pchanial would have to chime in on whether that PR already supports compound literals already. :)

To demonstrate @ntaylorwss 's example,

OptA = Literal['A']
OptB = Literal['B']

MyOpts = Literal[OptA, OptB]

### or, equivalently, but without needed OptA/B later ###
MyOpts = Literal['A', 'B']
pchanial commented 11 months ago

I haven't tried compound Literals, but that's something I could look at if @tiangolo is willing to add Literal handling on the roadmap.

dotKokott commented 9 months ago

+1 Would love to see Literal handling as well.

RuntimeError: Type not yet supported: typing.Literal

rafalkrupinski commented 8 months ago

Use case for a Union: I'd like my program to accept URL (custom type) or Path as a parameter source: Path | MyUrl

echus commented 3 months ago

Is there any workaround for this currently? I've got a usecase where I need to be able to accept any of bool | str | float | int in a data science CLI application. I'm taking a str and using literal_eval to cast it back to its original type. Is there a better option?

brianmedigate commented 1 month ago

It would be nice if typer would allow me to use a Union when passing a click_type. I don't mind implementing the union functionality myself, but typer throws an exception before even checking if I passed a click_type

bschoenbach commented 3 weeks ago

Please, please make UNION possible for typer as e.g.typer.options(Union[Enum, str].......)