bj.py

from typing_extensions import override

import argparse
import sys
import openshift_client as oc

from .basecommand import Command


class ListJobsCommand(Command):
    """
    Display the status of your jobs. This includes all jobs that have not been deleted.

    Note:
    Jobs must be explicitly deleted after they have completed.
    'brun' deletes jobs by default. However, if you specified WAIT=0 to 'brun',
    then it will not delete the job.

    See also:
        'brun -h' and the repository README.md for more documentation and examples.
    """

    name: str = "bj"
    help: str = "Display the status of GPU jobs"

    @staticmethod
    @override
    def run(args: argparse.Namespace):
        """
        Display the status of GPU jobs using 'oc get jobs'.
        """
        try:
            jobs = oc.selector("jobs").objects()
            if not jobs:
                print("No jobs found.")
                return

            print(f"Found {len(jobs)} jobs:\n")
            for job in jobs:
                print(f"- {job.model.metadata.name}")

        except oc.OpenShiftPythonException as e:
            sys.exit(f"Error occurred while retrieving jobs: {e}")