jenkins-job-builder/jenkins_jobs/cli/subcommand/base.py

136 lines
4.3 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 abc
import fnmatch
import logging
import time
from jenkins_jobs.builder import JenkinsManager
from jenkins_jobs.registry import ModuleRegistry
from jenkins_jobs.roots import Roots
from jenkins_jobs.xml_config import XmlJobGenerator
from jenkins_jobs.xml_config import XmlViewGenerator
from jenkins_jobs.loader import load_files
logger = logging.getLogger(__name__)
def matches(name, glob_list):
"""
Checks if the given string, ``name``, matches any of the glob patterns in
the iterable, ``glob_list``
:arg str name: String (job or view name) to test if it matches a pattern
:arg iterable glob_list: glob patterns to match (list, tuple, set, etc.)
"""
return any(fnmatch.fnmatch(name, glob) for glob in glob_list)
def filter_matching(item_list, glob_list):
if not glob_list:
return item_list
return [item for item in item_list if matches(item["name"], glob_list)]
class BaseSubCommand(metaclass=abc.ABCMeta):
"""Base class for Jenkins Job Builder subcommands, intended to allow
subcommands to be loaded as stevedore extensions by third party users.
"""
def __init__(self):
pass
@abc.abstractmethod
def parse_args(self, subparsers, recursive_parser):
"""Define subcommand arguments.
:param subparsers
A sub parser object. Implementations of this method should
create a new subcommand parser by calling
parser = subparsers.add_parser('command-name', ...)
This will return a new ArgumentParser object; all other arguments to
this method will be passed to the argparse.ArgumentParser constructor
for the returned object.
"""
@abc.abstractmethod
def execute(self, config):
"""Execute subcommand behavior.
:param config
JJBConfig object containing final configuration from config files,
command line arguments, and environment variables.
"""
@staticmethod
def parse_option_recursive_exclude(parser):
"""Add '--recursive' and '--exclude' arguments to given parser."""
parser.add_argument(
"-r",
"--recursive",
action="store_true",
dest="recursive",
default=False,
help="look for yaml files recursively",
)
parser.add_argument(
"-x",
"--exclude",
dest="exclude",
action="append",
default=[],
help="paths to exclude when using recursive search, "
"uses standard globbing.",
)
class JobsSubCommand(BaseSubCommand):
"""Base class for Jenkins Job Builder subcommands which generates jobs."""
def load_roots(self, jjb_config, path_list):
roots = Roots(jjb_config)
load_files(jjb_config, roots, path_list)
return roots
def make_jobs_and_views_xml(self, jjb_config, path_list, glob_list):
logger.info("Updating jobs in {0} ({1})".format(path_list, glob_list))
orig = time.time()
roots = self.load_roots(jjb_config, path_list)
builder = JenkinsManager(jjb_config)
registry = ModuleRegistry(jjb_config, builder.plugins_list)
registry.set_macros(roots.macros)
jobs = filter_matching(roots.generate_jobs(), glob_list)
views = filter_matching(roots.generate_views(), glob_list)
registry.amend_job_dicts(jobs)
xml_job_generator = XmlJobGenerator(registry)
xml_view_generator = XmlViewGenerator(registry)
xml_jobs = xml_job_generator.generateXML(jobs)
xml_views = xml_view_generator.generateXML(views)
step = time.time()
logging.debug("%d XML files generated in %ss", len(jobs), str(step - orig))
return builder, xml_jobs, xml_views