Description redacted.

--
MOS_MIGRATED_REVID=107379349

1 files changed, 20 insertions, 8 deletions

diff --git a/src/main/java/com/google/devtools/build/lib/rules/python/PrecompilePythonMode.java b/src/main/java/com/google/devtools/build/lib/rules/python/PrecompilePythonMode.java
index bcb6f483f2..a65e9a6088 100644
--- a/src/main/java/com/google/devtools/build/lib/rules/python/PrecompilePythonMode.java
+++ b/src/main/java/com/google/devtools/build/lib/rules/python/PrecompilePythonMode.java
@@ -14,13 +14,25 @@
 
 package com.google.devtools.build.lib.rules.python;
 
-/**
- * Enumerates the different modes of Python precompilation.
- *
- * <p>NONE denotes that no precompilation should take place, and PROTO causes
- * only the generated _pb/_pb2.py files to be precompiled. ALL compiles all
- * Python files.
- */
+/** Enumerates the different modes of Python precompilation. */
 public enum PrecompilePythonMode {
-  NONE, PROTO, ALL;
+  /** No Python precompilation should take place. */
+  NONE,
+
+  /** Only the _pb/_pb2.py files generated from protocol buffers should be precompiled. */
+  PROTO,
+
+  /** Compiles all Python files, but removes the .py sources from the runfiles. */
+  ONLY,
+
+  /** Compiles all Python files, and leaves the .py sources in the runfiles. */
+  ALL;
+
+  public boolean shouldPrecompileProtos() {
+    return this != NONE;
+  };
+
+  public boolean shouldPrecompilePythonSources() {
+    return this == ONLY || this == ALL;
+  };
 }