"""Provides a pre-kill method to run on macOS.""" from __future__ import print_function # system imports import subprocess import sys # third-party module imports import six def do_pre_kill(process_id, runner_context, output_stream, sample_time=3): """Samples the given process id, and puts the output to output_stream. @param process_id the local process to sample. @param runner_context a dictionary of details about the architectures and platform on which the given process is running. Expected keys are archs (array of architectures), platform_name, platform_url, and platform_working_dir. @param output_stream file-like object that should be used to write the results of sampling. @param sample_time specifies the time in seconds that should be captured. """ # Validate args. if runner_context is None: raise Exception("runner_context argument is required") if not isinstance(runner_context, dict): raise Exception("runner_context argument must be a dictionary") # We will try to run sample on the local host only if there is no URL # to a remote. if "platform_url" in runner_context and ( runner_context["platform_url"] is not None): import pprint sys.stderr.write( "warning: skipping timeout pre-kill sample invocation because we " "don't know how to run on a remote yet. runner_context={}\n" .format(pprint.pformat(runner_context))) output = subprocess.check_output(['sample', six.text_type(process_id), str(sample_time)]) output_stream.write(output)