Skip to content

Commit 06fd3b3

Browse files
committed
feat(main): Add treeid field
Add treeid field handling Signed-off-by: Denys Fedoryshchenko <denys.f@collabora.com>
1 parent abf2609 commit 06fd3b3

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

api/main.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -633,13 +633,14 @@ async def put_node(node_id: str, node: Node,
633633

634634

635635
async def _set_node_ownership_recursively(user: User, hierarchy: Hierarchy,
636-
submitter: str):
636+
submitter: str, treeid: str):
637637
"""Set node ownership information for a hierarchy of nodes"""
638638
if not hierarchy.node.owner:
639639
hierarchy.node.owner = user.username
640640
hierarchy.node.submitter = submitter
641+
hierarchy.node.treeid = treeid
641642
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)
643644

644645

645646
@app.put('/nodes/{node_id}', response_model=List[Node],
@@ -658,8 +659,9 @@ async def put_nodes(
658659
detail=f"Node not found with id: {node_id}"
659660
)
660661
submitter = node_from_id.submitter
662+
treeid = node_from_id.treeid
661663

662-
await _set_node_ownership_recursively(user, nodes, submitter)
664+
await _set_node_ownership_recursively(user, nodes, submitter, treeid)
663665
obj_list = await db.create_hierarchy(nodes, Node)
664666
data = _get_node_event_data('updated', obj_list[0], True)
665667
attributes = {}

0 commit comments

Comments
 (0)