diff options
Diffstat (limited to 'vendor/github.com/alecthomas/chroma/table.py')
-rw-r--r-- | vendor/github.com/alecthomas/chroma/table.py | 21 |
1 files changed, 19 insertions, 2 deletions
diff --git a/vendor/github.com/alecthomas/chroma/table.py b/vendor/github.com/alecthomas/chroma/table.py index 1caa7cc569..7794539821 100644 --- a/vendor/github.com/alecthomas/chroma/table.py +++ b/vendor/github.com/alecthomas/chroma/table.py @@ -1,8 +1,12 @@ #!/usr/bin/env python3 +import re from collections import defaultdict from subprocess import check_output -lines = check_output(["go", "run", "./cmd/chroma/main.go", "--list"]).decode('utf-8').splitlines() +README_FILE = "README.md" + + +lines = check_output(["go", "run", "./cmd/chroma/main.go", "--list"]).decode("utf-8").splitlines() lines = [line.strip() for line in lines if line.startswith(" ") and not line.startswith(" ")] lines = sorted(lines, key=lambda l: l.lower()) @@ -11,5 +15,18 @@ table = defaultdict(list) for line in lines: table[line[0].upper()].append(line) +rows = [] for key, value in table.items(): - print("{} | {}".format(key, ', '.join(value))) + rows.append("{} | {}".format(key, ", ".join(value))) +tbody = "\n".join(rows) + +with open(README_FILE, "r") as f: + content = f.read() + +with open(README_FILE, "w") as f: + marker = re.compile(r"(?P<start>:----: \\| --------\n).*?(?P<end>\n\n)", re.DOTALL) + replacement = r"\g<start>%s\g<end>" % tbody + updated_content = marker.sub(replacement, content) + f.write(updated_content) + +print(tbody) |