import monsoon.reflash as reflash

######################################
# Reflash unit with USB Protocol firmware
######################################
Mon = reflash.bootloaderMonsoon()
Mon.setup_usb()
Header, Hex = Mon.getHeaderFromFWM('LVPM_RevE_Prot1_Ver21_Beta.fwm')
if(Mon.verifyHeader(Header)):
   Mon.writeFlash(Hex)

######################################
# Return to the serial protocol firmware.
######################################
Mon = reflash.bootloaderMonsoon()
Mon.setup_usb()
Hex = Mon.getHexFile('PM_RevD_Prot17_Ver20.hex')
Mon.writeFlash(Hex)