94 lines
2.5 KiB
Python
Executable file
94 lines
2.5 KiB
Python
Executable file
#!/usr/bin/env python
|
|
# PYTHON_ARGCOMPLETE_OK
|
|
|
|
# (c) 2016 Red Hat, Inc.
|
|
#
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""CLI tool for starting new CI runs."""
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
# noinspection PyCompatibility
|
|
import argparse
|
|
import json
|
|
import os
|
|
import sys
|
|
import requests
|
|
import requests.auth
|
|
|
|
try:
|
|
import argcomplete
|
|
except ImportError:
|
|
argcomplete = None
|
|
|
|
# TODO: Dev does not have a token for AZP, somebody please test this.
|
|
|
|
# Following changes should be made to improve the overall style:
|
|
# TODO use new style formatting method.
|
|
# TODO type hints.
|
|
|
|
|
|
def main():
|
|
"""Main program body."""
|
|
|
|
args = parse_args()
|
|
|
|
key = os.environ.get('AZP_TOKEN', None)
|
|
if not key:
|
|
sys.stderr.write("please set you AZP token in AZP_TOKEN")
|
|
sys.exit(1)
|
|
|
|
start_run(args, key)
|
|
|
|
|
|
def parse_args():
|
|
"""Parse and return args."""
|
|
|
|
parser = argparse.ArgumentParser(description='Start a new CI run.')
|
|
|
|
parser.add_argument('-p', '--pipeline-id', type=int, default=20, help='pipeline to download the job from')
|
|
parser.add_argument('--ref', help='git ref name to run on')
|
|
|
|
parser.add_argument('--env',
|
|
nargs=2,
|
|
metavar=('KEY', 'VALUE'),
|
|
action='append',
|
|
help='environment variable to pass')
|
|
|
|
if argcomplete:
|
|
argcomplete.autocomplete(parser)
|
|
|
|
args = parser.parse_args()
|
|
|
|
return args
|
|
|
|
|
|
def start_run(args, key):
|
|
"""Start a new CI run."""
|
|
|
|
url = "https://dev.azure.com/ansible/ansible/_apis/pipelines/%s/runs?api-version=6.0-preview.1" % args.pipeline_id
|
|
payload = {"resources": {"repositories": {"self": {"refName": args.ref}}}}
|
|
|
|
resp = requests.post(url, auth=requests.auth.HTTPBasicAuth('user', key), data=payload)
|
|
resp.raise_for_status()
|
|
|
|
print(json.dumps(resp.json(), indent=4, sort_keys=True))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|