from workloadmgrclient.v1 import WorkloadmgrCommand


class GlobalJobSchedulerCommand(WorkloadmgrCommand):
    resource = "global_job_scheduler"


class GetGlobalJobScheduler(GlobalJobSchedulerCommand):
    """Show status of global job scheduler."""

    def take_action(self, parsed_args):
        client = self.get_client()
        job_scheduler = client.get()
        if job_scheduler:
            print("Global job scheduler enabled")
        else:
            print("Global job scheduler disabled")
        return


class DisableGlobalJobScheduler(GlobalJobSchedulerCommand):
    """Disable global job scheduler."""

    def take_action(self, parsed_args):
        client = self.get_client()
        job_scheduler = client.disable()
        if job_scheduler:
            print("Global job scheduler is successfully enabled")
        else:
            print("Global job scheduler is disabled")
        return


class EnableGlobalJobScheduler(GlobalJobSchedulerCommand):
    """Enable global job scheduler."""

    def take_action(self, parsed_args):
        client = self.get_client()
        job_scheduler = client.enable()
        if job_scheduler:
            print("Global job scheduler is successfully enabled")
        else:
            print("Global job scheduler is not enabled")
        return