merge gzip into firmware rename function

This commit is contained in:
proddy
2020-11-28 18:43:51 +01:00
parent d31758f69b
commit dc5c33c3c3
3 changed files with 22 additions and 60 deletions

View File

@@ -1,17 +1,16 @@
Import('env')
Import('env', "projenv")
import os
import re
import shutil
import gzip
OUTPUT_DIR = "build{}".format(os.path.sep)
def bin_copy(source, target, env):
# get the version
# get the EMS-ESP version
bag = {}
exprs = [
(re.compile(r'^#define EMSESP_APP_VERSION\s+"(\S+)"'), 'app_version'),
]
exprs = [(re.compile(r'^#define EMSESP_APP_VERSION\s+"(\S+)"'), 'app_version')]
with open('./src/version.h', 'r') as f:
for l in f.readlines():
for expr, var in exprs:
@@ -19,27 +18,29 @@ def bin_copy(source, target, env):
if m and len(m.groups()) > 0:
bag[var] = m.group(1)
app_version = bag.get('app_version')
variant = "EMS-ESP-" + app_version.replace(".", "_") + "-" + str(target[0]).split(os.path.sep)[2]
# esp8266 or esp32
platform = "esp" + env['PIOPLATFORM'].strip("espressif")
# str(target[0]).split(os.path.sep)[2]
variant = "EMS-ESP-" + bag.get('app_version').replace(".", "_") + "-" + platform
# check if output directories exist and create if necessary
if not os.path.isdir(OUTPUT_DIR):
os.mkdir(OUTPUT_DIR)
for d in ['firmware']:
if not os.path.isdir("{}{}".format(OUTPUT_DIR, d)):
os.mkdir("{}{}".format(OUTPUT_DIR, d))
# create string with location and file names based on variant
bin_file = "{}firmware{}{}.bin".format(OUTPUT_DIR, os.path.sep, variant)
gzip_file = "{}firmware{}{}.bin.gz".format(OUTPUT_DIR, os.path.sep, variant)
# check if new target files exist and remove if necessary
for f in [bin_file]:
if os.path.isfile(f):
os.remove(f)
if os.path.isfile(bin_file): os.remove(bin_file)
if os.path.isfile(gzip_file): os.remove(gzip_file)
# copy firmware.bin to firmware/<variant>.bin
# copy firmware.bin to new name
shutil.copy(str(target[0]), bin_file)
# create a zip'd version
with open(bin_file, "rb") as fp:
with gzip.open(gzip_file, "wb", compresslevel = 9) as f:
shutil.copyfileobj(fp, f)
print("Built firmwares: "+ bin_file + ", " + gzip_file)
env.AddPostAction("$BUILD_DIR/${PROGNAME}.bin", [bin_copy])