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

terminalwriter: Adding styling class for True Color Formatter.

This commit is contained in:
TyYan03 2023-12-08 18:12:36 -05:00
parent 764fb1af8e
commit cb499d7483
1 changed files with 31 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -228,11 +228,39 @@ class TerminalWriter:
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'):
terminal_formatter = TerminalTrueColorFormatter()
# Create new styling for True Color Formatter
class UpdatedTrueStyle(Style):
# Set color values for the True Color 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: ("ansibrightblue"),
}
from pygments.formatters.terminal256 import TerminalTrueColorFormatter
terminal_formatter = TerminalTrueColorFormatter(style=UpdatedTrueStyle)
elif '256' in os.environ.get('TERM', ''):
# Create new styling for 256 Formatter
class Updated256Style(Style):