在 PySpark 中将层次树菜单生成为 json

如何解决在 PySpark 中将层次树菜单生成为 json

我正在努力弄清楚如何使用 PySpark 为分层菜单生成嵌套的 json。

我有一个存储父子关系的 Athena 表。父子关系用于在网站上构建菜单。该表存储所有站点的关系。

data = [
  { 'id': 1,'parent_id': null,'name': "Section 1",'site_id': 1},{ 'id': 2,'parent_id': 1,'name': "Module 1",'site_id': 1 },{ 'id': 3,'name': "Module 2",{ 'id': 4,'parent_id': 3,'name': "Item in Module 1",{ 'id': 5,'site_id': 2},{ 'id': 6,'parent_id': 5,'site_id': 2 },{ 'id': 7,{ 'id': 8,'parent_id': 7,]

我需要做的是为每个站点获取一个嵌套的 JSON 语句并保存它？

例如：

[ 
{ 'id': 1,'site_id': 1,'name': 'Section 1','children':
   [ { 'id':2,'name': "Module 1" },'children':[ 
                       { 'id': 4,'name': "Item in Module 1"},]  
     } 
   ]
} 
]

非常感谢