Created
July 18, 2018 20:24
-
-
Save aneeshkp/3b8c40faef4db983b89ecd74d429764f to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import csv | |
import requests | |
import sys | |
def GetMetrixNames(url): | |
response = requests.get('{0}/api/v1/label/__name__/values'.format(url)) | |
results = response.json()['data'] | |
#Return metrix | |
return results | |
""" | |
A simple program to print the result of a Prometheus query as CSV. | |
""" | |
writer = csv.writer(sys.stdout) | |
# Write the header, | |
if len(sys.argv) != 2: | |
print('Usage: {0} http://prometheus:9090'.format(sys.argv[0])) | |
sys.exit(1) | |
metrixResults=GetMetrixNames(sys.argv[1]) | |
writeHeader=True | |
for metrixResult in metrixResults: | |
if metrixResult.startswith("collectd"): | |
response = requests.get('{0}/api/v1/query'.format(sys.argv[1]), | |
params={'query': metrixResult+'[1h]'}) | |
results = response.json()['data']['result'] | |
# Build a list of all labelnames used. | |
#gets all keys and discard __name__ | |
labelnames = set() | |
for result in results: | |
labelnames.update(result['metric'].keys()) | |
# Canonicalize | |
labelnames.discard('__name__') | |
labelnames = sorted(labelnames) | |
# Write the samples. | |
if writeHeader: | |
writer.writerow(['name', 'timestamp', 'value'] + labelnames) | |
writeHeader=False | |
for result in results: | |
l = [result['metric'].get('__name__', '')] + result['values'] | |
for label in labelnames: | |
l.append(result['metric'].get(label, '')) | |
writer.writerow(l) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment