From fbb3b5c6e44603f92cc6288bb91c6cc718d94d37 Mon Sep 17 00:00:00 2001 From: Otto Winter Date: Thu, 24 Oct 2019 16:01:16 +0200 Subject: [PATCH] Faster upload (4x) --- esphomeflasher/__main__.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/esphomeflasher/__main__.py b/esphomeflasher/__main__.py index 0276de8..d39055f 100644 --- a/esphomeflasher/__main__.py +++ b/esphomeflasher/__main__.py @@ -22,6 +22,8 @@ def parse_args(argv): group = parser.add_mutually_exclusive_group(required=False) group.add_argument('--esp8266', action='store_true') group.add_argument('--esp32', action='store_true') + group.add_argument('--upload-baud-rate', type=int, default=460800, + help="Baud rate to upload with (not for logging)") parser.add_argument('--bootloader', help="(ESP32-only) The bootloader to flash.", default=ESP32_DEFAULT_BOOTLOADER_FORMAT) @@ -110,6 +112,12 @@ def run_esphomeflasher(argv): stub_chip = chip_run_stub(chip) + if args.upload_baud_rate != 115200: + try: + stub_chip.change_baud(args.upload_baud_rate) + except esptool.FatalError as err: + raise EsphomeflasherError("Error changing ESP upload baud rate: {}".format(err)) + flash_size = detect_flash_size(stub_chip) print(" - Flash Size: {}".format(flash_size))