Add jobs stop command

Author: BartoszCki

paperspace/cli/__init__.py

@@ -0,0 +1,7 @@
+import click
+
+api_key_option = click.option(
+    "--apiKey",
+    "api_key",
+    help="API key to use this time only",
+)

paperspace/cli/cli.py

@@ -6,6 +6,7 @@
 import click
 
 from paperspace import constants, client, config
+from paperspace.cli import api_key_option
 from paperspace.cli.jobs.commands import jobs_group
 from paperspace.commands import experiments as experiments_commands, deployments as deployments_commands, \
     machines as machines_commands, login as login_commands
@@ -58,13 +59,6 @@ def del_if_value_is_none(dict_):
             del dict_[key]
 
 
-api_key_option = click.option(
-    "--apiKey",
-    "api_key",
-    help="API key to use this time only",
-)
-
-
 def validate_mutually_exclusive(options_1, options_2, error_message):
     used_option_in_options_1 = any(option is not None for option in options_1)
     used_option_in_options_2 = any(option is not None for option in options_2)

paperspace/cli/jobs/commands.py

@@ -1,6 +1,7 @@
 import click
 
 from paperspace import client, config
+from paperspace.cli import api_key_option
 from paperspace.commands import jobs as jobs_commands
 
 
@@ -15,7 +16,21 @@ def jobs_group():
     "job_id",
     required=True,
 )
+@api_key_option
 def delete_job(job_id, api_key=None):
     jobs_api = client.API(config.CONFIG_HOST, api_key=api_key)
     command = jobs_commands.DeleteJobCommand(api=jobs_api)
     command.execute(job_id)
+
+
+@jobs_group.command("stop", help="Stop running job")
+@click.option(
+    "--jobId",
+    "job_id",
+    required=True,
+)
+@api_key_option
+def stop_job(job_id, api_key=None):
+    jobs_api = client.API(config.CONFIG_HOST, api_key=api_key)
+    command = jobs_commands.StopJobCommand(api=jobs_api)
+    command.execute(job_id)

paperspace/commands/jobs.py

@@ -1,7 +1,7 @@
 from paperspace.commands import CommandBase
 
 
-class _JobsCommandBase(CommandBase):
+class JobsCommandBase(CommandBase):
     def _log_message(self, response, success_msg_template, error_msg):
         if response.ok:
             try:
@@ -19,10 +19,19 @@ def _log_message(self, response, success_msg_template, error_msg):
                 self.logger.log(error_msg)
 
 
-class DeleteJobCommand(_JobsCommandBase):
+class DeleteJobCommand(JobsCommandBase):
     def execute(self, job_id):
         url = "/jobs/{}/destroy/".format(job_id)
         response = self.api.post(url)
         self._log_message(response,
                           "Job deleted",
                           "Unknown error while deleting job")
+
+
+class StopJobCommand(JobsCommandBase):
+    def execute(self, job_id):
+        url = "/jobs/{}/stop/".format(job_id)
+        response = self.api.post(url)
+        self._log_message(response,
+                          "Job stopped",
+                          "Unknown error while stopping job")