fs: compute size lazily

scmrepo/fs.py

@@ -220,10 +220,12 @@ def _open(
     def info(self, path: str, **kwargs: Any) -> Dict[str, Any]:
         key = self._get_key(path)
         try:
-            return {
-                **self.trie.info(key),
-                "name": path,
-            }
+            # NOTE: to avoid wasting time computing object size, trie.info
+            # will return a LazyDict instance, that will compute compute size
+            # only when it is accessed.
+            ret = self.trie.info(key)
+            ret["name"] = path
+            return ret
         except KeyError:
             raise FileNotFoundError(
                 errno.ENOENT, os.strerror(errno.ENOENT), path

scmrepo/git/objects.py

@@ -1,7 +1,7 @@
 import stat
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
-from typing import Iterable, List, Optional
+from typing import Iterable, List, Optional, cast
 
 from pygtrie import Trie
 
@@ -136,13 +136,23 @@ def walk(self, top: tuple, topdown: Optional[bool] = True):
             yield top, dirs, nondirs
 
     def info(self, key: tuple) -> dict:
+        from scmrepo.utils import LazyDict
+
         obj = self.trie[key]
-        return {
-            "size": obj.size,
-            "type": "directory" if stat.S_ISDIR(obj.mode) else "file",
-            "sha": obj.sha,
-            "mode": obj.mode,
-        }
+
+        def size():
+            return obj.size
+
+        ret = LazyDict(
+            {
+                "size": size,
+                "type": "directory" if stat.S_ISDIR(obj.mode) else "file",
+                "sha": obj.sha,
+                "mode": obj.mode,
+            }
+        )
+
+        return cast(dict, ret)
 
 
 @dataclass

scmrepo/utils.py

@@ -1,4 +1,29 @@
 import os
+from collections.abc import MutableMapping
+
+
+class LazyDict(MutableMapping):
+    def __init__(self, values):
+        self._values = values
+
+    def __getitem__(self, item):
+        value = self._values[item]
+        if callable(value):
+            value = value()
+            self._values[item] = value
+        return value
+
+    def __setitem__(self, key, value):
+        self._values[key] = value
+
+    def __delitem__(self, key):
+        del self._values[key]
+
+    def __iter__(self):
+        return iter(self._values)
+
+    def __len__(self):
+        return len(self._values)
 
 
 def relpath(path, start=os.curdir):