Add .layouts() to Document entity

textractor/entities/document.py

@@ -21,6 +21,7 @@
 from textractor.entities.table import Table
 from textractor.entities.query import Query
 from textractor.entities.signature import Signature
+from textractor.entities.layout import Layout
 from textractor.exceptions import InputError
 from textractor.entities.key_value import KeyValue
 from textractor.entities.bbox import SpatialObject
@@ -192,6 +193,16 @@ def signatures(self) -> EntityList[Signature]:
         """
         return EntityList(sum([page.signatures for page in self.pages], []))
 
+    @property
+    def layouts(self) -> EntityList[Layout]:
+        """
+        Returns all the :class:`Layout` objects present in the Document
+
+        :return: List of Layout objects
+        :rtype: EntityList[Layout]
+        """
+        return EntityList(sum([page.layouts for page in self.pages], []))
+
     @property
     def identity_document(self) -> EntityList[IdentityDocument]:
         """