mirror of
https://github.com/IoTManagerProject/IoTManager.git
synced 2026-03-26 14:12:16 +03:00
112 lines
5.9 KiB
Python
112 lines
5.9 KiB
Python
# Скрипт для простой прошивки ESP с учетом профиля из myProfile.json и поиска нужной кнопочки в интерфейсе PlatformIO
|
||
# Если ничего не указано в параметрах, то выполняется последовательно набор команд:
|
||
# 1. run PrepareProject.py
|
||
# 2. platformio run -t clean
|
||
# 3. platformio run -t uploadfs -е default_envs
|
||
# 4. platformio run -t upload -е default_envs
|
||
# 5. platformio run -t monitor
|
||
# где default_envs - это параметр default_envs из myProfile.json
|
||
#
|
||
# Если указан параметр -p или --profile <ИмяФайла>, то выполняется первая команда PrepareProject.py -p <ИмяФайла>
|
||
# Если указан параметр -b или --board <board_name>, то выполняется первая команда PrepareProject.py -b <board_name>
|
||
# Если указан парамтер -l или --lite, то пропускаются команды 1, 2 и 5 с предварительной компиляцией
|
||
# Если указан параметр -d или --debug, то выполняется только команда 4 с предварительной компиляцией
|
||
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
import json
|
||
|
||
def get_platformio_path():
|
||
"""
|
||
Возвращает путь к PlatformIO в зависимости от операционной системы.
|
||
"""
|
||
if os.name == 'nt': # Windows
|
||
return os.path.join(os.environ['USERPROFILE'], '.platformio', 'penv', 'Scripts', 'pio.exe')
|
||
else: # Linux/MacOS
|
||
return os.path.join(os.environ['HOME'], '.platformio', 'penv', 'bin', 'pio')
|
||
|
||
def load_default_envs(profile_path="myProfile.json"):
|
||
"""
|
||
Загружает значение default_envs из файла myProfile.json.
|
||
"""
|
||
if not os.path.isfile(profile_path):
|
||
print(f"Файл профиля {profile_path} не найден.")
|
||
sys.exit(1)
|
||
|
||
try:
|
||
with open(profile_path, 'r', encoding='utf-8') as file:
|
||
profile_data = json.load(file)
|
||
return profile_data["projectProp"]["platformio"]["default_envs"]
|
||
except KeyError:
|
||
print("Не удалось найти ключ 'default_envs' в myProfile.json.")
|
||
sys.exit(1)
|
||
except json.JSONDecodeError:
|
||
print(f"Ошибка при чтении файла {profile_path}: некорректный JSON.")
|
||
sys.exit(1)
|
||
|
||
def run_command(command):
|
||
"""
|
||
Выполняет указанную команду в subprocess.
|
||
"""
|
||
try:
|
||
print(f"Выполнение команды: {' '.join(command)}")
|
||
subprocess.run(command, check=True)
|
||
except subprocess.CalledProcessError as e:
|
||
print(f"Ошибка при выполнении команды: {e}")
|
||
sys.exit(e.returncode)
|
||
|
||
def run_platformio():
|
||
"""
|
||
Основная логика выполнения команд в зависимости от параметров.
|
||
"""
|
||
pio_path = get_platformio_path()
|
||
|
||
# Проверяем, существует ли PlatformIO
|
||
if not os.path.isfile(pio_path):
|
||
print(f"PlatformIO не найден по пути: {pio_path}")
|
||
sys.exit(1)
|
||
# print(f"PlatformIO найден по пути: {pio_path}")
|
||
|
||
# Читаем аргументы командной строки
|
||
args = sys.argv[1:]
|
||
lite_mode = '-l' in args or '--lite' in args
|
||
debug_mode = '-d' in args or '--debug' in args
|
||
profile_index = next((i for i, arg in enumerate(args) if arg in ('-p', '--profile')), None)
|
||
profile_file = args[profile_index + 1] if profile_index is not None and len(args) > profile_index + 1 else "myProfile.json"
|
||
|
||
# Загружаем default_envs из myProfile.json, если не указан параметр -b, который имеет больший приоритет
|
||
board_index = next((i for i, arg in enumerate(args) if arg in ('-b', '--board')), None)
|
||
default_envs = args[board_index + 1] if board_index is not None and len(args) > board_index + 1 else load_default_envs(profile_path=profile_file)
|
||
|
||
print(f"Используем default_envs: {default_envs}")
|
||
print(f"Режим Lite: {lite_mode}, Режим отладки: {debug_mode}")
|
||
print(f"Профиль: {profile_file}")
|
||
|
||
# Выполнение команд в зависимости от параметров
|
||
if not lite_mode and not debug_mode:
|
||
# Полный набор команд
|
||
run_command(['python', 'PrepareProject.py', '-p', profile_file])
|
||
|
||
# Добавляем сообщение о необходимости дождаться завершения обновления конфигурации
|
||
input(f"\x1b[1;31;42m Подождите завершения обновления конфигурации PlatformIO, затем нажмите Ввод для продолжения...\x1b[0m")
|
||
|
||
run_command([pio_path, 'run', '-t', 'clean'])
|
||
run_command([pio_path, 'run', '-t', 'uploadfs', '--environment', default_envs])
|
||
run_command([pio_path, 'run', '-t', 'upload', '--environment', default_envs])
|
||
run_command([pio_path, 'run', '-t', 'monitor'])
|
||
elif lite_mode:
|
||
# Упрощенный режим (пропускаем команды 1, 2 и 5)
|
||
run_command([pio_path, 'run', '-t', 'buildfs', '--environment', default_envs])
|
||
run_command([pio_path, 'run', '-t', 'uploadfs', '--environment', default_envs])
|
||
run_command([pio_path, 'run', '--environment', default_envs])
|
||
run_command([pio_path, 'run', '-t', 'upload', '--environment', default_envs])
|
||
elif debug_mode:
|
||
# Режим отладки (только команда 4)
|
||
run_command([pio_path, 'run', '--environment', default_envs])
|
||
run_command([pio_path, 'run', '-t', 'upload', '--environment', default_envs])
|
||
|
||
if __name__ == "__main__":
|
||
run_platformio()
|
||
|