[Fix] Component table support State Vars (#571)

This commit is contained in:
Xiaojing Chen 2023-03-03 10:33:53 +08:00 committed by GitHub
parent 46cdda4125
commit 2e7fb4d082
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,6 +1,7 @@
"""Table components.""" """Table components."""
from pynecone.components.component import Component from pynecone.components.component import Component
from pynecone.components.layout.foreach import Foreach
from pynecone.components.libs.chakra import ChakraComponent from pynecone.components.libs.chakra import ChakraComponent
from pynecone.var import Var from pynecone.var import Var
@ -143,7 +144,10 @@ class Tr(ChakraComponent):
types = {"header": Th, "data": Td} types = {"header": Th, "data": Td}
cell_cls = types.get(cell_type) cell_cls = types.get(cell_type)
if len(children) == 0 and cell_cls: if len(children) == 0 and cell_cls:
children = [cell_cls.create(cell) for cell in cells or []] if isinstance(cells, Var):
children = [Foreach.create(cells, cell_cls.create)]
else:
children = [cell_cls.create(cell) for cell in cells or []]
return super().create(*children, **props) return super().create(*children, **props)