Skip to content

Commit 5cc8e71

Browse files
TheTechmagedbluhm
authored andcommitted
chore: move log code to default callback
Signed-off-by: Colton Wolkins (Indicio work address) <[email protected]>
1 parent 88f7e04 commit 5cc8e71

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

didcomm_messaging/quickstart.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -341,11 +341,17 @@ async def setup_relay(
341341
return new_did
342342

343343

344+
async def _message_callback(msg: Message) -> None:
345+
if msg.type == "https://didcomm.org/basicmessage/2.0/message":
346+
logmsg = msg.body["content"].replace("\n", " ").replace("\r", "")
347+
LOG.info("Got message: %s", logmsg)
348+
349+
344350
async def fetch_relayed_messages(
345351
dmp: DIDCommMessaging,
346352
my_did: DID,
347353
relay_did: DID,
348-
callback: Callable[[Message], Awaitable[None]] = None,
354+
callback: Callable[[Message], Awaitable[None]] = _message_callback,
349355
) -> List[Message]:
350356
"""Fetch stored messages from the relay."""
351357

@@ -389,10 +395,6 @@ async def fetch_relayed_messages(
389395
if callback:
390396
await callback(msg)
391397

392-
if msg.type == "https://didcomm.org/basicmessage/2.0/message":
393-
logmsg = msg.body["content"].replace("\n", " ").replace("\r", "")
394-
LOG.info(f"Got message: %s", logmsg)
395-
396398
message = Message(
397399
type="https://didcomm.org/messagepickup/3.0/messages-received",
398400
id=str(uuid.uuid4()),

0 commit comments

Comments
 (0)