run-and-prefix.py - mozsearch

Enable keyboard shortcuts

#!/usr/bin/env python

# This Source Code Form is subject to the terms of the Mozilla Public

# License, v. 2.0. If a copy of the MPL was not distributed with this

# file, You can obtain one at http://mozilla.org/MPL/2.0/.

# This script runs a process and prefixes its output with.

# Usage: run-and-prefix.py prefix command arg0 argv1...

import os

import subprocess

import sys

sys.stdout = os.fdopen(sys.stdout.fileno(), "wb", 0)

sys.stderr = os.fdopen(sys.stderr.fileno(), "wb", 0)

prefix = sys.argv[1].encode("utf-8")

args = sys.argv[2:]

p = subprocess.Popen(

    args,

    bufsize=0,

    stdout=subprocess.PIPE,

    stderr=subprocess.STDOUT,

    stdin=sys.stdin.fileno(),

    close_fds=False,

while True:

    data = p.stdout.readline()

    if data == b"":

        break

    if data.startswith(b"BUILDSTATUS"):

        sys.stdout.write(data)

    else:

        sys.stdout.write(b"%s> %s" % (prefix, data))

sys.exit(p.wait())