Basic kv_extract text utility

kv_extract.py

import re
import sys

keys = sys.argv[1:]
kv_re = re.compile(r'(?P<key>[^ =]+)=(?P<value>\S+)')

def strip_comma(s):
    if s[-1] == ',':
        return s[:-1]
    return s

for line in sys.stdin:
    line = line.strip()
    d = dict([(m.group('key'), strip_comma(m.group('value')))
                  for m in kv_re.finditer(line)])
    vs = [d.get(k, 'None') for k in keys]
    print '\t'.join(vs)