zulip/tools/setup/build_pygments_data

37 lines
1.0 KiB
Python
Executable File

#!/usr/bin/env python3
import json
import os
from pygments.lexers import get_all_lexers
ZULIP_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../../")
# The current priorities data set is based on StackOverflow's 2020 survey.
DATA_PATH = os.path.join(ZULIP_PATH, "tools", "setup", "lang.json")
OUT_PATH = os.path.join(ZULIP_PATH, "web", "generated", "pygments_data.json")
with open(DATA_PATH) as f:
pygments_data = json.load(f)
priorities = dict(
**pygments_data["default"], **pygments_data["custom"], **pygments_data["aliases"]
)
lexers = get_all_lexers()
langs = {
alias: {
"priority": priorities.get(alias, 0),
"pretty_name": longname,
}
for longname, aliases, filename_patterns, mimetypes in lexers
for alias in aliases
}
for name in priorities:
if langs.get(name) is None:
langs[name] = {
"priority": priorities[name],
"pretty_name": name,
}
with open(OUT_PATH, "w") as f:
json.dump({"langs": langs}, f)