refactored light weight webhook object for v1 & v2 cred exchange

Signed-off-by: Victor Lee <victorlee0505@gmail.com>

Author: victorlee0505

aries_cloudagent/protocols/issue_credential/v1_0/messages/credential_exchange_webhook.py

@@ -0,0 +1,45 @@
+"""v1.0 credential exchange light weight webhook"""
+
+class LightWeightV10CredentialExchangeWebhook:
+    """Class representing a state only credential exchange webhook."""
+
+    __acceptable_keys_list = [
+        "connection_id",
+        "credential_exchange_id",
+        "cred_ex_id",
+        "cred_def_id",
+        "role",
+        "initiator",
+        "revoc_reg_id",
+        "revocation_id",
+        "auto_offer",
+        "auto_issue",
+        "auto_remove",
+        "error_msg",
+        "thread_id",
+        "parent_thread_id",
+        "state",
+        "credential_definition_id",
+        "schema_id",
+        "credential_id",
+        "trace",
+        "public_did",
+        "cred_id_stored",
+        "conn_id",
+    ]
+
+    def __init__(
+        self,
+        **kwargs,
+    ):
+        """
+        Initialize webhook object from V10CredentialExchange
+        from a list of accepted attributes.
+        """
+        [
+            self.__setattr__(key, kwargs.get(key))
+            for key in self.__acceptable_keys_list
+            if kwargs.get(key) is not None
+        ]
+        if kwargs.get("_id") is not None:
+            self.credential_exchange_id = kwargs.get("_id")

aries_cloudagent/protocols/issue_credential/v1_0/models/credential_exchange.py

@@ -12,12 +12,14 @@
 from .....indy.models.cred_precis import IndyCredInfo, IndyCredInfoSchema
 from .....indy.models.cred_request import IndyCredRequest, IndyCredRequestSchema
 from .....messaging.models.base_record import BaseExchangeRecord, BaseExchangeSchema
-from .....messaging.models.light_webhook import LightWeightWebhook
 from .....messaging.valid import INDY_CRED_DEF_ID, INDY_SCHEMA_ID, UUIDFour
 from .....storage.base import StorageError
 
 from ..messages.credential_proposal import CredentialProposal, CredentialProposalSchema
 from ..messages.credential_offer import CredentialOffer, CredentialOfferSchema
+from ..messages.credential_exchange_webhook import (
+    LightWeightV10CredentialExchangeWebhook,
+)
 
 from . import UNENCRYPTED_TAGS
 
@@ -244,7 +246,7 @@ async def emit_event(self, session: ProfileSession, payload: Any = None):
             payload = self.serialize()
 
         if session.profile.settings.get("transport.light_weight_webhook"):
-            payload = LightWeightWebhook(1, **self.__dict__)
+            payload = LightWeightV10CredentialExchangeWebhook(**self.__dict__)
             payload = payload.__dict__
 
         await session.profile.notify(topic, payload)

…dagent/messaging/models/light_webhook.py …/v2_0/messages/cred_ex_record_webhook.pyaries_cloudagent/messaging/models/light_webhook.py renamed to aries_cloudagent/protocols/issue_credential/v2_0/messages/cred_ex_record_webhook.py aries_cloudagent/messaging/models/light_webhook.py renamed to aries_cloudagent/protocols/issue_credential/v2_0/messages/cred_ex_record_webhook.py

@@ -1,4 +1,8 @@
-class LightWeightWebhook:
+"""v2.0 credential exchange light weight webhook"""
+
+class LightWeightV20CredExRecordWebhook:
+    """Class representing a state only credential exchange webhook."""
+
     __acceptable_keys_list = [
         "connection_id",
         "credential_exchange_id",
@@ -26,15 +30,16 @@ class LightWeightWebhook:
 
     def __init__(
         self,
-        version,  # 2 = V20CredExRecord ; 1 = V10CredentialExchange
         **kwargs,
     ):
+        """
+        Initialize webhook object from V20CredExRecord
+        from a list of accepted attributes.
+        """
         [
             self.__setattr__(key, kwargs.get(key))
             for key in self.__acceptable_keys_list
             if kwargs.get(key) is not None
         ]
-        if version == 2:
+        if kwargs.get("_id") is not None:
             self.cred_ex_id = kwargs.get("_id")
-        else:
-            self.credential_exchange_id = kwargs.get("_id")

aries_cloudagent/protocols/issue_credential/v2_0/models/cred_ex_record.py

@@ -8,7 +8,6 @@
 
 from .....core.profile import ProfileSession
 from .....messaging.models.base_record import BaseExchangeRecord, BaseExchangeSchema
-from .....messaging.models.light_webhook import LightWeightWebhook
 from .....messaging.valid import UUIDFour
 from .....storage.base import StorageError
 
@@ -18,6 +17,7 @@
 from ..messages.cred_offer import V20CredOffer, V20CredOfferSchema
 from ..messages.cred_request import V20CredRequest, V20CredRequestSchema
 from ..messages.inner.cred_preview import V20CredPreviewSchema
+from ..messages.cred_ex_record_webhook import LightWeightV20CredExRecordWebhook
 
 from . import UNENCRYPTED_TAGS
 
@@ -204,7 +204,7 @@ async def emit_event(self, session: ProfileSession, payload: Any = None):
             payload = self.serialize()
 
         if session.profile.settings.get("transport.light_weight_webhook"):
-            payload = LightWeightWebhook(2, **self.__dict__)
+            payload = LightWeightV20CredExRecordWebhook(**self.__dict__)
             payload = payload.__dict__
 
         await session.profile.notify(topic, payload)