1 files changed, 7 insertions, 1 deletions

diff --git a/src/mailman_pgp/pgp/mime.py b/src/mailman_pgp/pgp/mime.py
index 81208bd..afb1c35 100644
--- a/src/mailman_pgp/pgp/mime.py
+++ b/src/mailman_pgp/pgp/mime.py
@@ -57,6 +57,9 @@ class MIMEWrapper:
         """
         self.msg = msg
 
+    def get_payload(self):
+        yield self.msg.as_string()
+
     def _is_mime(self):
         is_multipart = self.msg.is_multipart()
         payloads = len(self.msg.get_payload())
@@ -83,6 +86,9 @@ class MIMEWrapper:
     def has_signature(self):
         return self.is_signed()
 
+    def get_signed(self):
+        yield self.msg.get_payload(0).as_string(0)
+
     def is_encrypted(self):
         """
         Whether the whole message is MIME encrypted as per RFC3156 section 4.
@@ -275,7 +281,7 @@ class MIMEWrapper:
         payload = self.msg.as_string()
         pmsg = PGPMessage.new(payload)
         pmsg = self._encrypt(pmsg, *keys, cipher=cipher)
-        out =  self._wrap_encrypted(pmsg)
+        out = self._wrap_encrypted(pmsg)
         copy_headers(self.msg, out)
         return out