def stub_grouped_loans(data):
loans = []
for item in data:
loan = {}
for key in item:
loan[key] = item[key]
path = loan['groupName'].split('-')
if len(path) == 1:
loans.append(loan)
else:
parent_loan = find_parent(loans, path)
if not ('children' in parent_loan):
parent_loan['children'] = []
parent_loan['children'].append(loan)
stubs = [StubFactory.make_group_loans_stub(loans)]
return stubs
def stub_grouped_loans_by_count(count):
loans = generate_grouped_loans(int(count))
stubs = [StubFactory.make_group_loans_stub(loans)]
return stubs
