optionally pass env to AddBuildMiddleware callback (#4380)

This commit is contained in:
Benedikt Kleiner
2022-08-08 14:22:42 +02:00
committed by GitHub
parent 17a7293967
commit 32d501bed1

View File

@ -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)