Generate list of prometheus metrics in reStructuredText rather than markdown

This commit is contained in:
Rajat Goyal
2020-09-30 23:46:41 +05:30
parent 5b8a7fd191
commit b194135a0f

View File

@@ -1,7 +1,7 @@
import os import os
from os.path import join from os.path import join
from pytablewriter import MarkdownTableWriter from pytablewriter import RstSimpleTableWriter
from pytablewriter.style import Style from pytablewriter.style import Style
import jupyterhub.metrics import jupyterhub.metrics
@@ -12,7 +12,7 @@ HERE = os.path.abspath(os.path.dirname(__file__))
class Generator: class Generator:
@classmethod @classmethod
def create_writer(cls, table_name, headers, values): def create_writer(cls, table_name, headers, values):
writer = MarkdownTableWriter() writer = RstSimpleTableWriter()
writer.table_name = table_name writer.table_name = table_name
writer.headers = headers writer.headers = headers
writer.value_matrix = values writer.value_matrix = values
@@ -34,13 +34,17 @@ class Generator:
if not os.path.exists(generated_directory): if not os.path.exists(generated_directory):
os.makedirs(generated_directory) os.makedirs(generated_directory)
filename = f"{generated_directory}/metrics.md" filename = f"{generated_directory}/metrics.rst"
table_name = "List of Prometheus Metrics" table_name = ""
headers = ["Type", "Name", "Description"] headers = ["Type", "Name", "Description"]
values = self._parse_metrics() values = self._parse_metrics()
writer = self.create_writer(table_name, headers, values) writer = self.create_writer(table_name, headers, values)
title = "List of Prometheus Metrics"
underline = "============================"
content = f"{title}\n{underline}\n{writer.dumps()}"
with open(filename, 'w') as f: with open(filename, 'w') as f:
f.write(writer.dumps()) f.write(content)
def main(): def main():