Add logic to revert to 115200 upload rate if faster one fails (#16)

.gitignore

@@ -104,3 +104,6 @@ venv.bak/
 .mypy_cache/
 
 config/
+
+.DS_Store
+/.idea/

esphomeflasher/__main__.py

@@ -112,14 +112,28 @@ def run_esphomeflasher(argv):
     print(" - MAC Address: {}".format(info.mac))
 
     stub_chip = chip_run_stub(chip)
+    flash_size = None
 
     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)
+        # Check if the higher baud rate works
+        try:
+            flash_size = detect_flash_size(stub_chip)
+        except esptool.FatalError as err:
+            # Go back to old baud rate by recreating chip instance
+            print("Chip does not support baud rate {}, changing to 115200".format(args.upload_baud_Rate))
+            stub_chip._port.close()
+            chip = detect_chip(port, args.esp8266, args.esp32)
+            stub_chip = chip_run_stub(chip)
+
+    if flash_size is None:
+        flash_size = detect_flash_size(stub_chip)
+
+
     print(" - Flash Size: {}".format(flash_size))
 
     mock_args = configure_write_flash_args(info, firmware, flash_size,