Message: expose headers

HammadRafique29 · Mar 11, 2023 · e9e6c81 · e9e6c81
1 parent 982fec4
commit e9e6c81
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setuptools.setup(
     name="simplegmail",
-    version="4.0.0",
+    version="4.0.1",
     url="https://github.com/jeremyephron/simplegmail",
     author="Jeremy Ephron",
     author_email="jeremyephron@gmail.com",

diff --git a/simplegmail/gmail.py b/simplegmail/gmail.py
@@ -674,7 +674,7 @@ def _build_message_from_ref(
         self,
         user_id: str,
         message_ref: dict,
-        attachments: Union['ignore', 'reference', 'download'] = 'reference',
+        attachments: Union['ignore', 'reference', 'download'] = 'reference'
     ) -> Message:
         """
         Creates a Message object from a reference.
@@ -725,6 +725,7 @@ def _build_message_from_ref(
             sender = ''
             recipient = ''
             subject = ''
+            msg_hdrs = {}
             for hdr in headers:
                 if hdr['name'] == 'Date':
                     try:
@@ -737,6 +738,8 @@ def _build_message_from_ref(
                     recipient = hdr['value']
                 elif hdr['name'] == 'Subject':
                     subject = hdr['value']
+
+                msg_hdrs[hdr['name']] = hdr['value']
 
             parts = self._evaluate_message_payload(
                 payload, user_id, message_ref['id'], attachments
@@ -764,7 +767,7 @@ def _build_message_from_ref(
 
             return Message(self.service, user_id, msg_id, thread_id, recipient, 
                 sender, subject, date, snippet, plain_msg, html_msg, label_ids,
-                attms)
+                attms, msg_hdrs)
 
     def _evaluate_message_payload(
         self,

diff --git a/simplegmail/message.py b/simplegmail/message.py
@@ -16,8 +16,8 @@
 
 class Message(object):
     """
-    The Message class for emails in your Gmail mailbox. This class should not be
-    manually constructed. Contains all information about the associated 
+    The Message class for emails in your Gmail mailbox. This class should not 
+    be manually constructed. Contains all information about the associated 
     message, and can be used to modify the message's labels (e.g., marking as
     read/unread, archiving, moving to trash, starring, etc.).
 
@@ -66,7 +66,8 @@ def __init__(
         plain: Optional[str] = None,
         html: Optional[str] = None,
         label_ids: Optional[List[str]] = None,
-        attachments: Optional[List[Attachment]] = None
+        attachments: Optional[List[Attachment]] = None,
+        headers: Optional[dict] = None
     ) -> None:
         self._service = service
         self.user_id = user_id
@@ -81,6 +82,7 @@ def __init__(
         self.html = html
         self.label_ids = label_ids if label_ids is not None else []
         self.attachments = attachments if attachments is not None else []
+        self.headers = headers if headers else {}
 
     def __repr__(self) -> str:
         """Represents the object by its sender, recipient, and id."""