Coverage for typer / _click / _textwrap.py: 100%

1import textwrap 7 ctx1abcdefg

2from collections.abc import Generator 7 ctx1abcdefg

3from contextlib import contextmanager 7 ctx1abcdefg

4 

5 

6class TextWrapper(textwrap.TextWrapper): 7 ctx1abcdefg

7 def _handle_long_word( 7 ctx1abcdefg

8 self, 

9 reversed_chunks: list[str], 

10 cur_line: list[str], 

11 cur_len: int, 

12 width: int, 

13 ) -> None: 

14 space_left = max(width - cur_len, 1) 7 ctx1abcdefg

15 

16 last = reversed_chunks[-1] 7 ctx1abcdefg

17 cut = last[:space_left] 7 ctx1abcdefg

18 res = last[space_left:] 7 ctx1abcdefg

19 cur_line.append(cut) 7 ctx1abcdefg

20 reversed_chunks[-1] = res 7 ctx1abcdefg

21 

22 @contextmanager 7 ctx1abcdefg

23 def extra_indent(self, indent: str) -> Generator[None]: 7 ctx1abcdefg

24 old_initial_indent = self.initial_indent 7 ctx1abcdefg

25 old_subsequent_indent = self.subsequent_indent 7 ctx1abcdefg

26 self.initial_indent += indent 7 ctx1abcdefg

27 self.subsequent_indent += indent 7 ctx1abcdefg

28 

29 try: 7 ctx1abcdefg

30 yield 7 ctx1abcdefg

31 finally: 

32 self.initial_indent = old_initial_indent 7 ctx1abcdefg

33 self.subsequent_indent = old_subsequent_indent 7 ctx1abcdefg

34 

35 def indent_only(self, text: str) -> str: 7 ctx1abcdefg

36 rv = [] 7 ctx1abcdefg

37 

38 for idx, line in enumerate(text.splitlines()): 7 ctx1abcdefg

39 indent = self.initial_indent 7 ctx1abcdefg

40 

41 if idx > 0: 7 ctx1abcdefg

42 indent = self.subsequent_indent 7 ctx1abcdefg

43 

44 rv.append(f"{indent}{line}") 7 ctx1abcdefg

45 

46 return "\n".join(rv) 7 ctx1abcdefg