firesim/.github/scripts/change-workflow-instance-st...

23 lines
910 B
Python
Executable File

#!/usr/bin/env python3
# Changes the state of instances associated with the CI workflow run's unique tag.
# Can be used to start, stop, or terminate. This may run from either the manager
# or from the CI instance.
import argparse
from common import change_workflow_instance_states
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('tag_value',
help = "The tag used to identify workflow instances.")
parser.add_argument('state_change',
choices = ['terminate', 'stop', 'start'],
help = "The state transition to initiate on workflow instances.")
parser.add_argument('github_api_token',
help = "API token to modify self-hosted runner state.")
args = parser.parse_args()
change_workflow_instance_states(args.github_api_token, args.tag_value, args.state_change)