Gecko [ v3.cesa.co.za ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
bigquery_command.py	11.74 KB	-rw-r--r--	1980-01-01 08:00
bq_cached_client.py	7.57 KB	-rw-r--r--	1980-01-01 08:00
command_cancel.py	2.56 KB	-rw-r--r--	1980-01-01 08:00
command_copy.py	10.35 KB	-rw-r--r--	1980-01-01 08:00
command_delete.py	14.51 KB	-rw-r--r--	1980-01-01 08:00
command_extract.py	5.67 KB	-rw-r--r--	1980-01-01 08:00
command_head.py	3.3 KB	-rw-r--r--	1980-01-01 08:00
command_info.py	750 B	-rw-r--r--	1980-01-01 08:00
command_init.py	6.51 KB	-rw-r--r--	1980-01-01 08:00
command_insert.py	5.69 KB	-rw-r--r--	1980-01-01 08:00
command_list.py	25.87 KB	-rw-r--r--	1980-01-01 08:00
command_load.py	25.65 KB	-rw-r--r--	1980-01-01 08:00
command_make.py	58.42 KB	-rw-r--r--	1980-01-01 08:00
command_mkdef.py	9.94 KB	-rw-r--r--	1980-01-01 08:00
command_partition.py	8.14 KB	-rw-r--r--	1980-01-01 08:00
command_query.py	30.25 KB	-rw-r--r--	1980-01-01 08:00
command_repl.py	7.5 KB	-rw-r--r--	1980-01-01 08:00
command_show.py	15.89 KB	-rw-r--r--	1980-01-01 08:00
command_snapshot.py	22 B	-rw-r--r--	1980-01-01 08:00
command_truncate.py	11.2 KB	-rw-r--r--	1980-01-01 08:00
command_undelete.py	1.97 KB	-rw-r--r--	1980-01-01 08:00
command_update.py	51.52 KB	-rw-r--r--	1980-01-01 08:00
command_version.py	620 B	-rw-r--r--	1980-01-01 08:00
command_wait.py	3.42 KB	-rw-r--r--	1980-01-01 08:00
commands_iam.py	22.24 KB	-rw-r--r--	1980-01-01 08:00
flags.py	6.06 KB	-rw-r--r--	1980-01-01 08:00
utils.py	63.78 KB	-rw-r--r--	1980-01-01 08:00
utils_data_transfer.py	3.08 KB	-rw-r--r--	1980-01-01 08:00
utils_flags.py	1.46 KB	-rw-r--r--	1980-01-01 08:00
utils_formatting.py	47 KB	-rw-r--r--	1980-01-01 08:00
utils_id.py	1.69 KB	-rw-r--r--	1980-01-01 08:00

Rename

#!/usr/bin/env python
"""The BigQuery CLI cancel command."""

from typing import Optional

from absl import flags

import bq_flags
from clients import client_job
from clients import utils as bq_client_utils
from frontend import bigquery_command
from frontend import bq_cached_client
from frontend import utils as frontend_utils
from utils import bq_id_utils

# These aren't relevant for user-facing docstrings:
# pylint: disable=g-doc-return-or-yield
# pylint: disable=g-doc-args

class Cancel(bigquery_command.BigqueryCmd):
  """Attempt to cancel the specified job if it is running."""

usage = """cancel [--nosync] [<job_id>]"""
  command = 'cancel'

def __init__(self, name: str, fv: flags.FlagValues):
    super(Cancel, self).__init__(name, fv)
    self._ProcessCommandRc(fv)

def RunWithArgs(self, job_id: str = '') -> Optional[int]:
    # pylint: disable=g-doc-exception
    """Request a cancel and waits for the job to be cancelled.

Requests a cancel and then either:
    a) waits until the job is done if the sync flag is set [default], or
    b) returns immediately if the sync flag is not set.
    Not all job types support a cancel, an error is returned if it cannot be
    cancelled. Even for jobs that support a cancel, success is not guaranteed,
    the job may have completed by the time the cancel request is noticed, or
    the job may be in a stage where it cannot be cancelled.

Examples:
      bq cancel job_id  # Requests a cancel and waits until the job is done.
      bq --nosync cancel job_id  # Requests a cancel and returns immediately.

Arguments:
      job_id: Job ID to cancel.
    """
    client = bq_cached_client.Client.Get()
    job_reference_dict = dict(
        bq_client_utils.GetJobReference(
            id_fallbacks=client,
            identifier=job_id,
            default_location=bq_flags.LOCATION.value,
        )
    )
    job = client_job.CancelJob(
        bqclient=client,
        job_id=job_reference_dict['jobId'],
        location=job_reference_dict['location'],
    )
    frontend_utils.PrintObjectInfo(
        job,
        bq_id_utils.ApiClientHelper.JobReference.Create(**job['jobReference']),
        custom_format='show',
    )
    status = job['status']
    if status['state'] == 'DONE':
      if (
          'errorResult' in status
          and 'reason' in status['errorResult']
          and status['errorResult']['reason'] == 'stopped'
      ):
        print('Job has been cancelled successfully.')
      else:
        print('Job completed before it could be cancelled.')
    else:
      print('Job cancel has been requested.')
    return 0