@@ -633,13 +633,14 @@ async def put_node(node_id: str, node: Node,
633
633
634
634
635
635
async def _set_node_ownership_recursively (user : User , hierarchy : Hierarchy ,
636
- submitter : str ):
636
+ submitter : str , treeid : str ):
637
637
"""Set node ownership information for a hierarchy of nodes"""
638
638
if not hierarchy .node .owner :
639
639
hierarchy .node .owner = user .username
640
640
hierarchy .node .submitter = submitter
641
+ hierarchy .node .treeid = treeid
641
642
for node in hierarchy .child_nodes :
642
- await _set_node_ownership_recursively (user , node , submitter )
643
+ await _set_node_ownership_recursively (user , node , submitter , treeid )
643
644
644
645
645
646
@app .put ('/nodes/{node_id}' , response_model = List [Node ],
@@ -658,8 +659,9 @@ async def put_nodes(
658
659
detail = f"Node not found with id: { node_id } "
659
660
)
660
661
submitter = node_from_id .submitter
662
+ treeid = node_from_id .treeid
661
663
662
- await _set_node_ownership_recursively (user , nodes , submitter )
664
+ await _set_node_ownership_recursively (user , nodes , submitter , treeid )
663
665
obj_list = await db .create_hierarchy (nodes , Node )
664
666
data = _get_node_event_data ('updated' , obj_list [0 ], True )
665
667
attributes = {}
0 commit comments