Coverage for typer / _types.py: 100%

1from enum import Enum 7 ctx1abcdefg

2from typing import TypeVar 7 ctx1abcdefg

3 

4import click 7 ctx1abcdefg

5 

6ParamTypeValue = TypeVar("ParamTypeValue") 7 ctx1abcdefg

7 

8 

9class TyperChoice(click.Choice[ParamTypeValue]): 7 ctx1abcdefg

10 def normalize_choice( 7 ctx1abcdefg

11 self, choice: ParamTypeValue, ctx: click.Context | None 

12 ) -> str: 

13 # Click 8.2.0 added a new method `normalize_choice` to the `Choice` class 

14 # to support enums, but it uses the enum names, while Typer has always used the 

15 # enum values. 

16 # This class overrides that method to maintain the previous behavior. 

17 # In Click: 

18 # normed_value = choice.name if isinstance(choice, Enum) else str(choice) 

19 normed_value = str(choice.value) if isinstance(choice, Enum) else str(choice) 7 ctx1abcdefg

20 

21 if ctx is not None and ctx.token_normalize_func is not None: 7 ctx1abcdefg

22 normed_value = ctx.token_normalize_func(normed_value) 7 ctx1abcdefg

23 

24 if not self.case_sensitive: 7 ctx1abcdefg

25 normed_value = normed_value.casefold() 7 ctx1abcdefg

26 

27 return normed_value 7 ctx1abcdefg