Add number of columns subtitle (#14)

Co-authored-by: thread53 <[email protected]>

pqviewer/app.py

@@ -52,14 +52,15 @@ def watch_theme(self):
         datatable_schema.styles.scrollbar_color_active = color_hex
         datatable_schema.styles.scrollbar_color_hover = color_hex
         datatable_schema.border_title = "Schema"
+        datatable_schema.border_subtitle = f"Total columns: {self.parsed_file.number_columns()}"
 
         datatable_content = self.query_one("#datatable_content")
         datatable_content.styles.border = ("heavy", color_hex)
         datatable_content.styles.scrollbar_color = color_hex
         datatable_content.styles.scrollbar_color_active = color_hex
         datatable_content.styles.scrollbar_color_hover = color_hex
         datatable_content.border_title = "Table"
-        datatable_content.border_subtitle = f"Total rows: {len(self.parsed_file)}"
+        datatable_content.border_subtitle = f"Total rows: {self.parsed_file.number_rows()}"
 
     def action_help_screen(self) -> None:
         """Action to display the help screen dialog."""

pqviewer/file_parser.py

@@ -16,12 +16,18 @@ def schema(self) -> List[tuple]:
             for name in self.table.column_names
         ]
 
+    def number_columns(self) -> int:
+        """Get number of columns."""
+        return self.table.num_columns
+
+    def number_rows(self) -> int:
+        """Get number of rows."""
+        return self.table.num_rows
+
     def to_table(self) -> List[tuple]:
         """Transform table content to list."""
 
         table_header = [tuple(self.table.column_names)]
         table_content = [tuple(row.values()) for row in self.table.to_pylist()]
         return table_header + table_content
 
-    def __len__(self) -> int:
-        return len(self.table)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pqviewer"
-version = "0.2.6"
+version = "0.2.7"
 description = "View Apache Parquet Files In Your Terminal"
 authors = ["thread53 <[email protected]>"]
 license = "MIT"
@@ -16,7 +16,7 @@ classifiers = [
 [tool.poetry.dependencies]
 python = "^3.8"
 textual = "==0.46.0"
-pyarrow = "==14.0.1"
+pyarrow = ">=17.0.0"
 
 
 [build-system]

requirements-dev.txt

@@ -1,3 +1,3 @@
 pytest==8.0.0
 textual==0.46.0
-pyarrow==14.0.1
+pyarrow>=17.0.0