Notes

def modify_all_threads(jmx_file, thread_data):
tree = ET.parse(jmx_file)
root = tree.getroot()

# Find all ThreadGroup elements
thread_groups = root.findall(".//ThreadGroup")

for thread_id, attributes in thread_data.items():
# Find the corresponding ThreadGroup element
thread_group = thread_groups[thread_id - 1] # Adjust index to start from 0

# Update attributes for the ThreadGroup element
for attr_name, attr_value in attributes.items():
if attr_name not in ["name", "comment"]:
attr_element = thread_group.find(f".//stringProp[@name='ThreadGroup.{attr_name}']")
if attr_element is None:
print(f"{attr_name} not found for ThreadGroup {thread_id}")
continue
attr_element.text = str(attr_value)

# Update num_threads value
num_threads_element = thread_group.find(".//stringProp[@name='ThreadGroup.num_threads']")
if num_threads_element is not None:
num_threads_element.text = str(attributes.get('num_threads', 0))
else:
print(f"num_threads not found for ThreadGroup {thread_id}")

tree.write(jmx_file)