Feat 46 hpc endpoint (#54)

* feat: adds hpc job submit model

* feat: updates server

* feat: updates unit tests

* feat: adds endpoint to submit hpc configs

src/aind_data_transfer_service/configs/job_configs.py

@@ -314,6 +314,7 @@ def from_csv_row(
         )
 
 
+# Deprecating this class
 class HpcJobConfigs(BaseSettings):
     """Class to contain settings for hpc resources"""
 

src/aind_data_transfer_service/hpc/client.py

@@ -1,11 +1,14 @@
 """Module to manage connection with hpc cluster"""
 
-from typing import Optional, Union
+import json
+from typing import List, Optional, Union
 
 import requests
 from pydantic import BaseSettings, Field, SecretStr, validator
 from requests.models import Response
 
+from aind_data_transfer_service.hpc.models import HpcJobSubmitSettings
+
 
 class HpcClientConfigs(BaseSettings):
     """Configs needed to connect to the hpc cluster"""
@@ -100,3 +103,45 @@ def submit_job(self, job_def: dict) -> Response:
             url=self._job_submit_url, json=job_def, headers=self.__headers
         )
         return response
+
+    def submit_hpc_job(
+        self,
+        script: str,
+        job: Optional[HpcJobSubmitSettings] = None,
+        jobs: Optional[List[HpcJobSubmitSettings]] = None,
+    ) -> Response:
+        """
+        Submit a job following the v0.0.36 Slurm rest api job submission guide
+        Parameters
+        ----------
+        script : str
+          Executable script (full contents) to run in batch step
+        job : Optional[HpcJobSubmitSettings]
+          v0.0.36_job_properties (Default is None)
+        jobs : Optional[List[HpcJobSubmitSettings]]
+          List of properties of an HetJob (Default is None)
+
+        Returns
+        -------
+        Response
+
+        """
+        # Assert at least one of job or jobs is defined
+        assert job is not None or jobs is not None
+        # Assert not both job and jobs are defined
+        assert job is None or jobs is None
+        if job is not None:
+            job_def = {
+                "job": json.loads(job.json(exclude_none=True)),
+                "script": script,
+            }
+        else:
+            job_def = {
+                "jobs": [json.loads(j.json(exclude_none=True)) for j in jobs],
+                "script": script,
+            }
+
+        response = requests.post(
+            url=self._job_submit_url, json=job_def, headers=self.__headers
+        )
+        return response