Minor refactor for computed fields in tables (#325)

src/python-fastui/fastui/components/tables.py

@@ -43,22 +43,17 @@ def _fill_columns(self) -> _te.Self:
  except IndexError:
  raise ValueError('Cannot infer model from empty data, please set `Table(..., model=MyModel)`')
 
+ all_model_fields = {**data_model_type.model_fields, **data_model_type.model_computed_fields}
  if self.columns is None:
- self.columns = []
- for name, field in data_model_type.model_fields.items():
- self.columns.append(display.DisplayLookup(field=name, title=field.title))
- for name, field in data_model_type.model_computed_fields.items():
- self.columns.append(display.DisplayLookup(field=name, title=field.title))
-
+ self.columns = [
+ display.DisplayLookup(field=name, title=field.title) for name, field in all_model_fields.items()
+ ]
  else:
  # add pydantic titles to columns that don't have them
  for column in (c for c in self.columns if c.title is None):
- model_field = data_model_type.model_fields.get(column.field)
- computed_field = data_model_type.model_computed_fields.get(column.field)
- if model_field and model_field.title:
- column.title = model_field.title
- elif computed_field and computed_field.title:
- column.title = computed_field.title
+ field = all_model_fields.get(column.field)
+ if field and field.title:
+ column.title = field.title
  return self
 
  @classmethod