108 lines
4.0 KiB
Python
108 lines
4.0 KiB
Python
"""
|
|
mbed SDK
|
|
Copyright (c) 2011-2013 ARM Limited
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
"""
|
|
|
|
import re
|
|
from os.path import join
|
|
from host_test_plugins import HostTestPluginBase
|
|
|
|
|
|
class HostTestPluginCopyMethod_MPS2(HostTestPluginBase):
|
|
|
|
# MPS2 specific flashing / binary setup funcitons
|
|
def mps2_set_board_image_file(self, disk, images_cfg_path, image0file_path, image_name='images.txt'):
|
|
""" This function will alter image cfg file.
|
|
Main goal of this function is to change number of images to 1, comment all
|
|
existing image entries and append at the end of file new entry with test path.
|
|
@return True when all steps succeed.
|
|
"""
|
|
MBED_SDK_TEST_STAMP = 'test suite entry'
|
|
image_path = join(disk, images_cfg_path, image_name)
|
|
new_file_lines = [] # New configuration file lines (entries)
|
|
|
|
# Check each line of the image configuration file
|
|
try:
|
|
with open(image_path, 'r') as file:
|
|
for line in file:
|
|
if re.search('^TOTALIMAGES', line):
|
|
# Check number of total images, should be 1
|
|
new_file_lines.append(re.sub('^TOTALIMAGES:[\t ]*[\d]+', 'TOTALIMAGES: 1', line))
|
|
elif re.search('; - %s[\n\r]*$'% MBED_SDK_TEST_STAMP, line):
|
|
# Look for test suite entries and remove them
|
|
pass # Omit all test suite entries
|
|
elif re.search('^IMAGE[\d]+FILE', line):
|
|
# Check all image entries and mark the ';'
|
|
new_file_lines.append(';' + line) # Comment non test suite lines
|
|
else:
|
|
# Append line to new file
|
|
new_file_lines.append(line)
|
|
except IOError as e:
|
|
return False
|
|
|
|
# Add new image entry with proper commented stamp
|
|
new_file_lines.append('IMAGE0FILE: %s ; - %s\r\n'% (image0file_path, MBED_SDK_TEST_STAMP))
|
|
|
|
# Write all lines to file
|
|
try:
|
|
with open(image_path, 'w') as file:
|
|
for line in new_file_lines:
|
|
file.write(line),
|
|
except IOError:
|
|
return False
|
|
|
|
return True
|
|
|
|
def mps2_select_core(self, disk, mobo_config_name=""):
|
|
""" Function selects actual core
|
|
"""
|
|
# TODO: implement core selection
|
|
pass
|
|
|
|
def mps2_switch_usb_auto_mounting_after_restart(self, disk, usb_config_name=""):
|
|
""" Function alters configuration to allow USB MSD to be mounted after restarts
|
|
"""
|
|
# TODO: implement USB MSD restart detection
|
|
pass
|
|
|
|
# Plugin interface
|
|
name = 'HostTestPluginCopyMethod_MPS2'
|
|
type = 'CopyMethod'
|
|
capabilities = ['mps2']
|
|
required_parameters = ['image_path', 'destination_disk']
|
|
|
|
def setup(self, *args, **kwargs):
|
|
""" Configure plugin, this function should be called before plugin execute() method is used.
|
|
"""
|
|
return True
|
|
|
|
def execute(self, capabilitity, *args, **kwargs):
|
|
""" Executes capability by name.
|
|
Each capability may directly just call some command line
|
|
program or execute building pythonic function
|
|
"""
|
|
result = False
|
|
if self.check_parameters(capabilitity, *args, **kwargs) is True:
|
|
if capabilitity == 'mps2':
|
|
# TODO: Implement MPS2 firmware setup here
|
|
pass
|
|
return result
|
|
|
|
|
|
def load_plugin():
|
|
""" Returns plugin available in this module
|
|
"""
|
|
return HostTestPluginCopyMethod_MPS2()
|