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

1import click 8 ctx1habcdefg

2import typer 8 ctx1habcdefg

3 

4 

5class CustomClass: 8 ctx1habcdefg

6 def __init__(self, value: str): 8 ctx1habcdefg

7 self.value = value 8 ctx1habcdefg

8 

9 def __repr__(self): 8 ctx1habcdefg

10 return f"<CustomClass: value={self.value}>" 8 ctx1habcdefg

11 

12 

13class CustomClassParser(click.ParamType): 8 ctx1habcdefg

14 name = "CustomClass" 8 ctx1habcdefg

15 

16 def convert(self, value, param, ctx): 8 ctx1habcdefg

17 return CustomClass(value * 3) 8 ctx1habcdefg

18 

19 

20def main( 7 ctx1abcdefg

21 custom_arg: CustomClass = typer.Argument(click_type=CustomClassParser()), 

22 custom_opt: CustomClass = typer.Option("Foo", click_type=CustomClassParser()), 

23): 

24 print(f"custom_arg is {custom_arg}") 8 ctx1habcdefg

25 print(f"--custom-opt is {custom_opt}") 8 ctx1habcdefg

26 

27 

28if __name__ == "__main__": 8 ctx1habcdefg

29 typer.run(main) 8 ctx1habcdefg