natural-pdf 0.2.10__py3-none-any.whl → 0.2.11__py3-none-any.whl

natural_pdf/analyzers/guides.py

@@ -4246,12 +4246,26 @@ class _ColumnAccessor:
         """Return number of columns (vertical guides - 1)."""
         return max(0, len(self._guides.vertical) - 1)
 
-    def __getitem__(self, index: int) -> "Region":
-        """Get column at the specified index."""
-        # Handle negative indexing
-        if index < 0:
-            index = len(self) + index
-        return self._guides.column(index)
+    def __getitem__(self, index: Union[int, slice]) -> Union["Region", "ElementCollection"]:
+        """Get column at the specified index or slice."""
+        from natural_pdf.elements.element_collection import ElementCollection
+
+        if isinstance(index, slice):
+            # Handle slice notation - return multiple columns
+            columns = []
+            num_cols = len(self)
+
+            # Convert slice to range of indices
+            start, stop, step = index.indices(num_cols)
+            for i in range(start, stop, step):
+                columns.append(self._guides.column(i))
+
+            return ElementCollection(columns)
+        else:
+            # Handle negative indexing
+            if index < 0:
+                index = len(self) + index
+            return self._guides.column(index)
 
 
 class _RowAccessor:
@@ -4264,12 +4278,26 @@ class _RowAccessor:
         """Return number of rows (horizontal guides - 1)."""
         return max(0, len(self._guides.horizontal) - 1)
 
-    def __getitem__(self, index: int) -> "Region":
-        """Get row at the specified index."""
-        # Handle negative indexing
-        if index < 0:
-            index = len(self) + index
-        return self._guides.row(index)
+    def __getitem__(self, index: Union[int, slice]) -> Union["Region", "ElementCollection"]:
+        """Get row at the specified index or slice."""
+        from natural_pdf.elements.element_collection import ElementCollection
+
+        if isinstance(index, slice):
+            # Handle slice notation - return multiple rows
+            rows = []
+            num_rows = len(self)
+
+            # Convert slice to range of indices
+            start, stop, step = index.indices(num_rows)
+            for i in range(start, stop, step):
+                rows.append(self._guides.row(i))
+
+            return ElementCollection(rows)
+        else:
+            # Handle negative indexing
+            if index < 0:
+                index = len(self) + index
+            return self._guides.row(index)
 
 
 class _CellAccessor:
@@ -4278,33 +4306,82 @@ class _CellAccessor:
     def __init__(self, guides: "Guides"):
         self._guides = guides
 
-    def __getitem__(self, key) -> Union["Region", "_CellRowAccessor"]:
+    def __getitem__(self, key) -> Union["Region", "_CellRowAccessor", "ElementCollection"]:
         """
         Get cell(s) at the specified position.
 
         Supports:
-        - guides.cells[row, col] - tuple indexing
-        - guides.cells[row][col] - nested indexing
+        - guides.cells[row, col] - single cell
+        - guides.cells[row][col] - single cell (nested)
+        - guides.cells[row, :] - all cells in a row
+        - guides.cells[:, col] - all cells in a column
+        - guides.cells[:, :] - all cells
+        - guides.cells[row][:] - all cells in a row (nested)
         """
+        from natural_pdf.elements.element_collection import ElementCollection
+
         if isinstance(key, tuple) and len(key) == 2:
-            # Direct tuple access: guides.cells[row, col]
             row, col = key
-            # Handle negative indexing for both row and col
-            if row < 0:
-                row = len(self._guides.rows) + row
-            if col < 0:
-                col = len(self._guides.columns) + col
-            return self._guides.cell(row, col)
+
+            # Handle slices for row and/or column
+            if isinstance(row, slice) or isinstance(col, slice):
+                cells = []
+                num_rows = len(self._guides.rows)
+                num_cols = len(self._guides.columns)
+
+                # Convert slices to ranges
+                if isinstance(row, slice):
+                    row_indices = range(*row.indices(num_rows))
+                else:
+                    # Single row index
+                    if row < 0:
+                        row = num_rows + row
+                    row_indices = [row]
+
+                if isinstance(col, slice):
+                    col_indices = range(*col.indices(num_cols))
+                else:
+                    # Single column index
+                    if col < 0:
+                        col = num_cols + col
+                    col_indices = [col]
+
+                # Collect all cells in the specified ranges
+                for r in row_indices:
+                    for c in col_indices:
+                        cells.append(self._guides.cell(r, c))
+
+                return ElementCollection(cells)
+            else:
+                # Both are integers - single cell access
+                # Handle negative indexing for both row and col
+                if row < 0:
+                    row = len(self._guides.rows) + row
+                if col < 0:
+                    col = len(self._guides.columns) + col
+                return self._guides.cell(row, col)
+        elif isinstance(key, slice):
+            # First level slice: guides.cells[:] - return all rows as accessors
+            # For now, let's return all cells flattened
+            cells = []
+            num_rows = len(self._guides.rows)
+            row_indices = range(*key.indices(num_rows))
+
+            for r in row_indices:
+                for c in range(len(self._guides.columns)):
+                    cells.append(self._guides.cell(r, c))
+
+            return ElementCollection(cells)
         elif isinstance(key, int):
             # First level of nested access: guides.cells[row]
             # Handle negative indexing for row
             if key < 0:
                 key = len(self._guides.rows) + key
-            # Return a row accessor that allows [col] indexing
+            # Return a row accessor that allows [col] or [:] indexing
             return _CellRowAccessor(self._guides, key)
         else:
             raise TypeError(
-                f"Cell indices must be integers or tuple of two integers, got {type(key)}"
+                f"Cell indices must be integers, slices, or tuple of two integers/slices, got {type(key)}"
             )
 
 
@@ -4315,9 +4392,24 @@ class _CellRowAccessor:
         self._guides = guides
         self._row = row
 
-    def __getitem__(self, col: int) -> "Region":
-        """Get cell at [row][col]."""
-        # Handle negative indexing for column
-        if col < 0:
-            col = len(self._guides.columns) + col
-        return self._guides.cell(self._row, col)
+    def __getitem__(self, col: Union[int, slice]) -> Union["Region", "ElementCollection"]:
+        """Get cell at [row][col] or all cells in row with [row][:]."""
+        from natural_pdf.elements.element_collection import ElementCollection
+
+        if isinstance(col, slice):
+            # Handle slice notation - return all cells in this row
+            cells = []
+            num_cols = len(self._guides.columns)
+
+            # Convert slice to range of indices
+            start, stop, step = col.indices(num_cols)
+            for c in range(start, stop, step):
+                cells.append(self._guides.cell(self._row, c))
+
+            return ElementCollection(cells)
+        else:
+            # Handle single column index
+            # Handle negative indexing for column
+            if col < 0:
+                col = len(self._guides.columns) + col
+            return self._guides.cell(self._row, col)

{natural_pdf-0.2.10.dist-info → natural_pdf-0.2.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: natural-pdf
-Version: 0.2.10
+Version: 0.2.11
 Summary: A more intuitive interface for working with PDFs
 Author-email: Jonathan Soma <jonathan.soma@gmail.com>
 License-Expression: MIT

{natural_pdf-0.2.10.dist-info → natural_pdf-0.2.11.dist-info}/RECORD

@@ -2,7 +2,7 @@ natural_pdf/__init__.py,sha256=N4pR0LbuPEnUYFZqbdVqc_FGKldgwPQc1wjJhYKTBBM,3417
 natural_pdf/cli.py,sha256=0zO9ZoRiP8JmyGBaVavrMATnvbARWTl7WD2PEefu9BM,4061
 natural_pdf/text_mixin.py,sha256=eFCiHj6Okcw3aum4955BepcI2NPRalkf9UFFVTc_H30,4012
 natural_pdf/analyzers/__init__.py,sha256=3XGoNq3OgiVkZP7tOdeP5XVUl7fDgyztdA8DlOcMLXg,1138
-natural_pdf/analyzers/guides.py,sha256=O3MaeVLgH5l1qbj2fpJCUsDKF44350CiUb_T1J6HTiQ,178025
+natural_pdf/analyzers/guides.py,sha256=GQZHjl7wUpaxAKRpv4cyQf1SJuccMSjpngBKFCKNHK4,181795
 natural_pdf/analyzers/shape_detection_mixin.py,sha256=mgpyJ4jIulz9l9HCqThabJIsLSrXh9BB2AmLxUoHmw0,62584
 natural_pdf/analyzers/text_options.py,sha256=qEkDaYWla0rIM_gszEOsu52q7C_dAfV81P2HLJZM2sw,3333
 natural_pdf/analyzers/text_structure.py,sha256=3WWusi-BI0krUnJxB05DD6XmKj5qRNvQBqH7zOQGm1M,28451
@@ -107,7 +107,7 @@ natural_pdf/vision/results.py,sha256=F2zXG3MVZIpOUvPkJHotOq6-9rFz68BaO_8pnSndlOs
 natural_pdf/vision/similarity.py,sha256=YH8legN-t9uf1b_XULi4JLNDaRfPNKQwU1FZ4Qu08jY,11740
 natural_pdf/widgets/__init__.py,sha256=QTVaUmsw__FCweFYZebwPssQxxUFUMd0wpm_cUbGZJY,181
 natural_pdf/widgets/viewer.py,sha256=KW3JogdR2TMg2ECUMYp8hwd060hfg8EsYBWxb5IEzBY,24942
-natural_pdf-0.2.10.dist-info/licenses/LICENSE,sha256=9zfwINwJlarbDmdh6iJV4QUG54QSJlSAUcnC1YiC_Ns,1074
+natural_pdf-0.2.11.dist-info/licenses/LICENSE,sha256=9zfwINwJlarbDmdh6iJV4QUG54QSJlSAUcnC1YiC_Ns,1074
 optimization/memory_comparison.py,sha256=0i_foFSRmppj-fY069qjwH36s_zkx-1L2ASAAlepWzA,6541
 optimization/pdf_analyzer.py,sha256=HjrmTgu2qchxPeDckc5kjgxppGwd40UESrYS9Myj7pY,19352
 optimization/performance_analysis.py,sha256=JBXnR9hc7Ix7YCnt3EJPSpsyqIUgKsc7GEffQ_TDCBk,13033
@@ -124,8 +124,8 @@ tools/bad_pdf_eval/llm_enrich.py,sha256=mCh4KGi1HmIkzGjj5rrHz1Osd7sEX1IZ_FW08H1t
 tools/bad_pdf_eval/llm_enrich_with_retry.py,sha256=XUtPF1hUvqd3frDXT0wDTXoonuAivhjM5vgFdZ-tm0A,9373
 tools/bad_pdf_eval/reporter.py,sha256=e1g__mkSB4q02p3mGWOwMhvFs7F2HJosNBxup0-LkyU,400
 tools/bad_pdf_eval/utils.py,sha256=hR95XQ7qf7Cu6BdyX0L7ggGVx-ah5sK0jHWblTJUUic,4896
-natural_pdf-0.2.10.dist-info/METADATA,sha256=DnjO1O3cHjhT-sXzLjIHQ286xqBAzeSJdPR50yol4D4,6960
-natural_pdf-0.2.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-natural_pdf-0.2.10.dist-info/entry_points.txt,sha256=1R_KMv7g60UBBpRqGfw7bppsMNGdayR-iJlb9ohEk_8,81
-natural_pdf-0.2.10.dist-info/top_level.txt,sha256=80t0F2ZeX4vN4Ke5iTflcOk_PN_0USn33ha3X6X86Ik,36
-natural_pdf-0.2.10.dist-info/RECORD,,
+natural_pdf-0.2.11.dist-info/METADATA,sha256=RBk1DrwdyTY0cJvFdWqfvM2hMd4OY30Lc0BlXopknD4,6960
+natural_pdf-0.2.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+natural_pdf-0.2.11.dist-info/entry_points.txt,sha256=1R_KMv7g60UBBpRqGfw7bppsMNGdayR-iJlb9ohEk_8,81
+natural_pdf-0.2.11.dist-info/top_level.txt,sha256=80t0F2ZeX4vN4Ke5iTflcOk_PN_0USn33ha3X6X86Ik,36
+natural_pdf-0.2.11.dist-info/RECORD,,