Coverage for docs_src/parameter_types/custom_types/tutorial002_an.py: 100%

1import click 9 ctx1abcdefghi

2import typer 9 ctx1abcdefghi

3from typing_extensions import Annotated 9 ctx1abcdefghi

4 

5 

6class CustomClass: 9 ctx1abcdefghi

7 def __init__(self, value: str): 9 ctx1abcdefghi

8 self.value = value 9 ctx1abcdefghi

9 

10 def __repr__(self): 9 ctx1abcdefghi

11 return f"<CustomClass: value={self.value}>" 9 ctx1abcdefghi

12 

13 

14class CustomClassParser(click.ParamType): 9 ctx1abcdefghi

15 name = "CustomClass" 9 ctx1abcdefghi

16 

17 def convert(self, value, param, ctx): 9 ctx1abcdefghi

18 return CustomClass(value * 3) 9 ctx1abcdefghi

19 

20 

21app = typer.Typer() 9 ctx1abcdefghi

22 

23 

24@app.command() 9 ctx1abcdefghi

25def main( 9 ctx1abcdefghi

26 custom_arg: Annotated[CustomClass, typer.Argument(click_type=CustomClassParser())], 

27 custom_opt: Annotated[ 

28 CustomClass, typer.Option(click_type=CustomClassParser()) 

29 ] = "Foo", 

30): 

31 print(f"custom_arg is {custom_arg}") 9 ctx1abcdefghi

32 print(f"--custom-opt is {custom_opt}") 9 ctx1abcdefghi

33 

34 

35if __name__ == "__main__": 9 ctx1abcdefghi

36 app() 9 ctx1abcdefghi