import urllib, re, sys, optparse, os.path, datetime
import sqlite3 as dbapi2
+from xml.sax.saxutils import escape
blueprints_base_url = 'https://blueprints.launchpad.net'
cur = db.cursor()
cur.execute('''CREATE TABLE work_items (
blueprint VARCHAR(255) NOT NULL,
- workitem VARCHAR(255) NOT NULL,
+ workitem VARCHAR(1000) NOT NULL,
status VARCHAR(20) NOT NULL,
+ assignee VARCHAR(200) NOT NULL,
date TIMESTAMP NOT NULL)''')
cur.execute('''CREATE TABLE status (
blueprint VARCHAR(255) NOT NULL,
help='Path to database', dest='database', metavar='PATH')
optparser.add_option('-r', '--release',
help='Release name', dest='release')
+ optparser.add_option('--milestone',
+ help='Milestone name', dest='milestone')
optparser.add_option('-p', '--pattern', metavar='REGEX',
help='Regex pattern for blueprint name', dest='pattern')
optparser.add_option('-D', '--dump', action='store_true', default=False,
return (opts, args)
-def get_blueprints(url, name_pattern):
+def get_blueprints(url, name_pattern, milestone):
'''Return a list of blueprint URLs for the current release.'''
blueprint_name_filter = re.compile('href="(/ubuntu/\+spec/%s[^"]+)"' %
name_pattern)
result = []
+ scan_tr_end = False
+ found_ms = False
+ bp = None
for l in urllib.urlopen(url):
- m = blueprint_name_filter.search(l)
- if m:
- result.append(blueprints_base_url + m.group(1))
+ if scan_tr_end:
+ if milestone:
+ if ('/+milestone/%s"' % milestone) in l:
+ found_ms = True
+ if '</tr>' in l:
+ scan_tr_end = False
+ if bp and (not milestone or found_ms):
+ result.append(bp)
+ bp = None
+ found_ms = False
+ else:
+ m = blueprint_name_filter.search(l)
+ if m:
+ bp = blueprints_base_url + m.group(1)
+ scan_tr_end = True
return result
def get_blueprint_workitems(blueprint_url):
'''Collect work items from a particular blueprint URL.
- This will return a list of ('item', 'status') pairs.
+ This will return a list of ('item', 'status', 'assignee') tuples.
'''
work_items_re = re.compile('(<p>|^)work items:\s*<br />', re.I)
+ assignee_re = re.compile('<a href="https://.*launchpad.net/~([a-zA-Z0-9_-]+)" class=".*person">')
found_workitems = False
+ found_assignee = False
result = []
+ default_assignee = 'nobody'
for l in urllib.urlopen(blueprint_url):
end = False
+
+ if '<dt>Assignee:' in l:
+ found_assignee = True
+ continue
+
+ if found_assignee and not found_workitems:
+ m = assignee_re.search(l)
+ if m:
+ default_assignee = m.group(1)
+ found_assignee = False
+
if not found_workitems:
if work_items_re.search(l):
found_workitems = True
if '</p>' in l:
end = True
- l = l.replace('<br />', '').replace('</div>', '').replace('</p>', '').strip()
+ l = l.replace('<br />', '').replace('</div>', '').replace('</p>', '').replace('<wbr></wbr>', '').strip()
if not l:
break
print >> sys.stderr, 'ERROR: invalid state "%s" for work item "%s"' % (
state, desc)
continue
- result.append((desc, state))
+
+ if desc.startswith('['):
+ try:
+ off = desc.index(']')
+ assignee = desc[1:off]
+ desc = desc[off+1:].strip()
+ except ValueError:
+ print >> sys.stderr, 'ERROR: missing closing "]" for assignee for work item "%s"' % desc
+ else:
+ assignee = default_assignee
+
+ result.append((desc, state, assignee))
if end:
break
cur = db.cursor()
cur.execute('SELECT * FROM work_items')
print '== Work items: =='
- for (blueprint, workitem, item_status, date) in cur:
- print '%s [%s]\t%s: %s' % (date, blueprint, workitem, item_status)
+ for (blueprint, workitem, item_status, assignee, date) in cur:
+ print '%s [%s, %s]\t%s: %s' % (date, blueprint, assignee, workitem, item_status)
print '\n== Status =='
cur = db.cursor()
for (blueprint, status, date) in cur:
print '%s: %s [%s]' % (blueprint, status, date)
-def add_work_item(db, blueprint, item, status):
+def add_work_item(db, blueprint, item, status, assignee):
'''Add work item to database.'''
cur = db.cursor()
- cur.execute('INSERT INTO work_items VALUES (?, ?, ?, date(CURRENT_TIMESTAMP))',
- (blueprint, item, status))
+ cur.execute('INSERT INTO work_items VALUES (?, ?, ?, ?, date(CURRENT_TIMESTAMP))',
+ (blueprint, item, status, assignee))
def add_status(db, blueprint, status):
'''Add blueprint status to database.'''
cur.execute('INSERT INTO status VALUES (?, ?, date(CURRENT_TIMESTAMP))',
(blueprint, status))
-def import_lp(db, name_pattern, release):
+def import_lp(db, name_pattern, release, milestone):
'''Collect blueprint work items and status from Launchpad into DB.'''
blueprints = get_blueprints('%s//ubuntu/%s/+specs?batch=300' % (blueprints_base_url,
- opts.release), name_pattern)
+ release), name_pattern, milestone)
cur = db.cursor()
cur.execute('DELETE FROM work_items WHERE date = date(CURRENT_TIMESTAMP)')
status = get_blueprint_status(bp)
if not work_items:
print >> sys.stderr, 'WARNING: %s has no work items' % bpname
- for (item, state) in work_items:
- add_work_item(db, bpname, item, state)
+ for (item, state, assignee) in work_items:
+ add_work_item(db, bpname, item, state, assignee)
add_status(db, bpname, status)
def workitems_over_time(db):
return data
+def assignee_completion(db):
+ '''Determine current by-assignee completion.
+
+ Return assignee -> [todo, done, postponed] mapping.
+ '''
+ data = {}
+
+ # last date
+ cur = db.cursor()
+ cur.execute('SELECT max(date) FROM work_items')
+ (last_date,) = cur.fetchone()
+
+ index = 0
+ for s in valid_states:
+ cur = db.cursor()
+ cur.execute('SELECT assignee, count(workitem) FROM work_items '
+ 'WHERE date=? and status=? GROUP BY assignee',
+ (last_date, s))
+ for (a, num) in cur:
+ data.setdefault(a, [0, 0, 0])[index] = num
+ index += 1
+
+ return data
+
def text(db):
'''Print work item completion as text.'''
completion.sort(key=lambda k: k[1], reverse=True)
for (bp, percent) in completion:
- print ' <tr><td><a href="%s/ubuntu/+spec/%s">%s</a></td> <td>%i/%i/%i</td> <td>%i%%</td> <td>%s</td></tr>' % (
- blueprints_base_url, bp, bp, data[bp][0], data[bp][2],
+ if bp.startswith('http:'):
+ url = bp
+ else:
+ url = '%s/ubuntu/+spec/%s' % (blueprints_base_url, escape(bp))
+ print ' <tr><td><a href="%s">%s</a></td> <td>%i/%i/%i</td> <td>%i%%</td> <td>%s</td></tr>' % (
+ url, escape(bp), data[bp][0], data[bp][2],
data[bp][1], percent,
- data[bp][-1])
+ escape(data[bp][-1]))
+
+ print '</table>'
+
+ print '''
+<h1>Status by assignee</h1>
+<table>
+ <tr><th>Assignee</th> <th>todo/postponed/done</th> <th>Completion</th></tr>
+'''
+ data = assignee_completion(db)
+
+ completion = []
+ for (a, (todo, done, postponed)) in data.iteritems():
+ completion.append((a,
+ int(float(postponed+done)/(todo+done+postponed)*100 + 0.5)))
+
+ completion.sort(key=lambda k: k[0], reverse=False)
+ for (a, percent) in completion:
+ url = '%s/~%s/+specs?role=assignee' % (blueprints_base_url, a)
+ print ' <tr><td><a href="%s">%s</a></td> <td>%i/%i/%i</td> <td>%i%%</td></tr>' % (
+ url, escape(a), data[a][0], data[a][2],
+ data[a][1], percent)
print '</table>'
print '</body></html>'
for url in urls:
for (d, s) in get_moin_workitems(url):
- add_work_item(db, url, d, s)
+ add_work_item(db, url, d, s, 'nobody')
#
# main
elif opts.csv:
csv(db, opts.from_date, opts.to_date)
else:
- import_lp(db, opts.pattern, opts.release)
+ import_lp(db, opts.pattern, opts.release, opts.milestone)
import_moin(db, opts.moin)
db.commit()