diff --git a/docs/generate-metrics.py b/docs/generate-metrics.py index 05cc6c35..00551380 100644 --- a/docs/generate-metrics.py +++ b/docs/generate-metrics.py @@ -1,7 +1,6 @@ import os -from pytablewriter import RstSimpleTableWriter -from pytablewriter.style import Style +from pytablewriter import MarkdownTableWriter import jupyterhub.metrics @@ -11,12 +10,11 @@ HERE = os.path.abspath(os.path.dirname(__file__)) class Generator: @classmethod def create_writer(cls, table_name, headers, values): - writer = RstSimpleTableWriter() + writer = MarkdownTableWriter() writer.table_name = table_name writer.headers = headers writer.value_matrix = values writer.margin = 1 - [writer.set_style(header, Style(align="center")) for header in headers] return writer def _parse_metrics(self): @@ -33,18 +31,17 @@ class Generator: if not os.path.exists(generated_directory): os.makedirs(generated_directory) - filename = f"{generated_directory}/metrics.rst" + filename = f"{generated_directory}/metrics.md" table_name = "" headers = ["Type", "Name", "Description"] values = self._parse_metrics() 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: - f.write(content) - print(f"Generated {filename}.") + f.write("# List of Prometheus Metrics\n\n") + f.write(writer.dumps()) + f.write("\n") + print(f"Generated {filename}") def main():