tabularize_top_stats/tabularize_top_stats.py at master · c0d3rb4b4/tabularize_top_stats · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import fileinput
import utils
import tableFunctions

def generateFileNames(file_prefix, currentTimestamp):
    cpuFileName = file_prefix + '_cpu_' + currentTimestamp.replace(':', '')
    memFileName = file_prefix + '_mem_' + currentTimestamp.replace(':', '')
    return cpuFileName, memFileName

currentTimestamp = ''
cpuMetDict = {}
memMetDict = {}
config = utils.loadConfig('config.ini')
for line in fileinput.input():
    lineArray = line.strip().split()

    if config.getboolean('cfg', 'save_continous_stream') and (currentTimestamp != '') and (lineArray[0] != currentTimestamp):
        cpuFile, memFile = generateFileNames(config.get('cfg', 'file_prefix'), lineArray[0])
        cpuTable = tableFunctions.metDictToTable(cpuMetDict, config.get('cfg','folder_name'), cpuFile + '.svg')
        memTable = tableFunctions.metDictToTable(memMetDict, config.get('cfg','folder_name'), memFile + '.svg')
        utils.saveToFile(config.get('cfg','folder_name'), cpuFile + '.csv', cpuTable)
        utils.saveToFile(config.get('cfg','folder_name'), memFile + '.csv', memTable)
        print 'Saved for timestamp ' + lineArray[0]

    if lineArray[1] not in cpuMetDict:
        cpuMetDict[lineArray[1]] = {}
        memMetDict[lineArray[1]] = {}
    currentTimestamp = lineArray[0]
    cpuMetDict[lineArray[1]][lineArray[0]] = float(lineArray[2])
    memMetDict[lineArray[1]][lineArray[0]] = float(lineArray[3])

if config['save_static'] and cpuMetDict and memMetDict:
    cpuFile, memFile = generateFileNames(config['file_prefix'], currentTimestamp)
    cpuTable = tableFunctions.metDictToTable(cpuMetDict, config.get('cfg','folder_name'), cpuFile + '.svg')
    memTable = tableFunctions.metDictToTable(memMetDict, config.get('cfg','folder_name'), memFile + '.svg')
    utils.saveToFile(config[folder_name], cpuFile + '.csv', cpuTable)
    utils.saveToFile(config[folder_name], memFile + '.csv', memTable)