feature/key-revocation

1 files changed, 3 insertions, 6 deletions

diff --git a/src/mailman_pgp/workflows/key_set.py b/src/mailman_pgp/workflows/key_set.py
index e76e299..95e6e7c 100644
--- a/src/mailman_pgp/workflows/key_set.py
+++ b/src/mailman_pgp/workflows/key_set.py
@@ -22,7 +22,6 @@ from pgpy import PGPKey
 from public import public
 
 from mailman_pgp.database import transaction
-from mailman_pgp.model.address import PGPAddress
 
 KEY_REQUEST = """\
 ----------
@@ -51,15 +50,14 @@ class SetPubkeyMixin:
             self.pubkey = None
 
     def _step_pubkey_checks(self):
-        pgp_address = PGPAddress.for_address(self.address)
-        assert pgp_address is not None
+        assert self.pgp_address is not None
 
         if self.pubkey is None:
-            if pgp_address.key is None:
+            if self.pgp_address.key is None:
                 self.push('send_key_request')
         else:
             with transaction():
-                pgp_address.key = self.pubkey
+                self.pgp_address.key = self.pubkey
 
     def _step_send_key_request(self):
         self._set_token(TokenOwner.subscriber)
@@ -75,5 +73,4 @@ class SetPubkeyMixin:
         raise StopIteration
 
     def _step_receive_key(self):
-        self._restore_subscriber()
         self._set_token(TokenOwner.no_one)