config/run-and-prefix.py

   1 #!/usr/bin/env python
   2 # This Source Code Form is subject to the terms of the Mozilla Public
   3 # License, v. 2.0. If a copy of the MPL was not distributed with this
   4 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   5
   6 # This script runs a process and prefixes its output with.
   7 # Usage: run-and-prefix.py prefix command arg0 argv1...
   8
   9 from __future__ import absolute_import, print_function
  10
  11 import os
  12 import subprocess
  13 import sys
  14
  15 sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
  16 sys.stderr = os.fdopen(sys.stderr.fileno(), 'w', 0)
  17
  18 prefix = sys.argv[1]
  19 args = sys.argv[2:]
  20
  21 p = subprocess.Popen(args, bufsize=0,
  22                      stdout=subprocess.PIPE,
  23                      stderr=subprocess.STDOUT,
  24                      stdin=sys.stdin.fileno(),
  25                      universal_newlines=True)
  26
  27 while True:
  28     data = p.stdout.readline()
  29
  30     if data == b'':
  31         break
  32
  33     print('%s> %s' % (prefix, data), end=b'')
  34
  35 sys.exit(p.wait())