code4ward
diff --git a/‎.kokoro/trampoline_v2.sh
Copy file name to clipboardExpand all lines: .kokoro/trampoline_v2.sh
+17-12Lines changed: 17 additions & 12 deletions b/‎.kokoro/trampoline_v2.sh
Copy file name to clipboardExpand all lines: .kokoro/trampoline_v2.sh
+17-12Lines changed: 17 additions & 12 deletions
diff --git a/‎.trampolinerc
Copy file name to clipboardExpand all lines: .trampolinerc
+17Lines changed: 17 additions & 0 deletions b/‎.trampolinerc
Copy file name to clipboardExpand all lines: .trampolinerc
+17Lines changed: 17 additions & 0 deletions
@@ -36,10 +36,14 @@
 #
 # TRAMPOLINE_IMAGE: The docker image to use.
 # TRAMPOLINE_IMAGE_SOURCE: The location of the Dockerfile.
-# RUN_TESTS_SESSION: The nox session to run.
+# TRAMPOLINE_IMAGE_UPLOAD:
+#     (true|false): Whether to upload the Docker image after the
+#                   successful builds.
 # TRAMPOLINE_BUILD_FILE: The script to run in the docker container.
-# RUN_TESTS_DIR: A colon separated directory list relative to the
-#                project root.
+#
+# Potentially there are some repo specific envvars in .trampolinerc in
+# the project root.
+
 
 set -euo pipefail
 
@@ -55,10 +59,12 @@ else
   readonly IO_COLOR_RESET=""
 fi
 
-# Logs a message using the given color. The first argument must be one of the
-# IO_COLOR_* variables defined above, such as "${IO_COLOR_YELLOW}". The
-# remaining arguments will be logged in the given color. The log message will
-# also have an RFC-3339 timestamp prepended (in UTC).
+# Logs a message using the given color. The first argument must be one
+# of the IO_COLOR_* variables defined above, such as
+# "${IO_COLOR_YELLOW}". The remaining arguments will be logged in the
+# given color. The log message will also have an RFC-3339 timestamp
+# prepended (in UTC). You can disable the color output by setting
+# TERM=vt100.
 function log_impl() {
     local color="$1"
     shift
@@ -278,11 +284,10 @@ else
     log_red "Build finished with ${test_retval}"
 fi
 
-if [[ "${RUNNING_IN_CI}" == "true" ]] && \
-       [[ "${update_cache}" == "true" ]] && \
-       [[ -z "${KOKORO_GITHUB_PULL_REQUEST_NUMBER:-}" ]] && \
-       [[ $test_retval == 0 ]]; then
-    # Only upload it when the test passes.
+# Only upload it when the test passes.
+if [[ "${update_cache}" == "true" ]] && \
+       [[ $test_retval == 0 ]] && \
+       [[ "${TRAMPOLINE_IMAGE_UPLOAD:-false}" == "true" ]]; then
     log_yellow "Uploading the Docker image."
     if docker push "${TRAMPOLINE_IMAGE}"; then
 	log_green "Finished uploading the Docker image."
 
@@ -12,6 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Environment variables for python-docs-samples:
+# RUN_TESTS_SESSION: The nox session to run.
+# RUN_TESTS_DIR: A colon separated directory list relative to the
+# project root.
+
 required_envvars+=(
     # The nox session to run.
     "RUN_TESTS_SESSION"
@@ -25,6 +30,14 @@ pass_down_envvars+=(
     "RUN_TESTS_SESSION"
 )
 
+# Prevent unintentional override on the default image.
+
+if [[ "${TRAMPOLINE_IMAGE_UPLOAD:-false}" == "true" ]] && \
+   [[ -z "${TRAMPOLINE_IMAGE:-}" ]]; then
+   echo "Please set TRAMPOLINE_IMAGE if you want to upload the Docker image."
+   exit 1
+fi
+
 # Set default values. You can override them with env vars.
 if [[ -z "${TRAMPOLINE_IMAGE:-}" ]]; then
     # Set default image.
@@ -42,3 +55,7 @@ fi
 if [[ -z "${TRAMPOLINE_BUILD_FILE:-}" ]]; then
    TRAMPOLINE_BUILD_FILE=".kokoro/tests/run_tests_diff_head.sh"
 fi
+
+if [[ -z "${TRAMPOLINE_IMAGE_UPLOAD:-}" ]]; then
+    TRAMPOLINE_IMAGE_UPLOAD="false"
+fi