Merge pull request #19 from jitsuin-inc/dev/eccles/cache-subclasses-for-efficiency

Dev/eccles/cache subclasses for efficiency

Author: leflambeur

archivist/archivist.py

@@ -67,10 +67,38 @@ def __init__(self, url, *, auth=None, cert=None, verify=True):
 
         self._cert = cert
 
-        self.assets = _AssetsClient(self)
-        self.events = _EventsClient(self)
-        self.locations = _LocationsClient(self)
-        self.attachments = _AttachmentsClient(self)
+        self._assets = None
+        self._events = None
+        self._locations = None
+        self._attachments = None
+
+    @property
+    def assets(self):
+        """docstring"""
+        if self._assets is None:
+            self._assets = _AssetsClient(self)
+        return self._assets
+
+    @property
+    def events(self):
+        """docstring"""
+        if self._events is None:
+            self._events = _EventsClient(self)
+        return self._events
+
+    @property
+    def locations(self):
+        """docstring"""
+        if self._locations is None:
+            self._locations = _LocationsClient(self)
+        return self._locations
+
+    @property
+    def attachments(self):
+        """docstring"""
+        if self._attachments is None:
+            self._attachments = _AttachmentsClient(self)
+        return self._attachments
 
     @property
     def headers(self):

unittests/testarchivist.py

@@ -33,6 +33,30 @@ def test_archivist(self):
         Test default archivist creation
         """
         arch = Archivist("url", auth="authauthauth")
+        self.assertIsNotNone(
+            arch.assets,
+            msg="Incorrect assets",
+        )
+        self.assertIsNotNone(
+            arch.events,
+            msg="Incorrect events",
+        )
+        self.assertIsNotNone(
+            arch.locations,
+            msg="Incorrect locations",
+        )
+        self.assertIsNotNone(
+            arch.locations,
+            msg="Incorrect locations",
+        )
+        self.assertIsNotNone(
+            arch.attachments,
+            msg="Incorrect attachments",
+        )
+        self.assertIsNotNone(
+            arch.attachments,
+            msg="Incorrect attachments",
+        )
         self.assertEqual(
             arch.url,
             "url",