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
|
import sys
from pygments import styles
from pygments.token import Error
from friendly_styles import friendly_light, friendly_dark
# When friendly is imported in environments that have previously
# imported Pygments, the styles defined in friendly_styles do not
# get automatically added to the list of available styles from pygments,
# and we must "patch" the existing list.
sys.modules["pygments.styles.friendly_light"] = friendly_light
styles.STYLE_MAP["friendly_light"] = "friendly_light::FriendlyLightStyle"
friendly_light = styles.get_style_by_name("friendly_light")
sys.modules["pygments.styles.friendly_dark"] = friendly_dark
styles.STYLE_MAP["friendly_dark"] = "friendly_dark::FriendlyDarkStyle"
friendly_dark = styles.get_style_by_name("friendly_dark")
# The following global variable can be changed from other modules.
# Yes, I know, global variables are not a good idea.
CURRENT_THEME = friendly_dark
default_dark_background_colour = friendly_dark.background_color
default_light_background_colour = friendly_light.background_color
default_dark_highlight_colour = friendly_dark.styles[Error]
default_light_highlight_colour = friendly_light.styles[Error]
def get_default_background_color():
if CURRENT_THEME == friendly_dark:
return default_dark_background_colour
else:
return default_light_background_colour
def set_pygments_background_color(color):
if color is None:
CURRENT_THEME.background_color = get_default_background_color()
return get_default_background_color()
CURRENT_THEME.background_color = color
return color
def set_pygments_error_token(bg, fg):
CURRENT_THEME.styles[Error] = f"bg:{bg} {fg}"
def get_pygments_error_token():
if CURRENT_THEME == friendly_dark:
colour = default_dark_highlight_colour
else:
colour = default_light_highlight_colour
bg, fg = colour.split(" ")
bg = bg[3:]
return bg, fg
|
