chriz
0d8a8b35f0
Build and Release OpenMQTTGateway Firmware / Build Firmware & Publish Gitea Release (push) Canceled after 3m53s
117 lines
4.4 KiB
Python
117 lines
4.4 KiB
Python
# Configure rtl_433_ESP decoders (enable or disable specific protocols).
|
|
#
|
|
# Newer rtl_433_ESP releases mark some decoders with `.disabled = 1` (e.g.
|
|
# `esa_energy` since v0.6.x) and enable many noisy OOK decoders by default.
|
|
# This script patches the decoder definitions inside the downloaded library in
|
|
# .pio/libdeps/<env>/ before compilation.
|
|
#
|
|
# Usage (in the environment section):
|
|
# extra_scripts =
|
|
# ${env.extra_scripts}
|
|
# scripts/enable_rtl433_decoders.py
|
|
# custom_rtl_433_enable_decoders = esa_energy
|
|
# custom_rtl_433_disable_decoders =
|
|
# skylink_motion
|
|
# acurite_rain_896
|
|
#
|
|
# Must run as a normal (post) script: lib_deps are installed while the platform
|
|
# builder script runs, so a "pre:" script would not find the library on a clean
|
|
# build. Sources are only compiled after all scripts have been read, and SCons
|
|
# recompiles patched files because their content changed.
|
|
import glob
|
|
import os
|
|
import re
|
|
|
|
Import("env")
|
|
|
|
|
|
def patch_decoders(libdeps_dir, enable_decoders, disable_decoders):
|
|
"""Patch `.disabled` for the given r_device names in rtl_433_ESP."""
|
|
device_files = glob.glob(
|
|
os.path.join(libdeps_dir, "**", "src", "rtl_433", "devices", "*.c"),
|
|
recursive=True,
|
|
)
|
|
enable_status = {name: "not found" for name in enable_decoders}
|
|
disable_status = {name: "not found" for name in disable_decoders}
|
|
|
|
for path in device_files:
|
|
with open(path, "r", encoding="utf-8", newline="") as f:
|
|
content = f.read()
|
|
changed = False
|
|
|
|
for name in enable_decoders:
|
|
block_re = re.compile(
|
|
r"(r_device\s+(?:const\s+)?" + re.escape(name) + r"\s*=\s*\{)(.*?)(\};)",
|
|
re.DOTALL,
|
|
)
|
|
match = block_re.search(content)
|
|
if not match:
|
|
continue
|
|
body = match.group(2)
|
|
new_body, count = re.subn(
|
|
r"(\.disabled\s*=\s*)[1-9]\d*", r"\g<1>0", body
|
|
)
|
|
if count:
|
|
content = content[: match.start(2)] + new_body + content[match.end(2) :]
|
|
changed = True
|
|
enable_status[name] = "enabled"
|
|
else:
|
|
enable_status[name] = "already enabled"
|
|
|
|
for name in disable_decoders:
|
|
block_re = re.compile(
|
|
r"(r_device\s+(?:const\s+)?" + re.escape(name) + r"\s*=\s*\{)(.*?)(\};)",
|
|
re.DOTALL,
|
|
)
|
|
match = block_re.search(content)
|
|
if not match:
|
|
continue
|
|
body = match.group(2)
|
|
if re.search(r"\.disabled\s*=\s*[1-9]\d*", body):
|
|
disable_status[name] = "already disabled"
|
|
continue
|
|
if re.search(r"\.disabled\s*=\s*0\b", body):
|
|
new_body = re.sub(r"(\.disabled\s*=\s*)0\b", r"\g<1>1", body)
|
|
else:
|
|
nl = "\r\n" if "\r\n" in body else "\n"
|
|
new_body, count = re.subn(
|
|
r"([ \t]*\.fields\s*=)",
|
|
r" .disabled = 1," + nl + r"\g<1>",
|
|
body,
|
|
count=1,
|
|
)
|
|
if not count:
|
|
new_body = body + " .disabled = 1," + nl
|
|
content = content[: match.start(2)] + new_body + content[match.end(2) :]
|
|
changed = True
|
|
disable_status[name] = "disabled"
|
|
|
|
if changed:
|
|
with open(path, "w", encoding="utf-8", newline="") as f:
|
|
f.write(content)
|
|
|
|
return enable_status, disable_status
|
|
|
|
|
|
enable_list = env.GetProjectOption("custom_rtl_433_enable_decoders", "").split()
|
|
disable_list = env.GetProjectOption("custom_rtl_433_disable_decoders", "").split()
|
|
|
|
if enable_list or disable_list:
|
|
libdeps_dir = os.path.join(env.subst("$PROJECT_LIBDEPS_DIR"), env.subst("$PIOENV"))
|
|
enable_status, disable_status = patch_decoders(libdeps_dir, enable_list, disable_list)
|
|
for name, state in enable_status.items():
|
|
print("rtl_433_ESP decoder %s: %s" % (name, state))
|
|
for name, state in disable_status.items():
|
|
print("rtl_433_ESP decoder %s: %s" % (name, state))
|
|
missing = [
|
|
name
|
|
for name, state in list(enable_status.items()) + list(disable_status.items())
|
|
if state == "not found"
|
|
]
|
|
if missing:
|
|
print(
|
|
"Error: rtl_433_ESP decoder(s) not found in %s: %s"
|
|
% (libdeps_dir, ", ".join(missing))
|
|
)
|
|
env.Exit(1)
|