Files
cproof-plugins/jenkins.py
2014-06-09 22:35:42 +01:00

390 lines
14 KiB
Python

"""
Manage jenkins jobs
Prerequisites:
The jenkinsapi module is required https://pypi.python.org/pypi/jenkinsapi
To install the module using pip:
sudo pip install jenkinsapi
The plugin polls the jenkins server at 'jenkins_url' every 'jenkins_poll_interval' seconds.
The results are checked by profanity every 'prof_cb_interval'.
New failures are reported in the jenkins window, and a desktop notification is sent.
The 'prof_remind_interval' specifies the time between reminder notifications of broken builds
The following commands are available:
/jenkins list - list all jobs
/jenkins build [job] - trigger a build for the job
/jenkins open [job] - open the job in the systems default browser
/jenkins remind on|off - Enable or disable reminder notifications
/jenkins notify on|off - Enable or disable build status change notifications
/jenkins settings - Show current settings
/jenkins help - Show help
"""
import prof
import os
import threading
import time
import webbrowser
import urllib2
import jenkinsapi
from jenkinsapi.jenkins import Jenkins
jenkins_url = "http://localhost:8080"
username = None
password = None
jenkins_poll_interval = 5
prof_cb_interval = 5
prof_remind_interval = 60 * 15
enable_notify = True
enable_remind = False
last_state = {}
STATE_SUCCESS = "SUCCESS"
STATE_UNSTABLE = "UNSTABLE"
STATE_FAILURE = "FAILURE"
STATE_QUEUED = "QUEUED"
STATE_RUNNING = "RUNNING"
STATE_NOBUILDS = "NOBUILDS"
STATE_UNKNOWN = "UNKNOWN"
win_tag = "Jenkins"
poll_fail = False
poll_fail_message = None
job_list = None
changes_list = None
def _safe_remove(jobname, state):
if jobname in last_state[state]:
last_state[state].remove(jobname)
def _set_state(jobname, state):
if not jobname in last_state[state]:
_safe_remove(jobname, STATE_SUCCESS)
_safe_remove(jobname, STATE_UNSTABLE)
_safe_remove(jobname, STATE_FAILURE)
_safe_remove(jobname, STATE_QUEUED)
_safe_remove(jobname, STATE_RUNNING)
_safe_remove(jobname, STATE_NOBUILDS)
_safe_remove(jobname, STATE_UNKNOWN)
last_state[state].append(jobname)
return True
else:
return False
def _show_help():
prof.win_show(win_tag, "Commands:")
prof.win_show(win_tag, " /jenkins help - Show this help")
prof.win_show(win_tag, " /jenkins list - List all jobs")
prof.win_show(win_tag, " /jenkins show [job] - Details of specific job")
prof.win_show(win_tag, " /jenkins build [job] - Trigger build for job")
prof.win_show(win_tag, " /jenkins open [job] - Open job in browser")
prof.win_show(win_tag, "")
def _open_job_url(url):
savout = os.dup(1)
saverr = os.dup(2)
os.close(1)
os.close(2)
os.open(os.devnull, os.O_RDWR)
try:
webbrowser.open(url, new=2)
finally:
os.dup2(savout, 1)
os.dup2(saverr, 2)
class JobList():
def __init__(self):
self.jobs = []
def add_job(self, name, build_number, state):
self.jobs.append((name, build_number, state))
def get_jobs(self):
return self.jobs
def contains_job(self, jobname):
for name, build_number, state in self.jobs:
if name == jobname:
return True
return False
def num_failing(self):
count = 0
for name, build_number, state in self.jobs:
if state == "FAILURE":
count = count + 1
return count
def num_unstable(self):
count = 0
for name, build_number, state in self.jobs:
if state == "UNSTABLE":
count = count + 1
return count
class JobUpdates():
def __init__(self):
self.failing = []
self.passing = []
self.unstable = []
self.queued = []
self.running = []
def add_failing(self, name, build_number):
self.failing.append((name, build_number))
def get_failing(self):
return self.failing
def add_passing(self, name, build_number):
self.passing.append((name, build_number))
def get_passing(self):
return self.passing
def add_unstable(self, name, build_number):
self.unstable.append((name, build_number))
def get_unstable(self):
return self.unstable
def add_queued(self, name):
self.queued.append(name)
def get_queued(self):
return self.queued
def add_running(self, name):
self.running.append(name)
def get_running(self):
return self.running
def _jenkins_poll():
global poll_fail
global poll_fail_message
global job_list
global changes_list
while True:
time.sleep(jenkins_poll_interval)
try:
j = Jenkins(jenkins_url, username, password)
except Exception, e:
poll_fail = True
poll_fail_message = str(e)
else:
poll_fail = False
poll_fail_message = None
new_job_list = JobList()
new_changes_list = JobUpdates()
for name, job in j.get_jobs():
if not job.is_queued_or_running():
build = job.get_last_build_or_none()
if build:
if build.get_status() == "FAILURE":
new_job_list.add_job(name, build.get_number(), "FAILURE")
changed = _set_state(name, STATE_FAILURE)
if changed:
new_changes_list.add_failing(name, build.get_number())
elif build.get_status() == "SUCCESS":
new_job_list.add_job(name, build.get_number(), "SUCCESS")
changed = _set_state(name, STATE_SUCCESS)
if changed:
new_changes_list.add_passing(name, build.get_number())
elif build.get_status() == "UNSTABLE":
new_job_list.add_job(name, build.get_number(), "UNSTABLE")
changed = _set_state(name, STATE_UNSTABLE)
if changed:
new_changes_list.add_unstable(name, build.get_number())
else:
new_job_list.add_job(name, None, "NOBUILDS")
else:
if job.is_queued():
new_job_list.add_job(name, None, "QUEUED")
changed = _set_state(name, STATE_QUEUED)
if changed:
new_changes_list.add_queued(name)
elif job.is_running():
new_job_list.add_job(name, None, "RUNNING")
changed = _set_state(name, STATE_RUNNING)
if changed:
new_changes_list.add_running(name)
job_list = new_job_list
changes_list = new_changes_list
def _prof_callback():
global changes_list
if poll_fail:
prof.win_show(win_tag, "Could not connect to jenkins, see the logs.")
if poll_fail_message:
prof.log_warning("Jenkins poll failed: " + str(poll_fail_message))
else:
prof.log_warning("Jenkins poll failed")
elif changes_list:
for name in changes_list.get_queued():
prof.win_show_cyan(win_tag, name + " QUEUED")
for name in changes_list.get_running():
prof.win_show_cyan(win_tag, name + " RUNNING")
for name, build_number in changes_list.get_passing():
prof.win_show_green(win_tag, name + " #" + str(build_number) + " SUCCESS")
if enable_notify:
prof.notify(name + " SUCCESS", 5000, "Jenkins")
for name, build_number in changes_list.get_unstable():
prof.win_show_yellow(win_tag, name + " #" + str(build_number) + " UNSTABLE")
if enable_notify:
prof.notify(name + " UNSTABLE", 5000, "Jenkins")
for name, build_number in changes_list.get_failing():
prof.win_show_red(win_tag, name + " #" + str(build_number) + " FAILURE")
if enable_notify:
prof.notify(name + " FAILURE", 5000, "Jenkins")
changes_list = None
def _handle_input(win, line):
prof.win_show(win_tag, "Handled input.")
def _cmd_jenkins(cmd=None, arg=None):
global enable_remind
global enable_notify
if not prof.win_exists(win_tag):
prof.win_create(win_tag, _handle_input)
prof.win_focus(win_tag)
if cmd == "list":
if job_list and job_list.get_jobs():
prof.win_show(win_tag, "Jobs:")
for name, build_number, state in job_list.get_jobs():
if state == "SUCCESS":
prof.win_show_green(win_tag, " " + name + " #" + str(build_number) + " SUCCESS")
elif state == "UNSTABLE":
prof.win_show_yellow(win_tag, " " + name + " #" + str(build_number) + " UNSTABLE")
elif state == "FAILURE":
prof.win_show_red(win_tag, " " + name + " #" + str(build_number) + " FAILURE")
elif state == "NOBUILDS":
prof.win_show(win_tag, " " + name + ", no builds")
else:
prof.win_show_cyan(win_tag, " " + name + " " + state)
else:
prof.win_show(win_tag, "No job data yet.")
elif cmd == "build":
if not arg:
prof.win_show(win_tag, "You must supply a job argument.")
elif job_list and job_list.contains_job(arg):
try:
urllib2.urlopen(jenkins_url + "/job/" + arg + "/build")
except Exception, e:
prof.win_show(win_tag, "Failed to build " + arg + ", see the logs.")
prof.log_warning("Failed to build " + arg + ": " + str(e))
else:
prof.win_show(win_tag, "Build request sent for " + arg)
else:
prof.win_show(win_tag, "No such job: " + arg)
elif cmd == "open":
if not arg:
prof.win_show(win_tag, "You must supply a job argument.")
elif job_list and job_list.contains_job(arg):
_open_job_url(jenkins_url + "/job/" + arg)
else:
prof.win_show(win_tag, "No such job: " + arg)
elif cmd == "remind":
if not arg:
prof.win_show(win_tag, "You must specify either 'on' or 'off'.")
elif arg == "on":
enable_remind = True
prof.win_show(win_tag, "Reminder notifications enabled.")
elif arg == "off":
enable_remind = False
prof.win_show(win_tag, "Reminder notifications disabled.")
else:
prof.win_show(win_tag, "You must specify either 'on' or 'off'.")
elif cmd == "notify":
if not arg:
prof.win_show(win_tag, "You must specify either 'on' or 'off'.")
elif arg == "on":
enable_notify = True
prof.win_show(win_tag, "Build notifications enabled.")
elif arg == "off":
enable_notify = False
prof.win_show(win_tag, "Build notifications disabled.")
else:
prof.win_show(win_tag, "You must specify either 'on' or 'off'.")
elif cmd == "settings":
prof.win_show(win_tag, "Jenkins settings:")
prof.win_show(win_tag, " Jenkins URL : " + jenkins_url)
prof.win_show(win_tag, " Jenkins poll interval : " + str(jenkins_poll_interval) + " seconds")
prof.win_show(win_tag, " Profanity update interval : " + str(prof_cb_interval) + " seconds")
prof.win_show(win_tag, " Reminder interval : " + str(prof_remind_interval) + " seconds")
prof.win_show(win_tag, " Notifications enabled : " + str(enable_notify))
prof.win_show(win_tag, " Reminders enabled : " + str(enable_remind))
elif cmd == "help":
prof.win_show(win_tag, "Commands:")
prof.win_show(win_tag, " /jenkins help - Show this help")
prof.win_show(win_tag, " /jenkins list - List all jobs")
prof.win_show(win_tag, " /jenkins build [job] - Trigger build for job")
prof.win_show(win_tag, " /jenkins open [job] - Open job in browser")
prof.win_show(win_tag, " /jenkins remind on|off - Enable/disable reminder notifications")
prof.win_show(win_tag, " /jenkins notify on|off - Enable/disable build notifications")
prof.win_show(win_tag, " /jenkins settings - Show current settings")
else:
prof.win_show(win_tag, "Unknown command.")
def _remind():
if enable_remind and job_list:
notify_string = ""
failing = job_list.num_failing()
unstable = job_list.num_unstable()
if failing == 1:
notify_string = notify_string + "1 failing build"
if failing > 1:
notify_string = notify_string + str(failing) + " failing builds"
if failing > 0 and unstable > 0:
notify_string = notify_string + "\n"
if unstable == 1:
notify_string = notify_string + "1 unstable build"
if unstable > 1:
notify_string = notify_string + str(unstable) + " unstable builds"
if not notify_string == "":
prof.notify(notify_string, 5000, "Jenkins")
def prof_init(version, status):
last_state[STATE_SUCCESS] = []
last_state[STATE_UNSTABLE] = []
last_state[STATE_FAILURE] = []
last_state[STATE_QUEUED] = []
last_state[STATE_RUNNING] = []
last_state[STATE_NOBUILDS] = []
last_state[STATE_UNKNOWN] = []
jenkins_t = threading.Thread(target=_jenkins_poll)
jenkins_t.daemon = True;
jenkins_t.start()
prof.register_timed(_prof_callback, prof_cb_interval)
prof.register_timed(_remind, prof_remind_interval)
prof.register_command("/jenkins", 0, 2, "/jenkins list|build|open|remind|notify|settings|help", "Do jenkins stuff.", "Do jenkins stuff.",
_cmd_jenkins)
def prof_on_start():
prof.win_create(win_tag, _handle_input)
prof.win_show(win_tag, "Jenkins plugin started.")