Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.11/site-packages/argcomplete/io.py: 50%

1from __future__ import annotations 

2 

3import contextlib 

4import os 

5import sys 

6from collections.abc import Generator 

7 

8_DEBUG = "_ARC_DEBUG" in os.environ 

9 

10debug_stream = sys.stderr 

11 

12 

13def debug(*args: object) -> None: 

14 if _DEBUG: 

15 print(file=debug_stream, *args) 

16 

17 

18@contextlib.contextmanager 

19def mute_stdout() -> Generator[None]: 

20 stdout = sys.stdout 

21 sys.stdout = open(os.devnull, "w") 

22 try: 

23 yield 

24 finally: 

25 sys.stdout = stdout 

26 

27 

28@contextlib.contextmanager 

29def mute_stderr() -> Generator[None]: 

30 stderr = sys.stderr 

31 sys.stderr = open(os.devnull, "w") 

32 try: 

33 yield 

34 finally: 

35 sys.stderr.close() 

36 sys.stderr = stderr 

37 

38 

39def warn(*args: object) -> None: 

40 """ 

41 Prints **args** to standard error when running completions. This will interrupt the user's command line interaction; 

42 use it to indicate an error condition that is preventing your completer from working. 

43 """ 

44 print(file=debug_stream) 

45 print(file=debug_stream, *args)