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

1import click 9 ctx1iabcdefgh

2import typer 9 ctx1iabcdefgh

3from typing_extensions import Annotated 9 ctx1iabcdefgh

4 

5 

6class CustomClass: 9 ctx1iabcdefgh

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

8 self.value = value 9 ctx1iabcdefgh

9 

10 def __repr__(self): 9 ctx1iabcdefgh

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

12 

13 

14class CustomClassParser(click.ParamType): 9 ctx1iabcdefgh

15 name = "CustomClass" 9 ctx1iabcdefgh

16 

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

18 return CustomClass(value * 3) 9 ctx1iabcdefgh

19 

20 

21def main( 8 ctx1abcdefgh

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

23 custom_opt: Annotated[ 

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

25 ] = "Foo", 

26): 

27 print(f"custom_arg is {custom_arg}") 9 ctx1iabcdefgh

28 print(f"--custom-opt is {custom_opt}") 9 ctx1iabcdefgh

29 

30 

31if __name__ == "__main__": 9 ctx1iabcdefgh

32 typer.run(main) 9 ctx1iabcdefgh