import json, os
import xml.etree.ElementTree as ET
import xml.dom.minidom as minidom

from org.apache.jmeter.services import FileServer
projectPath = FileServer.getFileServer().getBaseDir()
fileName = os.path.join(projectPath, "test-data/admin-accounts.xml")

responseBody = json.loads(prev.getResponseDataAsString())
data = responseBody['data']

createFile = False
doBeauty = True
parentID = None
for arg in args:
    if '=' in arg:
        name, value = arg.split('=')
        exec("%s = \"%s\"" % (name, value))

# to XML
if createFile or not os.path.isfile(fileName):
    root = ET.Element('admin-accounts')
else:
    tree = ET.parse(fileName)
    root = tree.getroot()

# get root
if parentID != None:
    node = root.find(".//account[@id='%s']" % parentID)
    if node == None:
        node = ET.SubElement(root, 'account')
        node.attrib['id'] = parentID
else:
    node = root

for i in range(0, len(data)):
	account = ET.SubElement(node, 'account')
	account.attrib['id'] = str(data[i]['accountId'])
	account.attrib['name'] = data[i]['username']
	account.attrib['type'] = data[i]['type'].lower()

# beautifier
beautifier = minidom.parseString(ET.tostring(root, 'utf-8').replace('\n', "").replace('\t', "").replace('\r', ""))

# save XML to file
f = open(fileName, "w")
if doBeauty:
    beautifier.writexml(f, indent="\n", addindent ="\t")
else:
    beautifier.writexml(f)
f.close()