mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
feat: add SuppressOutput
util
This context manager suppresses/hides stdout.
This commit is contained in:
parent
5c1aa02e7b
commit
857e9c9b5f
24
invokeai/app/util/suppress_output.py
Normal file
24
invokeai/app/util/suppress_output.py
Normal file
@ -0,0 +1,24 @@
|
||||
import io
|
||||
import sys
|
||||
from typing import Any
|
||||
|
||||
|
||||
class SuppressOutput:
|
||||
"""Context manager to suppress stdout.
|
||||
|
||||
Example:
|
||||
```
|
||||
with SuppressOutput():
|
||||
print("This will not be printed")
|
||||
```
|
||||
"""
|
||||
|
||||
def __enter__(self):
|
||||
# Save the original stdout
|
||||
self._original_stdout = sys.stdout
|
||||
# Redirect stdout to a dummy StringIO object
|
||||
sys.stdout = io.StringIO()
|
||||
|
||||
def __exit__(self, *args: Any, **kwargs: Any):
|
||||
# Restore stdout
|
||||
sys.stdout = self._original_stdout
|
Loading…
Reference in New Issue
Block a user