v0.50.0

TheElementalOfDestruction · TheElementalOfDestruction · commit 978c8d1753d1 · 2024-10-07T11:54:22.000-07:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,6 @@
 **v0.50.0**
-* [[TeamMsgExtractor #432](https://github.com/TeamMsgExtractor/msg-extractor/issues/432)] Adjust html header code to replace non-ascii characters with escaped versions.
+* [[TeamMsgExtractor #432](https://github.com/TeamMsgExtractor/msg-extractor/issues/432)] Adjust html header code to replace non-ascii characters with escaped versions. Also adujusted plain text to html conversion to ensure non-ascii character from the body are encoded to escpaed values to be safe.
+* Made some corrections to `NullDate`.
 
 **v0.49.0**
 * [[TeamMsgExtractor #427](https://github.com/TeamMsgExtractor/msg-extractor/issues/427)] Adjusted code for converting time stamps to create null dates for any time stamp beyond a certain point. The point was determined to be close to the existing null dates.
diff --git a/extract_msg/__init__.py b/extract_msg/__init__.py
@@ -27,7 +27,7 @@
 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 __author__ = 'Destiny Peterson & Matthew Walker'
-__date__ = '2024-10-02'
+__date__ = '2024-10-07'
 __version__ = '0.50.0'
 
 __all__ = [
diff --git a/extract_msg/msg_classes/message_base.py b/extract_msg/msg_classes/message_base.py
@@ -1163,14 +1163,14 @@ def htmlBody(self) -> Optional[bytes]:
             pass
         elif self.rtfBody:
             logger.info('HTML body was not found, attempting to generate from RTF.')
-            htmlBody = self.deencapsulateBody(self.rtfBody, DeencapType.HTML)
+            htmlBody = cast(bytes, self.deencapsulateBody(self.rtfBody, DeencapType.HTML))
         # This is it's own if statement so we can ensure it will generate
         # even if there is an rtfBody, in the event it doesn't have HTML.
         if not htmlBody and self.body:
             # Convert the plain text body to html.
             logger.info('HTML body was not found, attempting to generate from plain text body.')
             correctedBody = html.escape(self.body).replace('\r', '').replace('\n', '<br />')
-            htmlBody = f'<html><body>{correctedBody}</body></head>'.encode('utf-8')
+            htmlBody = f'<html><body>{correctedBody}</body></head>'.encode('ascii', 'xmlreplace')
 
         if not htmlBody:
             logger.info('HTML body could not be found nor generated.')
@@ -1213,7 +1213,7 @@ def htmlInjectableHeader(self) -> str:
         prefix = '<div id="injectedHeader"><div><p class="MsoNormal">'
         suffix = '<o:p></o:p></p></div></div>'
         joinStr = '<br/>'
-        formatter = (lambda name, value: f'<b>{name}:</b>&nbsp;{inputToString(htmlSanitize(value), self.stringEncoding).encode("ascii", "xmlcharrefreplace").decode()}')
+        formatter = (lambda name, value: f'<b>{name}:</b>&nbsp;{inputToString(htmlSanitize(value), self.stringEncoding).encode('ascii', 'xmlcharrefreplace').decode()}')
 
         return self.getInjectableHeader(prefix, joinStr, suffix, formatter)
 
diff --git a/extract_msg/null_date.py b/extract_msg/null_date.py
@@ -26,9 +26,9 @@ class NullDate(datetime.datetime):
     def __eq__(self, other) -> bool:
         if isinstance(other, NullDate):
             return True
-        return super.__eq__(self, other)
+        return super().__eq__(other)
 
     def __ne__(self, other) -> bool:
         if isinstance(other, NullDate):
             return False
-        return super.__eq__(self, other)
+        return super().__eq__(other)
