From 9698fbbc9ae8f5babbe277b56d17132902b3d664 Mon Sep 17 00:00:00 2001
From: Hasan Ramezani <hasan.r67@gmail.com>
Date: Sun, 5 May 2024 22:58:04 +0200
Subject: [PATCH] Respect computed_field title in Table and Details

---
 .../fastui/components/display.py              |  8 ++--
 .../tests/test_tables_display.py              | 38 +++++++++++++++++++
 2 files changed, 42 insertions(+), 4 deletions(-)

diff --git a/src/python-fastui/fastui/components/display.py b/src/python-fastui/fastui/components/display.py
index 5f147def..794dddb1 100644
--- a/src/python-fastui/fastui/components/display.py
+++ b/src/python-fastui/fastui/components/display.py
@@ -79,14 +79,14 @@ class Details(BaseModel, extra='forbid'):
 
     @pydantic.model_validator(mode='after')
     def _fill_fields(self) -> _te.Self:
+        fields = {**self.data.model_fields, **self.data.model_computed_fields}
+
         if self.fields is None:
-            self.fields = [
-                DisplayLookup(field=name, title=field.title) for name, field in self.data.model_fields.items()
-            ]
+            self.fields = [DisplayLookup(field=name, title=field.title) for name, field in fields.items()]
         else:
             # add pydantic titles to fields that don't have them
             for field in (c for c in self.fields if c.title is None):
-                pydantic_field = self.data.model_fields.get(field.field)
+                pydantic_field = fields.get(field.field)
                 if pydantic_field and pydantic_field.title:
                     field.title = pydantic_field.title
         return self
diff --git a/src/python-fastui/tests/test_tables_display.py b/src/python-fastui/tests/test_tables_display.py
index c1c2274b..ed76bf9a 100644
--- a/src/python-fastui/tests/test_tables_display.py
+++ b/src/python-fastui/tests/test_tables_display.py
@@ -102,3 +102,41 @@ def test_display_fields():
         'fields': [{'title': 'ID', 'field': 'id'}, {'title': 'Name', 'field': 'name'}],
         'type': 'Details',
     }
+
+
+def test_table_respect_computed_field_title():
+    class Foo(BaseModel):
+        id: int
+
+        @computed_field(title='Foo Name')
+        def name(self) -> str:
+            return f'foo{self.id}'
+
+    foos = [Foo(id=1)]
+    table = components.Table(data=foos)
+
+    # insert_assert(table.model_dump(by_alias=True, exclude_none=True))
+    assert table.model_dump(by_alias=True, exclude_none=True) == {
+        'data': [{'id': 1, 'name': 'foo1'}],
+        'columns': [{'field': 'id'}, {'title': 'Foo Name', 'field': 'name'}],
+        'type': 'Table',
+    }
+
+
+def test_details_respect_computed_field_title():
+    class Foo(BaseModel):
+        id: int
+
+        @computed_field(title='Foo Name')
+        def name(self) -> str:
+            return f'foo{self.id}'
+
+    foos = Foo(id=1)
+    details = components.Details(data=foos)
+
+    # insert_assert(table.model_dump(by_alias=True, exclude_none=True))
+    assert details.model_dump(by_alias=True, exclude_none=True) == {
+        'data': {'id': 1, 'name': 'foo1'},
+        'fields': [{'field': 'id'}, {'title': 'Foo Name', 'field': 'name'}],
+        'type': 'Details',
+    }