bin/lts-cve-triage.py: Colorise output

git-svn-id: svn+ssh://svn.debian.org/svn/secure-testing@43989 e39458fd-73e7-0310-bf30-c45bca0a0e42

1 files changed, 15 insertions, 2 deletions

diff --git a/bin/lts-cve-triage.py b/bin/lts-cve-triage.py
index 4bcdab6c9f..075c7f907a 100755
--- a/bin/lts-cve-triage.py
+++ b/bin/lts-cve-triage.py
@@ -15,12 +15,22 @@
 # You should have received a copy of the GNU General Public License
 # along with this file.  If not, see <https://www.gnu.org/licenses/>.
 
+import sys
 import argparse
 import collections
 
 from tracker_data import TrackerData, RELEASES
 from unsupported_packages import UnsupportedPackages
 
+def colored(x, *args, **kwargs):
+    return x
+
+try:
+    if sys.stdout.isatty():
+        from termcolor import colored
+except ImportError:
+    pass
+
 LIST_NAMES = (
     ('triage_already_in_dsa_needed',
      ('Issues to triage for {lts} that are already in dsa-needed'
@@ -117,9 +127,12 @@ for key, desc in LIST_NAMES:
         continue
     if not len(lists[key]):
         continue
-    print('\n{}:'.format(desc))
+    print('\n{}:'.format(colored(desc, attrs=('bold',))))
     for pkg in sorted(lists[key].keys()):
         cve_list = ' '.join(
             [i.name for i in sorted(lists[key][pkg], key=lambda i: i.name)])
-        print('\n* {:20s}\n  {}'.format(pkg, cve_list))
+        print('\n* {:20s}\n  {}'.format(
+            colored(pkg, 'red', attrs=('bold', 'underline')),
+            cve_list,
+        ))
     print('')