buildframework/helium/sf/python/pythoncore/lib/ats3/aste.py
changeset 628 7c4a911dc066
parent 588 c7c26511138f
equal deleted inserted replaced
588:c7c26511138f 628:7c4a911dc066
    19 #Description:
    19 #Description:
    20 #===============================================================================
    20 #===============================================================================
    21 
    21 
    22 """ASTE test drop generation."""
    22 """ASTE test drop generation."""
    23 
    23 
    24 # pylint: disable-msg=R0201,R0903,R0902,W0142
    24 
    25 #W0142 => * and ** were used
    25 #W0142 => * and ** were used
    26 #R* removed during refactoring
    26 #R* removed during refactoring
    27 
    27 
    28 from optparse import OptionParser
    28 from optparse import OptionParser
    29 from xml.etree import ElementTree as et
    29 from xml.etree import ElementTree as et
    31 import os
    31 import os
    32 import re
    32 import re
    33 import tempfile
    33 import tempfile
    34 import zipfile
    34 import zipfile
    35 import pkg_resources
    35 import pkg_resources
    36 from path import path # pylint: disable-msg=F0401
    36 from path import path # pylint: disable=F0401
    37 import amara
    37 import amara
    38 import ntpath as atspath
    38 import ntpath as atspath
    39 import jinja2 # pylint: disable-msg=F0401
    39 import jinja2 # pylint: disable=F0401
    40 
    40 
    41 _logger = logging.getLogger('ats3')
    41 _logger = logging.getLogger('ats3')
    42 
    42 
    43 # Shortcuts
    43 # Shortcuts
    44 E = et.Element
    44 E = et.Element
   373     
   373     
   374     def generate_xml(self, test_plan):
   374     def generate_xml(self, test_plan):
   375         """ generate an XML file"""
   375         """ generate an XML file"""
   376         loader = jinja2.ChoiceLoader([jinja2.PackageLoader(__name__, 'templates')])
   376         loader = jinja2.ChoiceLoader([jinja2.PackageLoader(__name__, 'templates')])
   377         env = jinja2.Environment(loader=loader)
   377         env = jinja2.Environment(loader=loader)
   378         template = env.from_string(pkg_resources.resource_string(__name__, 'aste_template.xml'))# pylint: disable-msg=E1101
   378         template = env.from_string(pkg_resources.resource_string(__name__, 'aste_template.xml'))# pylint: disable=E1101
   379         
   379         
   380         xmltext = template.render(test_plan=test_plan, os=os, atspath=atspath, atsself=self).encode('ISO-8859-1')
   380         xmltext = template.render(test_plan=test_plan, os=os, atspath=atspath, atsself=self).encode('ISO-8859-1')
   381         return et.ElementTree(et.XML(xmltext))
   381         return et.ElementTree(et.XML(xmltext))
   382 
   382 
   383 def create_drop(config):
   383 def create_drop(config):