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(
  22     args,
  23     bufsize=0,
  24     stdout=subprocess.PIPE,
  25     stderr=subprocess.STDOUT,
  26     stdin=sys.stdin.fileno(),
  27     close_fds=False,
  28 )
  29
  30 while True:
  31     data = p.stdout.readline()
  32
  33     if data == b"":
  34         break
  35
  36     sys.stdout.write(b"%s> %s" % (prefix, data))
  37
  38 sys.exit(p.wait())