qubes-installer-qubes-os/anaconda/tests/lib/shutup.py
M. Vefa Bicakci 38f3e28d77
anaconda: Update to 23.19.10
Use the output of

  git diff --full-index --binary anaconda-22.20.13-1..anaconda-23.19.10-1

from anaconda's git repository and fix-up merge conflicts.
2016-04-10 00:00:00 -04:00

71 lines
2.2 KiB
Python

#
# shutup.py: context manager for ignoring overly verbose output
#
# Copyright (C) 2015 Red Hat, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published
# by the Free Software Foundation; either version 2.1 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# Author: David Shea <dshea@redhat.com>
import os
from contextlib import contextmanager
@contextmanager
def shutup():
"""Run something with stdout and stderr redirected to /dev/null
The redirections will be process-wide, so this is not recommended
for multithreaded applications.
"""
# Ignore all the warnings about interruptible calls
# pylint: disable=ignorable-system-call, interruptible-system-call
# Wrap the whole thing a try-finally to ensure errors don't leak file descriptor
old_stdout = None
old_stderr = None
devnull_fd = None
try:
# Save the current file descriptors
old_stdout = os.dup(1)
old_stderr = os.dup(2)
# Redirect to /dev/null
# Try to undo partial redirects if something goes wrong
devnull_fd = os.open(os.devnull, os.O_WRONLY)
try:
os.dup2(devnull_fd, 1)
os.dup2(devnull_fd, 2)
except:
os.dup2(old_stderr, 2)
os.dup2(old_stdout, 1)
raise
# Run the body. Cleanup in finally to ensure stderr is restored before
# an exception is raised.
try:
yield
finally:
os.dup2(old_stderr, 2)
os.dup2(old_stdout, 1)
finally:
if old_stdout is not None:
os.close(old_stdout)
if old_stderr is not None:
os.close(old_stderr)
if devnull_fd is not None:
os.close(devnull_fd)