1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
import logging
import os
from pathlib import Path
from typing import Union
from pygments.formatters import HtmlFormatter
from pygments.styles import get_style_by_name
from pygments.token import Text
from setuptools import find_packages
def find_all_themes_packages() -> list[str]:
"""Finds the currently supported themes in the a11y_pygments package.
Returns:
themes: list of themes under the a11y_pygments package
"""
exclude = {"test", "scripts", "a11y_pygments", "a11y_pygments.utils"}
packages = set(find_packages())
themes = list(packages - exclude)
# drop the a11y_pygments part of the pkg name
themes = [x.split(".")[1] for x in themes]
return themes
def get_themes_names() -> list[str]:
"""Get themes names from the a11y_pygments package.
Returns:
themes: list of themes names
"""
themes = find_all_themes_packages()
themes = [x.replace("_", "-") for x in themes]
logging.info(f"Found pygment themes: {themes}")
return themes
def generate_css(themes: list[str], save_dir: Union[str, Path]):
"""Generate css for the available themes.
Args:
themes (list): list of themes names
"""
assert save_dir, "Must provide directory"
for theme in themes:
style = get_style_by_name(theme)
formatter = HtmlFormatter(style=style, full=True, hl_lines=[2, 3, 4])
css = formatter.get_style_defs()
color = style.style_for_token(Text)["color"]
css += (
f"\n.highlight {{ background: {style.background_color}; color: #{color}; }}"
)
package = theme.replace("-", "_")
if not Path(save_dir).joinpath("css").exists():
os.mkdir(Path(save_dir).joinpath("css"))
if "docs" in save_dir:
out = Path(save_dir).joinpath("css", f"{package}-style.css")
else:
out = Path(save_dir) / package / "style.css"
logging.info(f"Saving css to {out}")
with open(out, "w") as f:
f.write(css)
|
