Creating A Dynamic nested Json Using For Loops - Python

I debugged your code, it was kind of strange but I worked it out. Your code only works for one Ward at the moment, I've written my own version to show you a way to do this for multiple wards.

I commented your code to show you the major changes.

import collections
def add_element(root, path, data):
    if len(path) == 1:
        root[path[0]] = data
    else:
        add_element(root[path[0]], path[1:], data)


def create_tree(Wards,WardName,Rooms):
    count = 1
    ward_list=[]
    room_list= []
    tree = lambda: collections.defaultdict(tree)
    root = tree()
    path_list= ['BuildingGroup', 'Building1']
    for i in range(1,Wards):
        Ward = 'Ward' + str(count)
        # Check your typos, this line had a ',' in it instead of a '.'
        path_list.append(Ward)
        ward_list.append(Ward)
        print (ward_list)
        print (path_list)
        count += 1
        add_element(root,path_list, 1 )
    path_list.append(WardName)
    for i in range(1, Rooms):
        # Initialise room_list with the path_list from the ward
        room_list = path_list.copy()
        # We have a new variable, 'i' for counting rooms, so don't use 'count' here as it never changes
        Room = 'Room' + str(i)
        # path_list.append(Room) - No! Bad! Don't change the path_list, this is the same for each room
        room_list.append(Room)
        # Use the room_list here instead of the path_list
        add_element(root,room_list, 1 )
        print(root)

my_tree = create_tree(1, 'Ward1', 10)

Now here's my version:

import json


def create_tree(ward_list, number_of_rooms):
    tree = {'BuildingGroup': {'Building1': {}}}
    for ward in ward_list:
        tree['BuildingGroup']['Building1'][ward] = []
        for i in range(1, number_of_rooms + 1):
            tree['BuildingGroup']['Building1'][ward].append("Room" + str(i))
    return tree


wards = ['Ward1', 'Ward2']
my_tree = create_tree(wards, 10)
tree_json = json.dumps(my_tree)
print(tree_json)