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, unicode_literals
  10
  11 import os
  12 import subprocess
  13 import sys
  14
  15 sys.stdout = os.fdopen(sys.stdout.fileno(), 'wb', 0)
  16 sys.stderr = os.fdopen(sys.stderr.fileno(), 'wb', 0)
  17
  18 prefix = sys.argv[1].encode('utf-8')
  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
  26 while True:
  27     data = p.stdout.readline()
  28
  29     if data == b'':
  30         break
  31
  32     sys.stdout.write(b'%s> %s' % (prefix, data))
  33
  34 sys.exit(p.wait())