From bd8009d3f0d2840ceaaa260ee1e4148ff1b1a944 Mon Sep 17 00:00:00 2001 From: bkleiner Date: Fri, 5 Aug 2022 00:06:14 +0200 Subject: [PATCH] optionally pass env to AddBuildMiddleware callback --- platformio/builder/tools/piobuild.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/platformio/builder/tools/piobuild.py b/platformio/builder/tools/piobuild.py index 4a32a1d750..06937e5864 100644 --- a/platformio/builder/tools/piobuild.py +++ b/platformio/builder/tools/piobuild.py @@ -292,7 +292,10 @@ def CollectBuildFiles( for callback, pattern in middlewares: if pattern and not fnmatch.fnmatch(node.srcnode().get_path(), pattern): continue - new_node = callback(new_node) + if callback.__code__.co_argcount == 2: + new_node = callback(env, new_node) + else: + new_node = callback(new_node) if new_node: new_sources.append(new_node)