diff --git a/bootloader/crc_patch.py b/bootloader/crc_patch.py index 4ba61e8e93c2360a82ca4990a807f21bf19b44df..7d93353d507531c097a3e69d2186c15c1b8ebded 100755 --- a/bootloader/crc_patch.py +++ b/bootloader/crc_patch.py @@ -3,21 +3,26 @@ import sys try: import crc16 + crcfun = crc16.crc16xmodem except ImportError: try: import crcmod - crcfun = crcmod.predefined.mkCrcFun('xmodem') + + crcfun = crcmod.predefined.mkCrcFun("xmodem") except ImportError: try: import crcelk + crcfun = crcelk.CRC_XMODEM.calc_bytes except ImportError: - raise Exception('Could not find a CRC implementation. Tried: crc16, crcmod, crcelk.') + raise Exception( + "Could not find a CRC implementation. Tried: crc16, crcmod, crcelk." + ) def main(): - data = open(sys.argv[1], 'rb').read() + data = open(sys.argv[1], "rb").read() crc = crcfun(data) # print(crc) @@ -26,7 +31,7 @@ def main(): crc = crcfun(padded) # print(crc) - open(sys.argv[1], 'wb').write(padded) + open(sys.argv[1], "wb").write(padded) if __name__ == "__main__":