Tobias Henkel
eca4620efa
When using a branch and pull model on a shared repository there are usually one or more protected branches which are gated and a dynamic number of temporary personal/feature branches which are the source for the pull requests. These temporary branches while ungated can potentially include broken zuul config and therefore break the global tenant wide configuration. In order to deal with this model add support for excluding unprotected branches. This can be configured on tenant level and overridden per project. Change-Id: I8a45fd41539a3c964a84142f04c1644585c0fdcf
106 lines
4.0 KiB
Python
106 lines
4.0 KiB
Python
# Copyright 2012 Hewlett-Packard Development Company, L.P.
|
|
# Copyright 2013 OpenStack Foundation
|
|
# Copyright 2016 Red Hat, Inc.
|
|
#
|
|
# 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
|
|
|
|
from apscheduler.schedulers.background import BackgroundScheduler
|
|
from apscheduler.triggers.cron import CronTrigger
|
|
|
|
from zuul.driver import Driver, TriggerInterface
|
|
from zuul.driver.timer import timertrigger
|
|
from zuul.driver.timer.timermodel import TimerTriggerEvent
|
|
|
|
|
|
class TimerDriver(Driver, TriggerInterface):
|
|
name = 'timer'
|
|
log = logging.getLogger("zuul.TimerDriver")
|
|
|
|
def __init__(self):
|
|
self.apsched = BackgroundScheduler()
|
|
self.apsched.start()
|
|
self.tenant_jobs = {}
|
|
|
|
def registerScheduler(self, scheduler):
|
|
self.sched = scheduler
|
|
|
|
def reconfigure(self, tenant):
|
|
self._removeJobs(tenant)
|
|
if not self.apsched:
|
|
# Handle possible reuse of the driver without connection objects.
|
|
self.apsched = BackgroundScheduler()
|
|
self.apsched.start()
|
|
self._addJobs(tenant)
|
|
|
|
def _removeJobs(self, tenant):
|
|
jobs = self.tenant_jobs.get(tenant.name, [])
|
|
for job in jobs:
|
|
job.remove()
|
|
|
|
def _addJobs(self, tenant):
|
|
jobs = []
|
|
self.tenant_jobs[tenant.name] = jobs
|
|
for pipeline in tenant.layout.pipelines.values():
|
|
for ef in pipeline.manager.event_filters:
|
|
if not isinstance(ef.trigger, timertrigger.TimerTrigger):
|
|
continue
|
|
for timespec in ef.timespecs:
|
|
parts = timespec.split()
|
|
if len(parts) < 5 or len(parts) > 6:
|
|
self.log.error(
|
|
"Unable to parse time value '%s' "
|
|
"defined in pipeline %s" % (
|
|
timespec,
|
|
pipeline.name))
|
|
continue
|
|
minute, hour, dom, month, dow = parts[:5]
|
|
if len(parts) > 5:
|
|
second = parts[5]
|
|
else:
|
|
second = None
|
|
trigger = CronTrigger(day=dom, day_of_week=dow, hour=hour,
|
|
minute=minute, second=second)
|
|
|
|
job = self.apsched.add_job(
|
|
self._onTrigger, trigger=trigger,
|
|
args=(tenant, pipeline.name, timespec,))
|
|
jobs.append(job)
|
|
|
|
def _onTrigger(self, tenant, pipeline_name, timespec):
|
|
for project_name in tenant.layout.project_configs.keys():
|
|
(trusted, project) = tenant.getProject(project_name)
|
|
for branch in project.source.getProjectBranches(project, tenant):
|
|
event = TimerTriggerEvent()
|
|
event.type = 'timer'
|
|
event.timespec = timespec
|
|
event.forced_pipeline = pipeline_name
|
|
event.project_hostname = project.canonical_hostname
|
|
event.project_name = project.name
|
|
event.ref = 'refs/heads/%s' % branch
|
|
event.branch = branch
|
|
self.log.debug("Adding event %s" % event)
|
|
self.sched.addEvent(event)
|
|
|
|
def stop(self):
|
|
if self.apsched:
|
|
self.apsched.shutdown()
|
|
self.apsched = None
|
|
|
|
def getTrigger(self, connection_name, config=None):
|
|
return timertrigger.TimerTrigger(self, config)
|
|
|
|
def getTriggerSchema(self):
|
|
return timertrigger.getSchema()
|