如何在python程序中搜索id树

def search_id_tree(tree, target_id):
    if tree['id'] == target_id:
        return tree
    
    for child in tree.get('children', []):
        result = search_id_tree(child, target_id)
        if result:
            return result
    
    return None

tree = {
    'id': 1,
    'name': 'Root',
    'children': [
        {
            'id': 2,
            'name': 'Child 1',
            'children': [
                {
                    'id': 3,
                    'name': 'Grandchild 1',
                    'children': []
                },
                {
                    'id': 4,
                    'name': 'Grandchild 2',
                    'children': []
                }
            ]
        },
        {
            'id': 5,
            'name': 'Child 2',
            'children': []
        }
    ]
}