import re
import sys

if len(sys.argv) == 2:
    filename = sys.argv[1]
else:
    filename = 'project_planning.xml'

f = file(filename, 'r')

task = re.compile(r'^<task id="wp([0-9\.]+)')
res = re.compile(r'^<use-resource idref="([0-9A-Za-z_]+)" cost="([0-9]+)"')

wpDict = {}
pCountDict ={}
personDict = {}

for line in f:
    match = task.search(line)
    if match:
        current = ('0' + match.group(1))[-2:]
        wpDict[current] = 0
    else:
        match = res.search(line)
        if match:
            months = int(match.group(2))
            person = match.group(1)
            wpDict[current] += months
            try:
                time = personDict[person]
                personDict[person] += months
            except KeyError:
                personDict[person] = months
            try:
                pCountDict[current] += 1
            except KeyError:
                pCountDict[current] = 1

print 'WP   mn  pp'
arr = []
sum = 0
for key, val in wpDict.iteritems():
    arr.append('%-5s%2i%4i' % (key, val, pCountDict[key]))
    sum += val
arr.sort()
for s in arr:
    print s
print 'Total:', sum
print
print
print 'Persons:'
arr = []
for key, val in personDict.iteritems():
    arr.append('%-30s%2i' % (key, val))
arr.sort()
for s in arr:
    print s

pbf = re.compile(r'pbf')
sumPbf = 0
sumOther = 0
for key, val in personDict.iteritems():
    match = pbf.search(key)
    if match:
        sumPbf += val
    else:
        sumOther += val

print
print 'PBF man months, other man months, percentage PBF'
print sumPbf, sumOther, 100.0 * sumPbf/(sumPbf + sumOther)