You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
126 lines
3.2 KiB
Python
126 lines
3.2 KiB
Python
#!/usr/bin/env python
|
|
# Copyright 2018 The Chromium Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
from __future__ import print_function
|
|
|
|
import json
|
|
import os
|
|
import platform
|
|
import subprocess
|
|
import sys
|
|
|
|
import ninjalog_uploader
|
|
import subprocess2
|
|
|
|
THIS_DIR = os.path.dirname(__file__)
|
|
UPLOADER = os.path.join(THIS_DIR, 'ninjalog_uploader.py')
|
|
CONFIG = os.path.join(THIS_DIR, 'ninjalog.cfg')
|
|
VERSION = 2
|
|
|
|
|
|
def LoadConfig():
|
|
if os.path.isfile(CONFIG):
|
|
with open(CONFIG, 'rb') as f:
|
|
config = json.load(f)
|
|
if config['version'] == VERSION:
|
|
config['countdown'] = max(0, config['countdown'] - 1)
|
|
return config
|
|
|
|
return {
|
|
'is-googler':
|
|
ninjalog_uploader.IsGoogler('chromium-build-stats.appspot.com'),
|
|
'countdown': 10,
|
|
'version': VERSION,
|
|
}
|
|
|
|
|
|
def SaveConfig(config):
|
|
with open(CONFIG, 'wb') as f:
|
|
json.dump(config, f)
|
|
|
|
|
|
def ShowMessage(countdown):
|
|
whitelisted = '\n'.join(
|
|
[' * %s' % config for config in ninjalog_uploader.WHITELISTED_CONFIGS])
|
|
print("""
|
|
Your ninjalog will be uploaded to build stats server. The uploaded log will be
|
|
used to analyze user side build performance.
|
|
|
|
The following information will be uploaded with ninjalog.
|
|
* OS (e.g. Win, Mac or Linux)
|
|
* number of cpu cores of building machine
|
|
* build targets (e.g. chrome, browser_tests)
|
|
* parallelism passed by -j flag
|
|
* following build configs
|
|
%s
|
|
|
|
Uploading ninjalog will be started after you run autoninja another %d time.
|
|
|
|
If you don't want to upload ninjalog, please run following command.
|
|
$ %s opt-out
|
|
|
|
If you want to allow upload ninjalog from next autoninja run, please run the
|
|
following command.
|
|
$ %s opt-in
|
|
|
|
If you have questions about this, please send mail to infra-dev@chromium.org
|
|
|
|
You can find a more detailed explanation in
|
|
%s
|
|
|
|
""" % (whitelisted, countdown, __file__, __file__,
|
|
os.path.abspath(os.path.join(THIS_DIR, "ninjalog.README.md"))))
|
|
|
|
|
|
def main():
|
|
config = LoadConfig()
|
|
|
|
if len(sys.argv) == 2 and sys.argv[1] == 'opt-in':
|
|
config['opt-in'] = True
|
|
config['countdown'] = 0
|
|
SaveConfig(config)
|
|
print('ninjalog upload is opted in.')
|
|
return 0
|
|
|
|
if len(sys.argv) == 2 and sys.argv[1] == 'opt-out':
|
|
config['opt-in'] = False
|
|
SaveConfig(config)
|
|
print('ninjalog upload is opted out.')
|
|
return 0
|
|
|
|
if 'opt-in' in config and not config['opt-in']:
|
|
# Upload is opted out.
|
|
return 0
|
|
|
|
if not config.get("is-googler", False):
|
|
# Not googler.
|
|
return 0
|
|
|
|
if config.get("countdown", 0) > 0:
|
|
# Need to show message.
|
|
ShowMessage(config["countdown"])
|
|
# Only save config if something has meaningfully changed.
|
|
SaveConfig(config)
|
|
return 0
|
|
|
|
if len(sys.argv) == 1:
|
|
# dry-run for debugging.
|
|
print("upload ninjalog dry-run")
|
|
return 0
|
|
|
|
# Run upload script without wait.
|
|
devnull = open(os.devnull, "w")
|
|
creationnflags = 0
|
|
if platform.system() == 'Windows':
|
|
creationnflags = subprocess.CREATE_NEW_PROCESS_GROUP
|
|
subprocess2.Popen(['vpython', UPLOADER] + sys.argv[1:],
|
|
stdout=devnull,
|
|
stderr=devnull,
|
|
creationflags=creationnflags)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|