def extractData (source):
'''
Extracts data from a single trace log, or from a directory of compressed
game logs using TournamentIterator
'''
if os.path.isdir(source):
# in this case, we use the iterator
for trace in ti.traceLogIter(source):
traceName = str(trace)
dataName = 'data/pp-data.csv'
m = gameRe.search(traceName)
if m:
dataName = 'data/pp-data-{}.csv'.format(m.group(1))
else:
print('Could not extract game ID from {}'.format(traceName))
extractFile(traceName, dataName)
else:
extractFile(source, source + '-pp.csv')
