119 lines
3.8 KiB
Python
119 lines
3.8 KiB
Python
#!/usr/bin/env python
|
|
# Copyright (C) 2015 Wayne Warren
|
|
#
|
|
# Licensed 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
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import logging
|
|
import sys
|
|
import time
|
|
|
|
from jenkins_jobs.builder import JenkinsManager
|
|
from jenkins_jobs.parser import YamlParser
|
|
from jenkins_jobs.registry import ModuleRegistry
|
|
from jenkins_jobs.xml_config import XmlJobGenerator
|
|
from jenkins_jobs.xml_config import XmlViewGenerator
|
|
from jenkins_jobs.errors import JenkinsJobsException
|
|
import jenkins_jobs.cli.subcommand.base as base
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class UpdateSubCommand(base.BaseSubCommand):
|
|
|
|
def parse_arg_path(self, parser):
|
|
parser.add_argument(
|
|
'path',
|
|
nargs='?',
|
|
default=sys.stdin,
|
|
help="colon-separated list of paths to YAML files "
|
|
"or directories")
|
|
|
|
def parse_arg_names(self, parser):
|
|
parser.add_argument(
|
|
'names',
|
|
help='name(s) of job(s)', nargs='*')
|
|
|
|
def parse_args(self, subparser):
|
|
update = subparser.add_parser('update')
|
|
|
|
self.parse_option_recursive_exclude(update)
|
|
|
|
self.parse_arg_path(update)
|
|
self.parse_arg_names(update)
|
|
|
|
update.add_argument(
|
|
'--delete-old',
|
|
action='store_true',
|
|
dest='delete_old',
|
|
default=False,
|
|
help='delete obsolete jobs')
|
|
update.add_argument(
|
|
'--workers',
|
|
type=int,
|
|
default=1,
|
|
dest='n_workers',
|
|
help="number of workers to use, 0 for autodetection and 1 "
|
|
"for just one worker.")
|
|
|
|
def _generate_xmljobs(self, options, jjb_config=None):
|
|
builder = JenkinsManager(jjb_config)
|
|
|
|
logger.info("Updating jobs in {0} ({1})".format(
|
|
options.path, options.names))
|
|
orig = time.time()
|
|
|
|
# Generate XML
|
|
parser = YamlParser(jjb_config)
|
|
registry = ModuleRegistry(jjb_config, builder.plugins_list)
|
|
xml_job_generator = XmlJobGenerator(registry)
|
|
xml_view_generator = XmlViewGenerator(registry)
|
|
|
|
parser.load_files(options.path)
|
|
registry.set_parser_data(parser.data)
|
|
|
|
job_data_list, view_data_list = parser.expandYaml(
|
|
registry, options.names)
|
|
|
|
xml_jobs = xml_job_generator.generateXML(job_data_list)
|
|
xml_views = xml_view_generator.generateXML(view_data_list)
|
|
|
|
jobs = parser.jobs
|
|
step = time.time()
|
|
logging.debug('%d XML files generated in %ss',
|
|
len(jobs), str(step - orig))
|
|
|
|
return builder, xml_jobs, xml_views
|
|
|
|
def execute(self, options, jjb_config):
|
|
|
|
if options.n_workers < 0:
|
|
raise JenkinsJobsException(
|
|
'Number of workers must be equal or greater than 0')
|
|
|
|
builder, xml_jobs, xml_views = self._generate_xmljobs(
|
|
options, jjb_config)
|
|
|
|
jobs, num_updated_jobs = builder.update_jobs(
|
|
xml_jobs, n_workers=options.n_workers)
|
|
logger.info("Number of jobs updated: %d", num_updated_jobs)
|
|
|
|
views, num_updated_views = builder.update_views(
|
|
xml_views, n_workers=options.n_workers)
|
|
logger.info("Number of views updated: %d", num_updated_views)
|
|
|
|
keep_jobs = [job.name for job in xml_jobs]
|
|
if options.delete_old:
|
|
n = builder.delete_old_managed(keep=keep_jobs)
|
|
logger.info("Number of jobs deleted: %d", n)
|