64d5045ccb
* simplify post_esp32.py * make sure path has correct "\" or "/" regarding OS * add os specific path separators * more path possible issues corrections * add function to normpath cmd * set board_build.variants_dir = variants/tasmota correctly for OS
79 lines
3.0 KiB
Python
79 lines
3.0 KiB
Python
Import("env")
|
|
|
|
import pathlib
|
|
import shutil
|
|
import tasmotapiolib
|
|
import gzip
|
|
from colorama import Fore, Back, Style
|
|
|
|
def map_gzip(source, target, env):
|
|
# create string with location and file names based on variant
|
|
map_file = pathlib.Path(tasmotapiolib.get_final_map_path(env))
|
|
|
|
if map_file.is_file():
|
|
gzip_file = map_file.with_suffix(".map.gz")
|
|
|
|
# check if new target map files exist and remove if necessary
|
|
if gzip_file.is_file():
|
|
gzip_file.unlink()
|
|
|
|
# write gzip map file
|
|
with map_file.open("rb") as fp:
|
|
with gzip.open(str(gzip_file), "wb", compresslevel=9) as f:
|
|
shutil.copyfileobj(fp, f)
|
|
|
|
# remove map file
|
|
if map_file.is_file():
|
|
map_file.unlink()
|
|
|
|
|
|
if not tasmotapiolib.is_env_set(tasmotapiolib.DISABLE_MAP_GZ, env):
|
|
silent_action = env.Action([map_gzip])
|
|
silent_action.strfunction = lambda target, source, env: '' # hack to silence scons command output
|
|
env.AddPostAction("$BUILD_DIR/${PROGNAME}.bin", silent_action)
|
|
|
|
if tasmotapiolib.is_env_set(tasmotapiolib.ENABLE_ESP32_GZ, env) or env["PIOPLATFORM"] != "espressif32":
|
|
import time
|
|
|
|
gzip_level = int(env['ENV'].get('GZIP_LEVEL', 10))
|
|
|
|
def bin_gzip(source, target, env):
|
|
# create string with location and file names based on variant
|
|
bin_file = pathlib.Path(tasmotapiolib.get_final_bin_path(env))
|
|
gzip_file = bin_file.with_suffix(".bin.gz")
|
|
|
|
# check if new target files exist and remove if necessary
|
|
if gzip_file.is_file():
|
|
gzip_file.unlink()
|
|
|
|
# write gzip firmware file
|
|
with bin_file.open("rb") as fp:
|
|
with gzip_file.open("wb") as f:
|
|
time_start = time.time()
|
|
gz = tasmotapiolib.compress(fp.read(), gzip_level)
|
|
time_delta = time.time() - time_start
|
|
f.write(gz)
|
|
|
|
ORG_FIRMWARE_SIZE = bin_file.stat().st_size
|
|
GZ_FIRMWARE_SIZE = gzip_file.stat().st_size
|
|
|
|
print()
|
|
if ORG_FIRMWARE_SIZE > 995326 and env["PIOPLATFORM"] != "espressif32":
|
|
print(Fore.RED + "!!! Tasmota firmware size is too big with {} bytes. Max size is 995326 bytes !!! ".format(
|
|
ORG_FIRMWARE_SIZE
|
|
)
|
|
)
|
|
else:
|
|
print(Fore.GREEN + "Compression reduced firmware size to {:.0f}% (was {} bytes, now {} bytes, took {:.3f} seconds)".format(
|
|
(GZ_FIRMWARE_SIZE / ORG_FIRMWARE_SIZE) * 100,
|
|
ORG_FIRMWARE_SIZE,
|
|
GZ_FIRMWARE_SIZE,
|
|
time_delta,
|
|
)
|
|
)
|
|
|
|
if not tasmotapiolib.is_env_set(tasmotapiolib.DISABLE_BIN_GZ, env):
|
|
silent_action = env.Action([bin_gzip])
|
|
silent_action.strfunction = lambda target, source, env: '' # hack to silence scons command output
|
|
env.AddPostAction("$BUILD_DIR/${PROGNAME}.bin", silent_action)
|