feature/key-revocation

1 files changed, 23 insertions, 18 deletions

diff --git a/src/mailman_pgp/workflows/subscription.py b/src/mailman_pgp/workflows/subscription.py
index 0ef1358..2546f1e 100644
--- a/src/mailman_pgp/workflows/subscription.py
+++ b/src/mailman_pgp/workflows/subscription.py
@@ -19,12 +19,11 @@
 
 from mailman.core.i18n import _
 from mailman.interfaces.workflows import ISubscriptionWorkflow
-from mailman.workflows.common import (ConfirmationMixin, SubscriptionBase,
-                                      VerificationMixin)
+from mailman.workflows.common import (ConfirmationMixin, VerificationMixin)
 from public import public
 from zope.interface import implementer
 
-from mailman_pgp.workflows.base import PGPMixin
+from mailman_pgp.workflows.base import PGPMixin, PGPSubscriptionBase
 from mailman_pgp.workflows.key_confirm import ConfirmPubkeyMixin
 from mailman_pgp.workflows.key_set import SetPubkeyMixin
 from mailman_pgp.workflows.mod_approval import ModeratorSubApprovalMixin
@@ -32,7 +31,7 @@ from mailman_pgp.workflows.mod_approval import ModeratorSubApprovalMixin
 
 @public
 @implementer(ISubscriptionWorkflow)
-class OpenSubscriptionPolicy(SubscriptionBase, VerificationMixin,
+class OpenSubscriptionPolicy(PGPSubscriptionBase, VerificationMixin,
                              SetPubkeyMixin, ConfirmPubkeyMixin,
                              PGPMixin):
     """"""
@@ -54,7 +53,7 @@ class OpenSubscriptionPolicy(SubscriptionBase, VerificationMixin,
     def __init__(self, mlist, subscriber=None, *,
                  pre_verified=False, pubkey=None,
                  pubkey_pre_confirmed=False):
-        SubscriptionBase.__init__(self, mlist, subscriber)
+        PGPSubscriptionBase.__init__(self, mlist, subscriber)
         VerificationMixin.__init__(self, pre_verified=pre_verified)
         SetPubkeyMixin.__init__(self, pubkey=pubkey)
         ConfirmPubkeyMixin.__init__(self, pre_confirmed=pubkey_pre_confirmed)
@@ -62,7 +61,9 @@ class OpenSubscriptionPolicy(SubscriptionBase, VerificationMixin,
 
     def _step_prepare(self):
         self.push('do_subscription')
+        self.push('restore_subscriber')
         self.push('pubkey_confirmation')
+        self.push('restore_address')
         self.push('pubkey_checks')
         self.push('create_address')
         self.push('verification_checks')
@@ -71,9 +72,9 @@ class OpenSubscriptionPolicy(SubscriptionBase, VerificationMixin,
 
 @public
 @implementer(ISubscriptionWorkflow)
-class ConfirmSubscriptionPolicy(SubscriptionBase, VerificationMixin,
+class ConfirmSubscriptionPolicy(PGPSubscriptionBase, VerificationMixin,
                                 ConfirmationMixin, SetPubkeyMixin,
-                                ConfirmPubkeyMixin, PGPMixin):
+                                ConfirmPubkeyMixin):
     """"""
 
     name = 'pgp-policy-confirm'
@@ -94,16 +95,17 @@ class ConfirmSubscriptionPolicy(SubscriptionBase, VerificationMixin,
     def __init__(self, mlist, subscriber=None, *,
                  pre_verified=False, pre_confirmed=False, pubkey=None,
                  pubkey_pre_confirmed=False):
-        SubscriptionBase.__init__(self, mlist, subscriber)
+        PGPSubscriptionBase.__init__(self, mlist, subscriber)
         VerificationMixin.__init__(self, pre_verified=pre_verified)
         ConfirmationMixin.__init__(self, pre_confirmed=pre_confirmed)
         SetPubkeyMixin.__init__(self, pubkey=pubkey)
         ConfirmPubkeyMixin.__init__(self, pre_confirmed=pubkey_pre_confirmed)
-        PGPMixin.__init__(self, mlist)
 
     def _step_prepare(self):
         self.push('do_subscription')
+        self.push('restore_subscriber')
         self.push('pubkey_confirmation')
+        self.push('restore_address')
         self.push('pubkey_checks')
         self.push('create_address')
         self.push('confirmation_checks')
@@ -113,9 +115,9 @@ class ConfirmSubscriptionPolicy(SubscriptionBase, VerificationMixin,
 
 @public
 @implementer(ISubscriptionWorkflow)
-class ModerationSubscriptionPolicy(SubscriptionBase, VerificationMixin,
+class ModerationSubscriptionPolicy(PGPSubscriptionBase, VerificationMixin,
                                    ModeratorSubApprovalMixin, SetPubkeyMixin,
-                                   ConfirmPubkeyMixin, PGPMixin):
+                                   ConfirmPubkeyMixin):
     """"""
 
     name = 'pgp-policy-moderate'
@@ -136,17 +138,18 @@ class ModerationSubscriptionPolicy(SubscriptionBase, VerificationMixin,
     def __init__(self, mlist, subscriber=None, *,
                  pre_verified=False, pre_approved=False, pubkey=None,
                  pubkey_pre_confirmed=False):
-        SubscriptionBase.__init__(self, mlist, subscriber)
+        PGPSubscriptionBase.__init__(self, mlist, subscriber)
         VerificationMixin.__init__(self, pre_verified=pre_verified)
         ModeratorSubApprovalMixin.__init__(self, pre_approved=pre_approved)
         SetPubkeyMixin.__init__(self, pubkey=pubkey)
         ConfirmPubkeyMixin.__init__(self, pre_confirmed=pubkey_pre_confirmed)
-        PGPMixin.__init__(self, mlist)
 
     def _step_prepare(self):
         self.push('do_subscription')
+        self.push('restore_subscriber')
         self.push('mod_approval')
         self.push('pubkey_confirmation')
+        self.push('restore_address')
         self.push('pubkey_checks')
         self.push('create_address')
         self.push('verification_checks')
@@ -155,11 +158,11 @@ class ModerationSubscriptionPolicy(SubscriptionBase, VerificationMixin,
 
 @public
 @implementer(ISubscriptionWorkflow)
-class ConfirmModerationSubscriptionPolicy(SubscriptionBase, VerificationMixin,
+class ConfirmModerationSubscriptionPolicy(PGPSubscriptionBase,
+                                          VerificationMixin,
                                           ConfirmationMixin,
                                           ModeratorSubApprovalMixin,
-                                          SetPubkeyMixin, ConfirmPubkeyMixin,
-                                          PGPMixin):
+                                          SetPubkeyMixin, ConfirmPubkeyMixin):
     """"""
 
     name = 'pgp-policy-confirm-moderate'
@@ -181,18 +184,20 @@ class ConfirmModerationSubscriptionPolicy(SubscriptionBase, VerificationMixin,
     def __init__(self, mlist, subscriber=None, *,
                  pre_verified=False, pre_confirmed=False, pre_approved=False,
                  pubkey=None, pubkey_pre_confirmed=False):
-        SubscriptionBase.__init__(self, mlist, subscriber)
+        PGPSubscriptionBase.__init__(self, mlist, subscriber)
         VerificationMixin.__init__(self, pre_verified=pre_verified)
         ConfirmationMixin.__init__(self, pre_confirmed=pre_confirmed)
         ModeratorSubApprovalMixin.__init__(self, pre_approved=pre_approved)
         SetPubkeyMixin.__init__(self, pubkey=pubkey)
         ConfirmPubkeyMixin.__init__(self, pre_confirmed=pubkey_pre_confirmed)
-        PGPMixin.__init__(self, mlist)
 
     def _step_prepare(self):
         self.push('do_subscription')
+        self.push('restore_subscriber')
         self.push('mod_approval')
+        self.push('restore_address')
         self.push('pubkey_confirmation')
+        self.push('restore_address')
         self.push('pubkey_checks')
         self.push('create_address')
         self.push('confirmation_checks')