alihan/textual
1
0
Fork 0
mirror of https://github.com/Textualize/textual.git synced 2025-10-17 02:38:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Small optimisations

Browse Source
This commit is contained in:
Darren Burns
2023-02-14 16:30:46 +00:00
parent f19263a974
commit 18debed888
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/textual/widgets/_data_table.py
View File

@@ -672,10 +672,9 @@ class DataTable(ScrollView, Generic[CellType], can_focus=True):
if row_key not in self._row_locations:
raise RowDoesNotExist(f"Row key {row_key!r} is not valid.")
cell_mapping: dict[ColumnKey, CellType] = self._data.get(row_key, {})
ordered_row: list[CellType] = []
for column in self.ordered_columns:
cell = cell_mapping[column.key]
ordered_row.append(cell)
ordered_row: list[CellType] = [
cell_mapping[column.key] for column in self.ordered_columns
]
return ordered_row
def get_row_at(self, row_index: int) -> list[CellType]:
@@ -712,10 +711,10 @@ class DataTable(ScrollView, Generic[CellType], can_focus=True):
if column_key not in self._column_locations:
raise ColumnDoesNotExist(f"Column key {column_key!r} is not valid.")
data = self._data
for row_metadata in self.ordered_rows:
row_key = row_metadata.key
row = self._data[row_key]
yield row[column_key]
yield data[row_key][column_key]
def get_column_at(self, column_index: int) -> Iterable[CellType]:
"""Get the values from the column at a given index.