diff --git a/Rakefile b/Rakefile index b7e9ed54..53f75133 100644 --- a/Rakefile +++ b/Rakefile @@ -1,6 +1,8 @@ require "bundler/gem_tasks" require "rspec/core/rake_task" +require_relative 'lib/tasks/slurm' + RSpec::Core::RakeTask.new(:spec) task :default => :spec diff --git a/lib/tasks/slurm.rb b/lib/tasks/slurm.rb new file mode 100644 index 00000000..eb40a090 --- /dev/null +++ b/lib/tasks/slurm.rb @@ -0,0 +1,17 @@ +# frozen_string_literal: true + +require_relative '../ood_core' +require_relative '../ood_core/job/adapters/slurm' + +namespace :slurm do + + desc 'Get squeue output in the format this gem expects' + task :squeue do + fields = OodCore::Job::Adapters::Slurm::Batch.new.all_squeue_fields + args = OodCore::Job::Adapters::Slurm::Batch.new.squeue_args(options: fields.values) + + single_job = `squeue #{args.join(' ')}`.split("\n")[0...2] + + puts single_job + end +end