diff --git a/bbcli/endpoints.py b/bbcli/endpoints.py index abaf3ced4c59bd24e21536de0d000c36bcc36675..2656e1670a495969f4bf1dd314ac1a0e3148e484 100644 --- a/bbcli/endpoints.py +++ b/bbcli/endpoints.py @@ -86,7 +86,7 @@ def get_course_contents(course_id: str): click.echo(map) -def get_children(session, worklist, url, acc, count: int = 0, is_root = True): +def get_children(session, worklist, url, acc, count: int = 0): count = count + 1 key = 'hasChildren' if len(worklist) == 0: @@ -94,10 +94,14 @@ def get_children(session, worklist, url, acc, count: int = 0, is_root = True): else: node = worklist.pop() id = node.data['id'] - if is_root is False: + if 'parentId' in node.data: old = f'{url}/children' else: old = f'{url}/{id}/children' + # if is_root is False: + # old = f'{url}/children' + # else: + # old = f'{url}/{id}/children' response = session.get(old, cookies=cli.cookies) if check_response(response) == False: return acc @@ -184,7 +188,7 @@ def get_assignments(course_id: str, folder_id=None): # print(len(data)) root = Node(data, True) worklist = [root] - res = get_children(session, worklist, url, [], is_root = False) + res = get_children(session, worklist, url, []) create_tree(root, res) else: data = response.json()['results'] @@ -192,6 +196,9 @@ def get_assignments(course_id: str, folder_id=None): root = Node(root, True) worklist = [root] res = get_children(session, worklist, url, []) + # for r in res: + # print() + # print(r.data['title']) create_tree(root, res) end = time.time()