add type annotations in two places where I found this helpful to unde…

…rstand the code
FelixSchwarz · Oct 18, 2024 · c7a089d · c7a089d
1 parent da05727
commit c7a089d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/schwarz/mailqueue/mailflow_check.py b/schwarz/mailqueue/mailflow_check.py
@@ -14,7 +14,7 @@
 
 __all__ = ['build_check_message', 'send_test_message']
 
-def build_check_message(recipient, sender=None):
+def build_check_message(recipient, sender=None) -> Message:
     mail = Message()
     sender = sender or recipient
     mail['From'] = sender

diff --git a/schwarz/mailqueue/message_handler.py b/schwarz/mailqueue/message_handler.py
@@ -164,8 +164,8 @@ def retries(self, value):
 
 
 class InMemoryMsg(BaseMsg):
-    def __init__(self, from_addr, to_addrs, msg_bytes):
-        msg_fp = BytesIO(msg_as_bytes(msg_bytes))
+    def __init__(self, from_addr, to_addrs, msg_bytes: bytes):
+        msg_fp = BytesIO(msg_bytes)
         msg = MsgInfo(from_addr, to_addrs, msg_fp)
         super().__init__(msg=msg)