From 226b82536ad6a4eda7d7afdacc7e838e79298d5a Mon Sep 17 00:00:00 2001
From: iceman1001 <iceman@iuse.se>
Date: Mon, 22 Feb 2021 21:25:25 +0100
Subject: [PATCH] color and text, which introduced a new dependency..

---
 tools/recover_pk.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/tools/recover_pk.py b/tools/recover_pk.py
index 0310b4750..c46180d41 100755
--- a/tools/recover_pk.py
+++ b/tools/recover_pk.py
@@ -5,6 +5,9 @@
 import binascii
 import sys
 import hashlib
+from colors import *
+
+# pip3 install ansicolors
 
 debug = False
 
@@ -460,7 +463,6 @@ def recover_multiple(uids, sigs, curvename, alghash=None):
             recovered &= recovered_tmp
     return recovered
 
-
 def selftests():
     tests = [
         {'name': "Mifare Ultralight EV1",
@@ -556,14 +558,17 @@ if __name__ == "__main__":
         print("Example: \n%s 04ee45daa34084 ebb6102bff74b087d18a57a54bc375159a04ea9bc61080b7f4a85afe1587d73b" % sys.argv[0])
         exit(1)
     uids, sigs = sys.argv[1:][::2], sys.argv[1:][1::2]
+    once = True
     curvenames = guess_curvename(sigs[0])
     for c in curvenames:
-        print("\nAssuming curve=%s" % c)
-        print("========================")
         for h in [None, "md5", "sha1", "sha256", "sha512"]:
-            print("Assuming hash=%s" % h)
             recovered = recover_multiple(uids, sigs, c, alghash=h)
             if recovered:
+                if once:
+                    print(color('curve=%s', fg='yellow') % c)
+                    once = False
+                print(color('hash=%s', fg='yellow') % h)
                 print("Possible uncompressed Pk(s):")
                 for pk in list(recovered):
                     print(binascii.hexlify(pk).decode('utf8'))
+        once = True