fune/tools/tryselect/selectors/chooser/__init__.py
Andrew Halberstadt 0c56961d92 Bug 1499822 - [tryselect] Implement |mach try chooser| r=sclements
Usage:

  $ ./mach try chooser

Will start a local flask server and server a "trychooser-like" page
that is dynamically generated from the taskgraph.

Differential Revision: https://phabricator.services.mozilla.com/D14903

--HG--
extra : moz-landing-system : lando
2019-01-08 16:46:57 +00:00

50 lines
1.7 KiB
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/.
from __future__ import absolute_import, print_function, unicode_literals
import os
import webbrowser
from threading import Timer
from tryselect.cli import BaseTryParser
from tryselect.tasks import generate_tasks
from tryselect.push import check_working_directory, push_to_try, vcs
here = os.path.abspath(os.path.dirname(__file__))
class ChooserParser(BaseTryParser):
name = 'chooser'
arguments = []
common_groups = ['push', 'task']
templates = ['artifact', 'env', 'rebuild', 'chemspill-prio', 'gecko-profile']
def run_try_chooser(update=False, query=None, templates=None, full=False, parameters=None,
save=False, preset=None, mod_presets=False, push=True, message='{msg}',
**kwargs):
from .app import create_application
check_working_directory(push)
tg = generate_tasks(parameters, full, root=vcs.path)
app = create_application(tg)
if os.environ.get('WERKZEUG_RUN_MAIN') == 'true':
# we are in the reloader process, don't open the browser or do any try stuff
app.run()
return
# give app a second to start before opening the browser
Timer(1, lambda: webbrowser.open('http://127.0.0.1:5000')).start()
app.run()
selected = app.tasks
if not selected:
print("no tasks selected")
return
msg = "Try Chooser Enhanced ({} tasks selected)".format(len(selected))
return push_to_try('chooser', message.format(msg=msg), selected, templates, push=push,
closed_tree=kwargs["closed_tree"])