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

terminalwriter: ran linting on file.

This commit is contained in:
TyYan03 2023-12-08 18:14:21 -05:00
parent cb499d7483
commit b6d830a748
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -229,7 +229,7 @@ class TerminalWriter:
)
# 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"):
# Create new styling for True Color Formatter
class UpdatedTrueStyle(Style):
# Set color values for the True Color Formatter
@ -259,9 +259,14 @@ class TerminalWriter:
Error: ("ansibrightblue"),
}
from pygments.formatters.terminal256 import TerminalTrueColorFormatter
terminal_formatter = TerminalTrueColorFormatter(style=UpdatedTrueStyle)
elif '256' in os.environ.get('TERM', ''):
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):
# Set color values for the 256 Formatter
@ -292,12 +297,13 @@ class TerminalWriter:
}
from pygments.formatters.terminal256 import Terminal256Formatter
terminal_formatter = Terminal256Formatter(style=Updated256Style)
else:
terminal_formatter = TerminalFormatter(
bg=os.getenv("PYTEST_THEME_MODE", "dark"),
style=os.getenv("PYTEST_THEME"),
)
bg=os.getenv("PYTEST_THEME_MODE", "dark"),
style=os.getenv("PYTEST_THEME"),
)
highlighted: str = highlight(
source,
Lexer(),