floraachy
/
pytest2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

terminalwriter: Create styling class for 256 Formatter.

This commit is contained in:
TyYan03 2023-12-08 17:53:33 -05:00
parent c7b5175d7f
commit 764fb1af8e
1 changed files with 46 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
src/_pytest/_io/terminalwriter.py
View File

@ -215,13 +215,56 @@ class TerminalWriter:
return source return source
else: else:
try: try:
from pygments.formatters.terminal256 import TerminalTrueColorFormatter from pygments.style import Style
from pygments.formatters.terminal256 import Terminal256Formatter from pygments.token import (
Token,
Comment,
Keyword,
Name,
String,
Error,
Generic,
Number,
Operator,
)
from pygments.formatters.terminal256 import TerminalTrueColorFormatter
# Set the terminal formatter to better performing formatters is user environement allows for it
if os.environ.get('COLORTERM','') in ('truecolor', '24bit'): if os.environ.get('COLORTERM','') in ('truecolor', '24bit'):
terminal_formatter = TerminalTrueColorFormatter() terminal_formatter = TerminalTrueColorFormatter()
elif '256' in os.environ.get('TERM', ''): elif '256' in os.environ.get('TERM', ''):
terminal_formatter = Terminal256Formatter() # Create new styling for 256 Formatter
class Updated256Style(Style):
# Set color values for the 256 Formatter
styles = {
Token: (""),
Comment: ("ansigray"),
Comment.Preproc: ("ansicyan"),
Keyword: ("ansiblue"),
Keyword.Type: ("ansicyan"),
Operator.Word: ("ansimagenta"),
Name.Builtin: ("ansicyan"),
Name.Function: ("ansigreen"),
Name.Namespace: ("ansicyan"),
Name.Class: ("ansigreen"),
Name.Exception: ("ansicyan"),
Name.Decorator: ("ansibrightblack"),
Name.Variable: ("ansired"),
Name.Constant: ("ansired"),
Name.Attribute: ("ansicyan"),
Name.Tag: ("ansibrightblue"),
String: ("ansiyellow"),
Number: ("ansiblue"),
Generic.Deleted: ("ansibrightred"),
Generic.Inserted: ("ansigreen"),
Generic.Subheading: ("ansimagenta"),
Generic.Error: ("ansibrightred"),
Error: ("ansibrightred"),
}
from pygments.formatters.terminal256 import Terminal256Formatter
terminal_formatter = Terminal256Formatter(style=Updated256Style)
else: else:
terminal_formatter = TerminalFormatter( terminal_formatter = TerminalFormatter(
bg=os.getenv("PYTEST_THEME_MODE", "dark"), bg=os.getenv("PYTEST_THEME_MODE", "dark"),