copy_fw.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. #This script is based on the Tasmota rename-firmware.py script. https://github.com/arendst/Tasmota
  2. Import('env')
  3. import os
  4. import shutil
  5. buildFlags = env.ParseFlags(env['BUILD_FLAGS'])
  6. def get_fw_version(source, target, env):
  7. global HASP_VER_MAJ
  8. global HASP_VER_MIN
  9. global HASP_VER_REV
  10. for item in buildFlags.get("CPPDEFINES"):
  11. if (type(item) is list):
  12. if (item[0]=="HASP_VER_MAJ"): HASP_VER_MAJ = item[1]
  13. elif (item[0]=="APP_VER_MAJ"): HASP_VER_MAJ = item[1]
  14. if (item[0]=="HASP_VER_MIN"): HASP_VER_MIN = item[1]
  15. elif (item[0]=="APP_VER_MIN"): HASP_VER_MIN = item[1]
  16. if (item[0]=="HASP_VER_REV"): HASP_VER_REV = item[1]
  17. elif (item[0]=="APP_VER_REV"): HASP_VER_REV = item[1]
  18. print(" * %s = %s" % (item[0],item[1]))
  19. else:
  20. print(" * %s" % item)
  21. OUTPUT_DIR = "build_output{}".format(os.path.sep)
  22. def bin_copy_rename(source, target, env):
  23. variant = str(target[0]).split(os.path.sep)[2] + '_v' + str(HASP_VER_MAJ) + '.' + str(HASP_VER_MIN) + '.' + str(HASP_VER_REV)
  24. # check if output directories exist and create if necessary
  25. if not os.path.isdir(OUTPUT_DIR):
  26. os.mkdir(OUTPUT_DIR)
  27. for d in ['firmware', 'map']:
  28. if not os.path.isdir("{}{}".format(OUTPUT_DIR, d)):
  29. os.mkdir("{}{}".format(OUTPUT_DIR, d))
  30. # create string with location and file names based on variant
  31. bin_file = "{}firmware{}{}.bin".format(OUTPUT_DIR, os.path.sep, variant)
  32. # check if new target files exist and remove if necessary
  33. for f in [bin_file]:
  34. if os.path.isfile(f):
  35. os.remove(f)
  36. # copy firmware.bin to firmware/<variant>.bin
  37. shutil.copy(str(target[0]), bin_file)
  38. env.AddPreAction("$BUILD_DIR/${PROGNAME}.bin", [get_fw_version])
  39. env.AddPostAction("$BUILD_DIR/${PROGNAME}.bin", [bin_copy_rename])