blob: df079f28f091ca7fbad71c11656e1ad51653e8a6 [file] [log] [blame]
#!/usr/bin/env python
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
# This script generates a header file which contains definitions
# for the current Kudu build (eg timestamp, git hash, etc)
from __future__ import print_function
import logging
import os
import subprocess
import sys
ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
_GET_UPSTREAM_COMMIT_SCRIPT = os.path.join(ROOT, "build-support", "")
# Alias raw_input() to input() in Python 2.
input = raw_input
except NameError:
def get_upstream_commit():
""" Return the last commit hash that appears to have been committed by gerrit. """
return check_output(_GET_UPSTREAM_COMMIT_SCRIPT).strip().decode('utf-8')
def init_logging():
format='%(asctime)s %(levelname)s: %(message)s')
class ColorFilter(logging.Filter):
""" logging.Filter implementation which colorizes the output to console. """
def filter(self, record):
if record.levelno >= logging.ERROR:
record.msg = Colors.RED + record.msg + Colors.RESET
elif record.levelno >= logging.WARNING:
record.msg = Colors.YELLOW + record.msg + Colors.RESET
return True
class Colors(object):
""" ANSI color codes. """
def __on_tty(x):
if not os.isatty(sys.stdout.fileno()):
return ""
return x
RED = __on_tty("\x1b[31m")
GREEN = __on_tty("\x1b[32m")
YELLOW = __on_tty("\x1b[33m")
RESET = __on_tty("\x1b[m")
def check_output(*popenargs, **kwargs):
r"""Run command with arguments and return its output as a byte string.
Backported from Python 2.7 as it's implemented as pure python on stdlib.
>>> check_output(['/usr/bin/python', '--version'])
Python 2.6.2
process = subprocess.Popen(stdout=subprocess.PIPE, *popenargs, **kwargs)
output, unused_err = process.communicate()
retcode = process.poll()
if retcode:
cmd = kwargs.get("args")
if cmd is None:
cmd = popenargs[0]
error = subprocess.CalledProcessError(retcode, cmd)
error.output = output
raise error
return output
def confirm_prompt(prompt):
Issue the given prompt, and ask the user to confirm yes/no. Returns true
if the user confirms.
while True:
print(prompt, end=' [Y/n]: ')
if not os.isatty(sys.stdout.fileno()):
print("Not running interactively. Assuming 'N'.")
return False
r = input().strip().lower()
if r in ['y', 'yes', '']:
return True
elif r in ['n', 'no']:
return False
def get_my_email():
""" Return the email address in the user's git config. """
return check_output(['git', 'config', '--get',
def get_thirdparty_dir():
env = os.environ.copy()
return env['THIRDPARTY_DIR'] if env.has_key('THIRDPARTY_DIR') else os.path.join(ROOT, "thirdparty")