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

added new terminal formatters using PYTEST_THEME for style

This commit is contained in:
pandula12 2024-05-13 15:52:42 +09:30
parent eea04c2891
commit 98121455e5
3 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
AUTHORS
View File

@ -308,6 +308,7 @@ Omar Kohl
Omer Hadari
Ondřej Súkup
Oscar Benjamin
Pandula Gajadeera
Parth Patel
Patrick Hayes
Patrick Lannigan

5
changelog/11666.improvement.rst Normal file
View File

@ -0,0 +1,5 @@
Improved the terminal writer to be able to use Terminal256Formatter and TerminalTrueColorFormatter.
These are newer versions of TerminalFormmater from the pygments library that is currently in use.
These style of these formatters are set by the users through the environmental variable PYTEST_THEME.

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

@ -218,13 +218,24 @@ class TerminalWriter:
return source
else:
try:
# Import new more rich color formatters from the pygments library
from pygments.formatters.terminal256 import TerminalTrueColorFormatter
from pygments.formatters.terminal256 import Terminal256Formatter
# Use terminal formatters depending on user environment variables
if os.environ.get('COLORTERM','') in ('truecolor', '24bit'):
# Style determined by user set environment variable, if none then use default style
terminal_formatter = TerminalTrueColorFormatter(style=os.getenv("PYTEST_THEME","default"))
elif '256' in os.environ.get('TERM', ''):
terminal_formatter = Terminal256Formatter(style=os.getenv("PYTEST_THEME","default"))
else:
terminal_formatter = TerminalFormatter(
bg=os.getenv("PYTEST_THEME_MODE", "dark"),
style=os.getenv("PYTEST_THEME"),
)
highlighted: str = highlight(
source,
Lexer(),
TerminalFormatter(
bg=os.getenv("PYTEST_THEME_MODE", "dark"),
style=os.getenv("PYTEST_THEME"),
),
)
# pygments terminal formatter may add a newline when there wasn't one.
# We don't want this, remove.