Slurmrest

Dockerfile

@@ -43,6 +43,11 @@ RUN printf "deb http://ftp.us.debian.org/debian buster main" > /etc/apt/sources.
     apt-get update && apt-get install libreadline7 && \
     rm -rf /var/lib/apt/lists/*
 
+# SSH support to run commands on slurmctld using ssh
+RUN mkdir -p ./ssh-keys
+RUN ssh-keygen -t ed25519 -f ./ssh-keys/id_ed25519 -N "" -y
+RUN echo "Generated SSH KEY"
+
 EXPOSE 80
 EXPOSE 25
 

coldfront/plugins/slurm/associations.py

@@ -28,7 +28,7 @@
     slurm_fixed_width_lines_to_dict,
     SlurmError,
 )
-
+from coldfront.plugins.slurm.integrations import SlurmError, get_cluster_connection
 
 logger = logging.getLogger(__name__)
 
@@ -226,7 +226,7 @@ def create_allocation_attributes(project, justification, quantity, resource):
             )
             return new_allocation
 
-        partitions = slurm_list_partitions()
+        partitions = get_cluster_connection(self.name).list_partitions()
         current_cluster_resource = Resource.objects.filter(
             resourceattribute__value=self.name, resource_type__name='Cluster').first()
         if not current_cluster_resource:
@@ -284,7 +284,7 @@ def map_shares(share_info):
             }
 
         if not file:
-            share_info = slurm_collect_shares(cluster=self.name)
+            share_info = get_cluster_connection(self.name).collect_shares()
         else:
             with open(file, 'r') as share_file:
                 share_data = list(share_file)