annotate vendor/github.com/alecthomas/chroma/v2/table.py @ 66:787b5ee0289d draft

Use vendored modules Signed-off-by: Izuru Yakumo <yakumo.izuru@chaotic.ninja>
author yakumo.izuru
date Sun, 23 Jul 2023 13:18:53 +0000
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
66
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
1 #!/usr/bin/env python3
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
2 import re
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
3 from collections import defaultdict
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
4 from subprocess import check_output
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
5
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
6 README_FILE = "README.md"
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
7
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
8
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
9 lines = check_output(["go", "run", "./cmd/chroma/main.go", "--list"]).decode("utf-8").splitlines()
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
10 lines = [line.strip() for line in lines if line.startswith(" ") and not line.startswith(" ")]
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
11 lines = sorted(lines, key=lambda l: l.lower())
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
12
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
13 table = defaultdict(list)
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
14
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
15 for line in lines:
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
16 table[line[0].upper()].append(line)
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
17
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
18 rows = []
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
19 for key, value in table.items():
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
20 rows.append("{} | {}".format(key, ", ".join(value)))
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
21 tbody = "\n".join(rows)
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
22
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
23 with open(README_FILE, "r") as f:
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
24 content = f.read()
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
25
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
26 with open(README_FILE, "w") as f:
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
27 marker = re.compile(r"(?P<start>:----: \\| --------\n).*?(?P<end>\n\n)", re.DOTALL)
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
28 replacement = r"\g<start>%s\g<end>" % tbody
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
29 updated_content = marker.sub(replacement, content)
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
30 f.write(updated_content)
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
31
787b5ee0289d Use vendored modules
yakumo.izuru
parents:
diff changeset
32 print(tbody)