Cache parsing result

Author: fcurella

datauri/__init__.py

@@ -13,6 +13,8 @@
 
 from urllib.parse import quote, unquote
 
+from cached_property import cached_property
+
 from .exceptions import InvalidCharset, InvalidDataURI, InvalidMimeType
 
 T = TypeVar("T")
@@ -131,7 +133,7 @@ def is_valid(self) -> bool:
             return False
         return True
 
-    @property
+    @cached_property
     def _parse(
         self,
     ) -> Tuple[Optional[str], Optional[str], Optional[str], bool, bytes]:

setup.py

@@ -39,6 +39,7 @@ def read(fname):
     ],
     test_suite="tests",
     install_requires=[
+        "cached-property",
         "typing_extensions",
     ],
 )

tests/test_parsing.py

@@ -1,5 +1,6 @@
 import os
 from pathlib import Path
+from unittest.mock import patch
 
 import pytest
 
@@ -14,6 +15,15 @@ def test_parse():
     DataURI(t)
 
 
+@patch("datauri.unquote", return_value="The quick brown fox jumped over the lazy dog.")
+def test_parse_cached(_parse):
+    t = "data:text/plain;charset=utf-8,VGhlIHF1aWNrIGJyb3duIGZveCBqdW1wZWQgb3ZlciB0aGUgbGF6eSBkb2cu"
+    parsed = DataURI(t)
+    assert parsed.charset is not None
+    assert parsed.data is not None
+    _parse.assert_called_once()
+
+
 def test_parse_base64():
     t = "data:text/plain;charset=utf-8;base64,VGhlIHF1aWNrIGJyb3duIGZveCBqdW1wZWQgb3ZlciB0aGUgbGF6eSBkb2cu"
     DataURI(t)