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

1import click 8 ctx1abcdefgh

2import typer 8 ctx1abcdefgh

3 

4 

5class CustomClass: 8 ctx1abcdefgh

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

7 self.value = value 8 ctx1abcdefgh

8 

9 def __repr__(self): 8 ctx1abcdefgh

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

11 

12 

13class CustomClassParser(click.ParamType): 8 ctx1abcdefgh

14 name = "CustomClass" 8 ctx1abcdefgh

15 

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

17 return CustomClass(value * 3) 8 ctx1abcdefgh

18 

19 

20app = typer.Typer() 8 ctx1abcdefgh

21 

22 

23@app.command() 8 ctx1abcdefgh

24def main( 8 ctx1abcdefgh

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}") 8 ctx1abcdefgh

29 print(f"--custom-opt is {custom_opt}") 8 ctx1abcdefgh

30 

31 

32if __name__ == "__main__": 8 ctx1abcdefgh

33 app() 8 ctx1abcdefgh