"""
A color palette that adapts to the time of day.
Solarized is a sixteen color palette (eight monotones and eight accents)
designed with both precise CIELAB lightness relationships and a refined
set of hues based on fixed color wheel relationships. It is intended for
use with terminal and GUI applications.
The [Solarized][1] color palette is designed by @[Ethan Schoonover][2].
[1]: http://ethanschoonover.com/solarized
[2]: http://ethanschoonover.com
Styles based upon the original implementation `pygments-style-solarized`
by @[Shoji Kumagai](mailto:take.this.2.your.grave@gmail.com) and released
under the MIT license.
"""
# TODO import uuid
import re
import lxml.html
import pygments
import pygments.formatters
import pygments.lexers
import pygments.style
from lxml.html import builder
from pygments.token import Literal # Token
from pygments.token import (
Comment,
Error,
Generic,
Keyword,
Name,
Number,
Operator,
Other,
Punctuation,
String,
Text,
Whitespace,
)
__all__ = ["highlight", "Solar", "Lunar"]
bases = _b = {
"03": "002b36",
"0": "839496",
"02": "073642",
"1": "93a1a1",
"01": "586e75",
"2": "eee8d5",
"00": "657b83",
"3": "fdf6e3",
}
dark_bases = (_b["03"], _b["02"], _b["01"], _b["0"], _b["1"])
light_bases = (_b["3"], _b["2"], _b["0"], _b["00"], _b["01"])
colors = _c = {
"yellow": "b58900",
"orange": "cb4b16",
"red": "dc322f",
"magenta": "d33682",
"violet": "6c71c4",
"blue": "268bd2",
"cyan": "2aa198",
"green": "859900",
}
palette = tuple(list(bases.values()) + list(colors.values()))
accented = [
(_b["02"], _b["01"], _b["0"], _b["1"], _c["red"]),
(_b["03"], _b["02"], _b["01"], _b["00"], _c["yellow"]),
(_b["00"], _b["1"], _b["2"], _b["3"], _c["blue"]),
(_b["00"], _b["0"], _b["1"], _b["2"], _c["orange"]),
(_b["02"], _b["1"], _b["2"], _c["magenta"], _c["violet"]),
]
def highlight(
code,
filename,
lines=True,
starting_line=1,
focus=None,
coverage=None,
failures=None,
search=None,
diff=None,
filename_anchors=False,
):
"""
Highlight given code.
Focus any lines containing `search`.
"""
if filename.endswith(".py"):
lexer = pygments.lexers.Python3Lexer()
else:
try:
lexer = pygments.lexers.get_lexer_for_filename(filename)
except pygments.util.ClassNotFound:
lexer = pygments.lexers.get_lexer_for_filename(".txt")
# XXX return code
lexer.add_filter("codetagify")
# TODO random_id = uuid.uuid4()
# XXX is_file = "." in filename # an actual filename, eg "date_util.py"
# XXX if not is_file:
# XXX filename = f"XaXaX{filename}"
# if diff:
# starting_line = diff[0]
starting_line = int(starting_line)
if diff:
formatter_args = {}
else:
formatter_args = {"linenos": "table", "linenostart": starting_line}
formatter = pygments.formatters.HtmlFormatter(linespans=filename, **formatter_args)
formatter = pygments.formatters.HtmlFormatter(linespans=filename, **formatter_args)
html = pygments.highlight(code, lexer, formatter)
doc = lxml.html.fromstring(html)
if search:
focus = []
for line in doc.cssselect("td.code > div > pre > span"):
if re.search(search, line.text_content()):
focus.append(line.attrib["id"].rpartition("-")[2])
if focus:
if isinstance(focus, int):
focus = [focus]
for focus_line in focus:
# XXX given_id = f"{random_id}-{focus_line}".format(random_id, focus_line)
given_id = str(focus_line)
html = html.replace(given_id, f'{given_id}" class="focus')
if coverage:
for line, covered in coverage:
if int(covered):
class_name = "covered"
else:
class_name = "uncovered"
current_id = f'{filename}-{line}"'
html = html.replace(current_id, f'{current_id} class="{class_name}"')
if failures:
for line in failures:
current_id = f"{filename}-{line}"
html = html.replace(current_id, f'{current_id}" class="failed')
if diff:
...
else:
doc = lxml.html.fromstring(html)
linenos = doc.cssselect(".linenodiv pre")[0]
ending_line = linenos.text_content().rpartition("\n")[2].partition('"')[0]
for lineno in linenos.iterchildren():
lineno.drop_tree()
linenos.text = ""
for line in range(starting_line, int(ending_line) + 1):
line_link = f"#L{line}"
if filename_anchors:
line_link = f"#{filename}-{line}"
linenos.append(builder.A(str(line), "\n", href=line_link))
for span in doc.cssselect(".highlight pre > span"):
if span.text:
span.text = span.text.replace(" ", "\u00a0")
html = lxml.html.tostring(doc).decode("utf-8")
if not filename_anchors:
html = html.replace(f"{filename}-", "")
return html
class Solar(pygments.style.Style):
"""A Pygments style based upon Solarized's light palette."""
background_color = "#fdf6e3"
default_style = ""
styles = {
Text: "#657b83", # base00
Whitespace: "#fdf6e3", # base3 w
Error: "#dc322f", # red err
Other: "#657b83", # base00 x
Comment: "italic #93a1a1", # base1 c
Comment.Multiline: "italic #93a1a1", # base1 cm
Comment.Preproc: "italic #93a1a1", # base1 cp
Comment.Single: "italic #93a1a1", # base1 c1
Comment.Special: "italic #93a1a1", # base1 cs
Keyword: "#859900", # green k
Keyword.Constant: "#859900", # green kc
Keyword.Declaration: "#859900", # green kd
Keyword.Namespace: "#cb4b16", # orange kn
Keyword.Pseudo: "#cb4b16", # orange kp
Keyword.Reserved: "#859900", # green kr
Keyword.Type: "#859900", # green kt
Operator: "#657b83", # base00 o
Operator.Word: "#859900", # green ow
Name: "#586e75", # base01 n
Name.Attribute: "#657b83", # base00 na
Name.Builtin: "#268bd2", # blue nb
Name.Builtin.Pseudo: "bold #268bd2", # blue bp
Name.Class: "#268bd2", # blue nc
Name.Constant: "#b58900", # yellow no
Name.Decorator: "#cb4b16", # orange nd
Name.Entity: "#cb4b16", # orange ni
Name.Exception: "#cb4b16", # orange ne
Name.Function: "#268bd2", # blue nf
Name.Property: "#268bd2", # blue py
Name.Label: "#657b83", # base00 nc
Name.Namespace: "#b58900", # yellow nn
Name.Other: "#657b83", # base00 nx
Name.Tag: "#859900", # green nt
Name.Variable: "#cd4b16", # orange nv
Name.Variable.Class: "#268bd2", # blue vc
Name.Variable.Global: "#268bd2", # blue vg
Name.Variable.Instance: "#268bd2", # blue vi
Number: "#2aa198", # cyan m
Number.Float: "#2aa198", # cyan mf
Number.Hex: "#2aa198", # cyan mh
Number.Integer: "#2aa198", # cyan mi
Number.Integer.Long: "#2aa198", # cyan il
Number.Oct: "#2aa198", # cyan mo
Literal: "#657b83", # base00 l
Literal.Date: "#657b83", # base00 ld
Punctuation: "#657b83", # base00 p
String: "#2aa198", # cyan s
String.Backtick: "#2aa198", # cyan sb
String.Char: "#2aa198", # cyan sc
String.Doc: "#2aa198", # cyan sd
String.Double: "#2aa198", # cyan s2
String.Escape: "#cb4b16", # orange se
String.Heredoc: "#2aa198", # cyan sh
String.Interpol: "#cb4b16", # orange si
String.Other: "#2aa198", # cyan sx
String.Regex: "#2aa198", # cyan sr
String.Single: "#2aa198", # cyan s1
String.Symbol: "#2aa198", # cyan ss
Generic: "#657b83", # base00 g
Generic.Deleted: "#657b83", # base00 gd
Generic.Emph: "#657b83", # base00 ge
Generic.Error: "#657b83", # base00 gr
Generic.Heading: "#657b83", # base00 gh
Generic.Inserted: "#657b83", # base00 gi
Generic.Output: "#657b83", # base00 go
Generic.Prompt: "#657b83", # base00 gp
Generic.Strong: "#657b83", # base00 gs
Generic.Subheading: "#657b83", # base00 gu
Generic.Traceback: "#657b83", # base00 gt
}
class Lunar(pygments.style.Style):
"""A Pygments style based upon Solarized's dark palette."""
background_color = "#002b36"
default_style = ""
styles = {
Text: "#839496", # base0
Whitespace: "#002b36", # base03 w
Error: "#dc322f", # red err
Other: "#839496", # base0 x
Comment: "italic #586e75", # base01 c
Comment.Multiline: "italic #586e75", # base01 cm
Comment.Preproc: "italic #586e75", # base01 cp
Comment.Single: "italic #586e75", # base01 c1
Comment.Special: "italic #586e75", # base01 cs
Keyword: "#859900", # green k
Keyword.Constant: "#859900", # green kc
Keyword.Declaration: "#859900", # green kd
Keyword.Namespace: "#cb4b16", # orange kn
Keyword.Pseudo: "#cb4b16", # orange kp
Keyword.Reserved: "#859900", # green kr
Keyword.Type: "#859900", # green kt
Operator: "#839496", # base0 o
Operator.Word: "#859900", # green ow
Name: "#93a1a1", # base1 n
Name.Attribute: "#839496", # base0 na
Name.Builtin: "#268bd2", # blue nb
Name.Builtin.Pseudo: "bold #268bd2", # blue bp
Name.Class: "#268bd2", # blue nc
Name.Constant: "#b58900", # yellow no
Name.Decorator: "#cb4b16", # orange nd
Name.Entity: "#cb4b16", # orange ni
Name.Exception: "#cb4b16", # orange ne
Name.Function: "#268bd2", # blue nf
Name.Property: "#268bd2", # blue py
Name.Label: "#839496", # base0 nc
Name.Namespace: "#b58900", # yellow nn
Name.Other: "#839496", # base0 nx
Name.Tag: "#859900", # green nt
Name.Variable: "#cd4b16", # orange nv
Name.Variable.Class: "#268bd2", # blue vc
Name.Variable.Global: "#268bd2", # blue vg
Name.Variable.Instance: "#268bd2", # blue vi
Number: "#2aa198", # cyan m
Number.Float: "#2aa198", # cyan mf
Number.Hex: "#2aa198", # cyan mh
Number.Integer: "#2aa198", # cyan mi
Number.Integer.Long: "#2aa198", # cyan il
Number.Oct: "#2aa198", # cyan mo
Literal: "#839496", # base0 l
Literal.Date: "#839496", # base0 ld
Punctuation: "#839496", # base0 p
String: "#2aa198", # cyan s
String.Backtick: "#2aa198", # cyan sb
String.Char: "#2aa198", # cyan sc
String.Doc: "#2aa198", # cyan sd
String.Double: "#2aa198", # cyan s2
String.Escape: "#cb4b16", # orange se
String.Heredoc: "#2aa198", # cyan sh
String.Interpol: "#cb4b16", # orange si
String.Other: "#2aa198", # cyan sx
String.Regex: "#2aa198", # cyan sr
String.Single: "#2aa198", # cyan s1
String.Symbol: "#2aa198", # cyan ss
Generic: "#839496", # base0 g
Generic.Deleted: "#839496", # base0 gd
Generic.Emph: "#839496", # base0 ge
Generic.Error: "#839496", # base0 gr
Generic.Heading: "#839496", # base0 gh
Generic.Inserted: "#839496", # base0 gi
Generic.Output: "#839496", # base0 go
Generic.Prompt: "#839496", # base0 gp
Generic.Strong: "#839496", # base0 gs
Generic.Subheading: "#839496", # base0 gu
Generic.Traceback: "#839496", # base0 gt
}
