Skip to content

Instantly share code, notes, and snippets.

@punchagan punchagan/issue_plot.py
Last active Oct 11, 2015

Embed
What would you like to do?
Simple script to plot Github turn around time
#!/usr/bin/env python
from github import Github
from datetime import datetime
from pylab import plot, show, figure, legend, title, ylabel
def set_repo_name(repo_name, username):
if '/' in repo_name:
return repo_name
else:
return "%s/%s" %(username, repo_name)
def get_repo(username, password, full_reponame):
print 'Fetching repository'
g = Github(username, password)
user, reponame = full_reponame.split('/')
user = g.get_user(user)
return user.get_repo(reponame)
def get_issues(repo, states=('closed', 'open')):
issues = dict()
for state in states:
print 'Fetching %s issues' % state
issues[state] = list(repo.get_issues(state=state))
return issues
def get_first_user_comment(issue, author=''):
#print 'Fetching first comment date for', issue.number
for comment in issue.get_comments():
if not author or comment.user.login == author:
return comment.created_at
def plot_issues(issues, author):
""" Issues is a dict with keys as states and values as a list of issues.
"""
figure(figsize=(12,9))
all_issues = []
for state in issues:
for issue in issues[state]:
#print 'Plotting issue', issue.number
start = issue.created_at
if state == 'closed':
end = issue.closed_at
color = 'g'
elif issue.comments:
end = get_first_user_comment(issue, author)
color = 'r'
else:
end = None
if not end:
color = 'k'
end = datetime.today()
plot([start, end], [issue.number]*2, color, linewidth=2)
all_issues.append((issue, start, end))
return all_issues
def mark_work_days(all_issues):
""" If three or more issues were fixed on a day, draw a vertical line
"""
FMT = '%Y-%m-%d'
n = len(all_issues)
ends = [datetime.strptime(i.strftime(FMT), FMT) for _, _, i in all_issues]
days = [day for day in ends if ends.count(day) >= 3]
for day in days:
plot([day]*n, range(n), 'b--')
def add_legend():
import matplotlib.patches as mpatches
r = mpatches.Patch(color='r', label='Time to first comment (open issues)')
g = mpatches.Patch(color='g', label='Time to close issue/PR')
k = mpatches.Patch(color='k', label='Time open with no comment')
b = mpatches.Patch(color='b', label='Work days (> 3 issues closed)')
legend(handles=[r, g, k, b], loc=2)
title('%s turnaround' % reponame)
ylabel('Issues/PRs')
if __name__ == "__main__":
from sys import argv, exit
from getpass import getpass
if len(argv) < 3:
print "Usage: %s username reponame" % argv[0]
print "reponame can be of the form <author>/<repository>"
exit(1)
username = argv[1]
reponame = set_repo_name(argv[2], username)
author = reponame.split('/')[0]
password = getpass('Github password for %s: ' % username)
repo = get_repo(username, password, reponame)
issues = get_issues(repo)
all_issues = plot_issues(issues, author)
mark_work_days(all_issues)
add_legend()
show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.