diff options
Diffstat (limited to 'src/mailman_pgp/pgp/mime.py')
| -rw-r--r-- | src/mailman_pgp/pgp/mime.py | 8 |
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 |