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

1import click 9 ctx1abcdefghi

2import typer 9 ctx1abcdefghi

3 

4 

5class CustomClass: 9 ctx1abcdefghi

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

7 self.value = value 9 ctx1abcdefghi

8 

9 def __repr__(self): 9 ctx1abcdefghi

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

11 

12 

13class CustomClassParser(click.ParamType): 9 ctx1abcdefghi

14 name = "CustomClass" 9 ctx1abcdefghi

15 

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

17 return CustomClass(value * 3) 9 ctx1abcdefghi

18 

19 

20app = typer.Typer() 9 ctx1abcdefghi

21 

22 

23@app.command() 9 ctx1abcdefghi

24def main( 9 ctx1abcdefghi

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

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

27): 

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

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

30 

31 

32if __name__ == "__main__": 9 ctx1abcdefghi

33 app() 9 ctx1abcdefghi