mirror of
https://github.com/espressif/esp-idf.git
synced 2024-10-05 20:47:46 -04:00
83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
#!/usr/bin/env python
|
|
#
|
|
# Demonstrates the use of otatool.py, a tool for performing ota partition level
|
|
# operations.
|
|
#
|
|
# SPDX-FileCopyrightText: 2018-2022 Espressif Systems (Shanghai) CO LTD
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
import argparse
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
from subprocess import CalledProcessError
|
|
|
|
import serial
|
|
|
|
|
|
def get_running_partition(port=None):
|
|
# Monitor the serial output of target device. The firmware outputs the currently
|
|
# running partition
|
|
|
|
IDF_PATH = os.path.expandvars('$IDF_PATH')
|
|
sys.path.append(os.path.join(IDF_PATH, 'components', 'esptool_py', 'esptool'))
|
|
|
|
try:
|
|
# esptool>=4.0
|
|
from esptool.loader import ESPLoader
|
|
except (AttributeError, ModuleNotFoundError):
|
|
# esptool<4.0
|
|
from esptool import ESPLoader
|
|
|
|
ESPTOOL_PY = os.path.join(IDF_PATH, 'components', 'esptool_py', 'esptool', 'esptool.py')
|
|
|
|
baud = os.environ.get('ESPTOOL_BAUD', ESPLoader.ESP_ROM_BAUD)
|
|
|
|
if not port:
|
|
error_message = 'Unable to obtain default target device port.\nSerial log:\n\n'
|
|
try:
|
|
# Check what esptool.py finds on what port the device is connected to
|
|
output = subprocess.check_output([sys.executable, ESPTOOL_PY, 'chip_id']) # may raise CalledProcessError
|
|
pattern = r'Serial port ([\S]+)'
|
|
pattern = re.compile(pattern.encode())
|
|
|
|
port = re.search(pattern, output).group(1) # may raise AttributeError
|
|
except CalledProcessError as e:
|
|
raise Exception(error_message + e.output)
|
|
except AttributeError:
|
|
raise Exception(error_message + output)
|
|
|
|
serial_instance = serial.serial_for_url(port.decode('utf-8'), baud, do_not_open=True)
|
|
|
|
serial_instance.dtr = False
|
|
serial_instance.rts = False
|
|
|
|
serial_instance.rts = True
|
|
serial_instance.open()
|
|
serial_instance.rts = False
|
|
|
|
# Read until example end and find the currently running partition string
|
|
content = serial_instance.read_until(b'Example end')
|
|
pattern = re.compile(b'Running partition: ([a-z0-9_]+)')
|
|
running = re.search(pattern, content).group(1)
|
|
|
|
return running.decode('utf-8')
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--port', default=None)
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
res = get_running_partition(args.port)
|
|
except Exception as e:
|
|
print(e.message)
|
|
sys.exit(1)
|
|
|
|
print(res)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|