115 lines
4.6 KiB
Python
115 lines
4.6 KiB
Python
import os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import subprocess
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import platform
|
|
import re
|
|
import configparser
|
|
from time import sleep as delay
|
|
|
|
|
|
# Get the current directory
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
# Define the path to the firmware folder
|
|
firmware_folder = os.path.join(current_dir, 'firmware')
|
|
|
|
# Create the firmware folder if it does not exist
|
|
if not os.path.exists(firmware_folder):
|
|
os.makedirs(firmware_folder)
|
|
|
|
# Define the path to the release folder
|
|
release_folder = os.path.join(current_dir, 'release')
|
|
|
|
# Get the platformio path from environment variable
|
|
platformio_path = os.environ.get('PLATFORMIO_PATH')
|
|
# If the environment variable is not set, use the default path
|
|
if platformio_path is None:
|
|
if platform.system() == 'Darwin':
|
|
platformio_path = os.path.expanduser('~/.platformio/penv/bin/platformio')
|
|
elif platform.system() == 'Windows':
|
|
platformio_path = os.path.expanduser('~/.platformio/penv/Scripts/platformio.exe')
|
|
elif platform.system() == 'Linux':
|
|
platformio_path = os.path.expanduser('~/.platformio/penv/bin/platformio')
|
|
|
|
if os.path.exists(release_folder):
|
|
# If the release folder exists, delete it
|
|
shutil.rmtree(release_folder)
|
|
os.makedirs(release_folder)
|
|
|
|
# Get the current Git branch name
|
|
git_branch = subprocess.check_output(['git', 'rev-parse', '--abbrev-ref', 'HEAD']).decode().strip()
|
|
commit_hash = subprocess.check_output(['git', 'rev-parse', 'HEAD']).decode().strip()[:10]
|
|
|
|
# If arguments are supplied, build the environments that match the arguments
|
|
if len(sys.argv) > 1:
|
|
environments = sys.argv[1:]
|
|
|
|
# If no argument are supplied, build all environments listed in platformio.ini
|
|
else:
|
|
environments = []
|
|
with open(os.path.join(current_dir, 'platformio.ini')) as f:
|
|
|
|
# Iterate over the lines in platformio.ini
|
|
for line in f.readlines():
|
|
|
|
# Check if the line contains the string "env:"
|
|
if '[env:' in line:
|
|
|
|
# Extract the environment using regular expressions
|
|
environment = re.search(r'\[env:(.*?)\]', line).group(1)
|
|
|
|
# Add the environment to the list of environments
|
|
environments.append(environment)
|
|
|
|
# Remove old platformio.ini file from firmware folder if it exists
|
|
if os.path.isfile(os.path.join(firmware_folder, 'platformio.ini')):
|
|
os.remove(os.path.join(firmware_folder, 'platformio.ini'))
|
|
|
|
# Copy the platformio.ini file to the firmware folder
|
|
shutil.copyfile(os.path.join(current_dir, 'platformio.ini'), os.path.join(firmware_folder, 'platformio.ini'))
|
|
|
|
# Check that the file is copied correctly
|
|
if not os.path.isfile(os.path.join(firmware_folder, 'platformio.ini')):
|
|
raise Exception('platformio.ini file not copied correctly')
|
|
|
|
# Read the platformio.ini file
|
|
config = configparser.ConfigParser()
|
|
config.read(os.path.join(firmware_folder, 'platformio.ini'))
|
|
|
|
# Add firmware version to build_flags in platformio.ini
|
|
for environment in environments:
|
|
if 'build_flags' not in config[f'env:{environment}']:
|
|
config[f'env:{environment}']['build_flags'] = f'-DFW_VERSION=\\"{git_branch}_{environment}_{commit_hash}\\"'
|
|
else:
|
|
config[f'env:{environment}']['build_flags'] = config[f'env:{environment}']['build_flags'] + f' -DFW_VERSION=\\"{git_branch}_{environment}_{commit_hash}\\"'
|
|
|
|
# Write the platformio.ini file
|
|
with open(os.path.join(firmware_folder, 'platformio.ini'), 'w') as configfile:
|
|
config.write(configfile)
|
|
|
|
# if argument is not supplied, build all environments listed in platformio.ini
|
|
if len(sys.argv) == 1:
|
|
subprocess.run([f'{platformio_path}', 'run','-c',f'{firmware_folder}/platformio.ini'], cwd=current_dir)
|
|
|
|
for environment in environments:
|
|
if(len(sys.argv) > 1):
|
|
subprocess.run([f'{platformio_path}', 'run', '-e', environment,'-c',f'{firmware_folder}/platformio.ini'], cwd=current_dir)
|
|
|
|
# Iterate over the subfolders in the firmware folder
|
|
for subfolder in os.listdir(firmware_folder):
|
|
subfolder_path = os.path.join(firmware_folder, subfolder)
|
|
|
|
# Check if the subfolder matches the environment argument
|
|
if subfolder == environment:
|
|
# Check if the subfolder contains a firmware.bin file
|
|
firmware_file = os.path.join(subfolder_path, 'firmware.bin')
|
|
if os.path.isfile(firmware_file):
|
|
# Move the firmware.bin file to the release folder with the Git branch name and commit hash appended
|
|
new_file_name = os.path.join(release_folder, f"{git_branch}_{subfolder}_{commit_hash}.bin")
|
|
shutil.move(firmware_file, new_file_name) |