xpk 0.17.2__py3-none-any.whl → 1.0.0__py3-none-any.whl

xpk/parser/job.py

@@ -1,147 +0,0 @@
-"""
-Copyright 2024 Google LLC
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-     https://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-import argparse
-from ..commands.job import job_info, job_list, job_cancel
-
-from .common import add_shared_arguments
-from .validators import name_type
-
-
-def set_job_parser(job_parser: argparse.ArgumentParser):
-  job_subcommands = job_parser.add_subparsers(
-      title='job subcommands',
-      dest='xpk_job_subcommands',
-      help=(
-          'These are commands related to job management. Look at help for'
-          ' specific subcommands for more details.'
-      ),
-  )
-  set_job_info_parser(
-      job_info_parser=job_subcommands.add_parser(
-          'info', help='Show information about specified job.'
-      )
-  )
-  set_job_list_parser(
-      job_list_parser=job_subcommands.add_parser('ls', help='List jobs.')
-  )
-  set_job_cancel_parser(
-      job_cancel_parser=job_subcommands.add_parser(
-          'cancel', help='Cancel job execution.'
-      )
-  )
-
-
-def set_job_info_parser(job_info_parser: argparse.ArgumentParser):
-  job_info_required_arguments = job_info_parser.add_argument_group(
-      'Required arguments',
-      'The basic information required to identify the job.',
-  )
-  job_info_optional_arguments = job_info_parser.add_argument_group(
-      'Optional Arguments', 'Arguments optional for job info.'
-  )
-
-  ### Required arguments
-  job_info_required_arguments.add_argument(
-      '--cluster',
-      type=name_type,
-      default=None,
-      help='The name of the cluster to info jobs on.',
-      required=True,
-  )
-
-  job_info_optional_arguments.add_argument(
-      '--kind-cluster',
-      type=bool,
-      action=argparse.BooleanOptionalAction,
-      default=False,
-      help='Apply command to a local test cluster.',
-  )
-  job_info_required_arguments.add_argument(
-      'name',
-      type=str,
-      default=None,
-      help='Name of the job.',
-  )
-  job_info_parser.set_defaults(func=job_info)
-  add_shared_arguments(job_info_parser)
-
-
-def set_job_list_parser(job_list_parser: argparse.ArgumentParser):
-  job_list_required_arguments = job_list_parser.add_argument_group(
-      'Required Arguments',
-      'Arguments required for job list.',
-  )
-  job_list_optional_arguments = job_list_parser.add_argument_group(
-      'Optional Arguments', 'Arguments optional for job list.'
-  )
-
-  ### Required arguments
-  job_list_required_arguments.add_argument(
-      '--cluster',
-      type=name_type,
-      default=None,
-      help='The name of the cluster to list jobs on.',
-      required=True,
-  )
-
-  job_list_optional_arguments.add_argument(
-      '--kind-cluster',
-      type=bool,
-      action=argparse.BooleanOptionalAction,
-      default=False,
-      help='Apply command to a local test cluster.',
-  )
-
-  job_list_parser.set_defaults(func=job_list)
-  add_shared_arguments(job_list_optional_arguments)
-
-
-def set_job_cancel_parser(job_cancel_parser: argparse.ArgumentParser):
-  job_cancel_required_arguments = job_cancel_parser.add_argument_group(
-      'Required Arguments',
-      'Arguments required for job cancel.',
-  )
-  job_cancel_optional_arguments = job_cancel_parser.add_argument_group(
-      'Optional Arguments', 'Arguments optional for job cancel.'
-  )
-
-  job_cancel_required_arguments.add_argument(
-      'name',
-      type=str,
-      default=None,
-      help='The name of the job to be cancelled.',
-      nargs='+',
-  )
-
-  job_cancel_required_arguments.add_argument(
-      '--cluster',
-      type=name_type,
-      default=None,
-      help='The name of the cluster to delete the job on.',
-      required=True,
-  )
-
-  job_cancel_optional_arguments.add_argument(
-      '--kind-cluster',
-      type=bool,
-      action=argparse.BooleanOptionalAction,
-      default=False,
-      help='Apply command to a local test cluster.',
-  )
-
-  job_cancel_parser.set_defaults(func=job_cancel)
-  add_shared_arguments(job_cancel_optional_arguments)

xpk/parser/kind.py

@@ -1,95 +0,0 @@
-"""
-Copyright 2024 Google LLC
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-     https://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-from ..commands.kind import (
-    cluster_create,
-    cluster_delete,
-    cluster_list,
-)
-from .common import add_global_arguments
-from .validators import name_type
-
-
-def set_kind_parser(kind_parser):
-  cluster_subcommands = kind_parser.add_subparsers(
-      title='kind subcommands',
-      dest='xpk_kind_subcommands',
-      help=(
-          'These are commands related to kind management. Look at help for'
-          ' specific subcommands for more details.'
-      ),
-  )
-
-  ### "cluster create" command parser ###
-  cluster_create_parser = cluster_subcommands.add_parser(
-      'create', help='Create local clusters.'
-  )
-
-  ### Optional Arguments
-  cluster_create_parser.add_argument(
-      '--cluster',
-      type=name_type,
-      default='kind',
-      help=(
-          'The name of the cluster. Will be used as the prefix for internal'
-          ' objects in the cluster.'
-      ),
-      required=False,
-  )
-
-  cluster_create_parser.add_argument(
-      '--k8s-version',
-      type=str,
-      default='',
-      help='The Kubernetes version of the cluster.',
-      required=False,
-  )
-
-  add_global_arguments(cluster_create_parser)
-  cluster_create_parser.set_defaults(func=cluster_create)
-
-  ### "cluster delete" command parser ###
-  cluster_delete_parser = cluster_subcommands.add_parser(
-      'delete',
-      help='Delete cloud clusters.',
-  )
-
-  cluster_delete_required_arguments = cluster_delete_parser.add_argument_group(
-      'Required Arguments',
-      'Arguments required for cluster delete.',
-  )
-
-  ### Required arguments
-  cluster_delete_required_arguments.add_argument(
-      '--cluster',
-      type=name_type,
-      default=None,
-      help='The name of the cluster to be deleted.',
-      required=True,
-  )
-
-  ### Optional Arguments
-  add_global_arguments(cluster_delete_parser)
-  cluster_delete_parser.set_defaults(func=cluster_delete)
-
-  # "cluster list" command parser.
-  cluster_list_parser = cluster_subcommands.add_parser(
-      'list', help='List cloud clusters.'
-  )
-
-  ### Optional Arguments
-  add_global_arguments(cluster_list_parser)
-  cluster_list_parser.set_defaults(func=cluster_list)

xpk/parser/run.py

@@ -1,47 +0,0 @@
-"""
-Copyright 2025 Google LLC
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-     https://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-from ..commands.run import run
-from .common import (
-    add_shared_arguments,
-    add_slurm_arguments,
-    add_cluster_arguments,
-    add_kind_cluster_arguments,
-)
-
-
-def set_run_parser(run_parser):
-  run_required_arguments = run_parser.add_argument_group(
-      'Required Arguments', 'Arguments required for `run`.'
-  )
-  run_optional_arguments = run_parser.add_argument_group(
-      'Optional Arguments', 'Arguments optional for `run`.'
-  )
-
-  run_required_arguments.add_argument('script', help='script with task to run')
-  run_optional_arguments.add_argument(
-      '--timeout',
-      type=int,
-      default=None,
-      help='Amount of time to wait for job in seconds',
-      required=False,
-  )
-
-  add_cluster_arguments(run_optional_arguments)
-  add_kind_cluster_arguments(run_optional_arguments)
-  add_slurm_arguments(run_optional_arguments)
-  add_shared_arguments(run_parser)
-  run_parser.set_defaults(func=run)

xpk/parser/shell.py

@@ -1,59 +0,0 @@
-"""
-Copyright 2024 Google LLC
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-     https://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-from ..commands.shell import shell, shell_stop
-from .common import (
-    add_shared_arguments,
-    add_cluster_arguments,
-    add_kind_cluster_arguments,
-)
-import argparse
-
-
-def set_shell_parser(shell_parser: argparse.ArgumentParser) -> None:
-  shell_optional_arguments = shell_parser.add_argument_group(
-      'Optional Arguments', 'Arguments optional for shell.'
-  )
-  add_shared_arguments(shell_optional_arguments)
-  shell_parser.set_defaults(func=shell)
-
-  add_cluster_arguments(shell_optional_arguments)
-  add_kind_cluster_arguments(shell_optional_arguments)
-
-  shell_subcommands = shell_parser.add_subparsers(
-      title='shell subcommands',
-      dest='xpk_shell_subcommands',
-      help=(
-          'These are commands related to interactive shell. Look at help for'
-          ' specific subcommands for more details.'
-      ),
-  )
-
-  set_shell_stop_parser(
-      shell_stop_parser=shell_subcommands.add_parser(
-          name='stop', help='Stop the running shell.'
-      )
-  )
-
-
-def set_shell_stop_parser(shell_stop_parser: argparse.ArgumentParser):
-  shell_stop_optional_arguments = shell_stop_parser.add_argument_group(
-      'Optional Arguments', 'Arguments optional for shell stop.'
-  )
-  add_shared_arguments(shell_stop_optional_arguments)
-  shell_stop_parser.set_defaults(func=shell_stop)
-  add_cluster_arguments(shell_stop_parser)
-  add_kind_cluster_arguments(shell_stop_optional_arguments)