Hopiu/linkchecker
1
0
Fork 0
mirror of https://github.com/Hopiu/linkchecker.git synced 2026-05-02 11:54:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improved terminal color detection with curses.

Browse source
This commit is contained in:
Bastian Kleineidam 2012-09-17 15:24:04 +02:00
parent 6e1841cf1f
commit 1f9ee987f9
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
linkcheck/ansicolor.py
View file

@ -194,7 +194,7 @@ def get_win_color(color):
def has_colors (fp):
"""Test if given file is an ANSI color enabled tty."""
# The isatty() function ensures that we do not colorize
# The is_tty() function ensures that we do not colorize
# redirected streams, as this is almost never what we want
if not is_tty(fp):
return False
@ -203,13 +203,12 @@ def has_colors (fp):
elif has_curses:
import curses
try:
curses.setupterm()
if curses.tigetnum("colors") >= 8:
# We have at least 8 colors - allright!
return True
curses.initscr()
return curses.has_colors()
except curses.error:
# initialize error, eg. no terminfo database found
return False
finally:
curses.endwin()
return False