def level_generator(number_of_levels):
    levels = []
    rooms_1 = open("rooms/rooms_1.txt", 'r')
    rooms_2 = None
    rooms_3 = None
    for i in range(number_of_levels):
        rooms = room_generator(rooms_1, i + 1)
        levels.append(Level(rooms, i + 1))
    rooms_1.close()
    return levels


def room_generator(rooms_d, lvl_no):
    rooms_out = []
    if lvl_no == 1:
        for i in range(3):
            rooms_in = []
            for j in range(2):
                rooms_desc = (rooms_d.readline().rstrip(), rooms_d.readline().rstrip())
                if i == 2 and j == 0:
                    rooms_in.append(Room(rooms_desc[0], rooms_desc[1], j, i, [], Quest(quests.check_number), Enemy(
                        "troll", {"health": 200, "strength": 20}
                    )))
                else:
                    rooms_in.append(Room(rooms_desc[0], rooms_desc[1], j, i, [], None, None))
            rooms_out.append(rooms_in)

    if lvl_no == 2:
        for i in range(3):
            rooms_in = []
            for j in range(2):
                rooms_desc = (rooms_d.readline().rstrip(), rooms_d.readline().rstrip())
                if i == 2 and j == 0:
                    rooms_in.append(Room(rooms_desc[0], rooms_desc[1], j, i, [], Quest(quests.check_number), Enemy(
                        "troll", {"health": 200, "strength": 20}
                    )))
                else:
                    rooms_in.append(Room(rooms_desc[0], rooms_desc[1], j, i, [], None, None))
            rooms_out.append(rooms_in)

    if lvl_no == 3:
        for i in range(3):
            rooms_in = []
            for j in range(2):
                rooms_desc = (rooms_d.readline().rstrip(), rooms_d.readline().rstrip())
                if i == 2 and j == 0:
                    rooms_in.append(Room(rooms_desc[0], rooms_desc[1], j, i, [], Quest(quests.check_number), Enemy(
                        "troll", {"health": 200, "strength": 20}
                    )))
                else:
                    rooms_in.append(Room(rooms_desc[0], rooms_desc[1], j, i, [], None, None))
            rooms_out.append(rooms_in)

    return rooms_out