From c401f2c6cc5f0aa4c6de245dcf252d3bbcc920fd Mon Sep 17 00:00:00 2001
From: J08nY
Date: Thu, 27 Jul 2017 17:30:51 +0200
Subject: Add get_signature and get_encrypted to wrappers.

---
 src/mailman_pgp/pgp/mime.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'src/mailman_pgp/pgp/mime.py')

diff --git a/src/mailman_pgp/pgp/mime.py b/src/mailman_pgp/pgp/mime.py
index 9978b54..c702eb8 100644
--- a/src/mailman_pgp/pgp/mime.py
+++ b/src/mailman_pgp/pgp/mime.py
@@ -88,8 +88,19 @@ class MIMEWrapper:
         return self.is_signed()
 
     def get_signed(self):
+        """
+
+        :return:
+        """
         yield self.msg.get_payload(0).as_string()
 
+    def get_signature(self):
+        """
+
+        :return:
+        """
+        yield self.msg.get_payload(1).get_payload()
+
     def is_encrypted(self):
         """
         Whether the whole message is MIME encrypted as per RFC3156 section 4.
@@ -115,6 +126,13 @@ class MIMEWrapper:
     def has_encryption(self):
         return self.is_encrypted()
 
+    def get_encrypted(self):
+        """
+
+        :return:
+        """
+        yield self.msg.get_payload(1).get_payload()
+
     def is_keys(self):
         """
         Whether the message has only keys as per RFC3156 section 7.
-- 
cgit v1.2.3-70-g09d2