Add PGPPacket._str_generic_key_packet and fingerprint calculation

[gpg-migrate.git] / gpg-migrate.py

diff --git a/gpg-migrate.py b/gpg-migrate.py
index 5b9ef10f5e2cada317096032b66201c7af4c032e..a2519f8b8355d4a3d0dd6041790867bdd0c41e08 100755 (executable)
--- a/gpg-migrate.py
+++ b/gpg-migrate.py
@@ -1,5 +1,6 @@
 #!/usr/bin/python
 
+import hashlib as _hashlib
 import re as _re
 import subprocess as _subprocess
 import struct as _struct
@@ -198,6 +199,21 @@ class PGPPacket (dict):
         details = method()
         return '{}: {}'.format(self['type'], details)
 
+    def _str_public_key_packet(self):
+        return self._str_generic_key_packet()
+
+    def _str_public_subkey_packet(self):
+        return self._str_generic_key_packet()
+
+    def _str_secret_key_packet(self):
+        return self._str_generic_key_packet()
+
+    def _str_secret_subkey_packet(self):
+        return self._str_generic_key_packet()
+
+    def _str_generic_key_packet(self):
+        return self['fingerprint'][-8:].upper()
+
     def _str_user_id_packet(self):
         return self['user']
 
@@ -337,6 +353,11 @@ class PGPPacket (dict):
             raise NotImplementedError(
                 'algorithm-specific key fields for {}'.format(
                     self['public-key-algorithm']))
+        fingerprint = _hashlib.sha1()
+        fingerprint.update(b'\x99')
+        fingerprint.update(_struct.pack('>H', len(data)))
+        fingerprint.update(data)
+        self['fingerprint'] = fingerprint.hexdigest()
         return offset
 
     def _parse_secret_key_packet(self, data):