buildframework/helium/sf/python/pythoncore/lib/ats3/atsconfigparser.py
author wbernard
Fri, 13 Aug 2010 14:59:05 +0300
changeset 628 7c4a911dc066
parent 587 85df38eb4012
child 645 b8d81fa19e7d
permissions -rw-r--r--
helium_11.0.0-e00f171ca185

#============================================================================ 
#Name        : atsconfigparser.py
#Part of     : Helium 

#Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
#All rights reserved.
#This component and the accompanying materials are made available
#under the terms of the License "Eclipse Public License v1.0"
#which accompanies this distribution, and is available
#at the URL "http://www.eclipse.org/legal/epl-v10.html".
#
#Initial Contributors:
#Nokia Corporation - initial contribution.
#
#Contributors:
#
#Description:
#===============================================================================

""" handles the parameters  used in configuring the ATS """

import configuration
import amara

class TestXML:
    """ class used to create XML file"""
    def __init__(self, testxml):
        """ init file"""
        self.testxml = testxml
        self.doc = amara.parse(testxml)

    def containsproperty(self, name, value):
        """ returns the value of property if it exists or false"""
        for p_temp in self.doc.xml_xpath("//property"):
            if str(p_temp.name) == name:
                return str(p_temp.value) == value
        return False
    
    def containssetting(self, name, value):
        """ returns the value of setting if it exists or false"""
        for p_temp in self.doc.xml_xpath("//setting"):
            if str(p_temp.name) == name:
                return str(p_temp.value) == value
        return False
    
    def addorreplacesetting(self, name, value):
        """ Add or replace 'setting' value """
        changed = False
        for p_temp in self.doc.xml_xpath("//setting"):
            if str(p_temp.name) == name:
                p_temp.value = value
                changed = True
        if not changed:
            for device in self.doc.test.target.device:
                device.xml_append(self.doc.xml_create_element(u"setting", attributes = {u'name': unicode(name), u'value': unicode(value)}))
        
    def containsattribute(self, name, value):
        """ returns true or false """
        for p_temp in self.doc.xml_xpath("//*[@" + name + "]"):
            if p_temp[name] == value:
                return True
        return False
        
    def replaceattribute(self, name, value):
        """sets the xpath to the passed in value"""
        for p_temp in self.doc.xml_xpath("//*[@" + name + "]"):
            p_temp[name] = value
            
    def addorreplaceproperty(self, name, value):
        """ add or replace property value"""
        changed = False
        for p_temp in self.doc.xml_xpath("//property"):
            if str(p_temp.name) == name:
                p_temp.value = value
                changed = True
        if not changed:
            for device in self.doc.test.target.device:
                device.xml_append(self.doc.xml_create_element(u"property", attributes = {u'name': unicode(name), u'value': unicode(value)}))


class ATSConfigParser:
    """ ATS configuration parser"""
    def __init__(self, specfilename):
        specfile = open(specfilename)
        builder = configuration.NestedConfigurationBuilder(specfile)
        self.configs = builder.getConfigurations("common")

    def properties(self):
        """ retrieve the property values"""
        props = {}
        for config in self.configs:
            if (config.type == "properties"):
                for subconfig in config:
                    props[subconfig] = config[subconfig]
        return props

    def settings(self):
        """ retrieve the settings values"""
        settings = {}
        for config in self.configs:
            if (config.type == "settings"):
                for subconfig in config:
                    settings[subconfig] = config[subconfig]
        return settings

def converttestxml(specfilename, testxmldata): # pylint: disable=R0912
    """ convert the specfilename to xml"""
    specfile = open(specfilename)

    builder = configuration.NestedConfigurationBuilder(specfile)
    configs = builder.getConfigurations("common")# + builder.getConfigurations("ats3")

    testxml = TestXML(testxmldata)

    for config in configs:
        if (config.type == "properties"):
            for subconfig in config:
                testxml.addorreplaceproperty(subconfig, config[subconfig])
        if (config.type == "conditional_properties"):
            check = config.name.split(',')
            if testxml.containsproperty(check[0], check[1]):
                for subconfig in config:
                    testxml.addorreplaceproperty(subconfig, config[subconfig])
        if (config.type == "settings"):
            for subconfig in config:
                testxml.addorreplacesetting(subconfig, config[subconfig])
        if (config.type == "conditional_settings"):
            check = config.name.split(',')
            if testxml.containssetting(check[0], check[1]):
                for subconfig in config:
                    testxml.addorreplacesetting(subconfig, config[subconfig])
        if (config.type == "attributes"):
            for subconfig in config:
                testxml.replaceattribute(subconfig, config[subconfig])
        if (config.type == "conditional_attributes"):
            check = config.name.split(',')
            if testxml.containsattribute(check[0], check[1]):
                for subconfig in config:
                    testxml.replaceattribute(subconfig, config[subconfig])
    
    return testxml.doc.xml(indent=u"yes")