Enable simple compilation of Python artifacts for py_binary and py_library.

--
MOS_MIGRATED_REVID=99163140

1 files changed, 6 insertions, 3 deletions

diff --git a/src/main/java/com/google/devtools/build/lib/rules/python/PyBinary.java b/src/main/java/com/google/devtools/build/lib/rules/python/PyBinary.java
index 337d63b1a5..b732cae2ed 100644
--- a/src/main/java/com/google/devtools/build/lib/rules/python/PyBinary.java
+++ b/src/main/java/com/google/devtools/build/lib/rules/python/PyBinary.java
@@ -27,6 +27,7 @@ import com.google.devtools.build.lib.rules.cpp.CcLinkParams;
 import com.google.devtools.build.lib.rules.cpp.CcLinkParamsProvider;
 import com.google.devtools.build.lib.rules.cpp.CcLinkParamsStore;
 
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -53,10 +54,13 @@ public abstract class PyBinary implements RuleConfiguredTargetFactory {
 
   static RuleConfiguredTargetBuilder init(
       RuleContext ruleContext, PythonSemantics semantics, PyCommon common) {
-    List<Artifact> srcs = common.validateSrcs();
     CcLinkParamsStore ccLinkParamsStore = initializeCcLinkParamStore(ruleContext);
 
-    common.initBinary(srcs);
+    List<Artifact> srcs = common.validateSrcs();
+    List<Artifact> allOutputs = new ArrayList<>(srcs);
+    allOutputs.addAll(semantics.precompiledPythonFiles(ruleContext, srcs, common));
+
+    common.initBinary(allOutputs);
     semantics.validate(ruleContext, common);
     if (ruleContext.hasErrors()) {
       return null;
@@ -127,4 +131,3 @@ public abstract class PyBinary implements RuleConfiguredTargetFactory {
     };
   }
 }
-