forked from borgmatic-collective/borgmatic
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
import subprocess
|
|
|
|
from borgmatic.logger import get_logger
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
def execute_and_log_output(full_command):
|
|
process = subprocess.Popen(full_command, stdout=None, stderr=subprocess.PIPE)
|
|
|
|
while process.poll() is None:
|
|
line = process.stderr.readline().rstrip().decode()
|
|
if line.startswith('borg: error:'):
|
|
logger.error(line)
|
|
else:
|
|
logger.info(line)
|
|
|
|
logger.info(process.stderr.read().rstrip().decode())
|
|
exit_code = process.poll()
|
|
|
|
if exit_code != 0:
|
|
raise subprocess.CalledProcessError(exit_code, full_command)
|
|
|
|
|
|
def execute_command(full_command, capture_output=False):
|
|
'''
|
|
Execute the given command (a sequence of command/argument strings). If capture output is True,
|
|
then return the command's output as a string.
|
|
'''
|
|
logger.debug(' '.join(full_command))
|
|
|
|
if capture_output:
|
|
output = subprocess.check_output(full_command)
|
|
return output.decode() if output is not None else None
|
|
else:
|
|
execute_and_log_output(full_command)
|