sushi_fabric 0.9.2 → 0.9.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 1d779ebb90dfe148f547bf8421ca4b065b8e88d4
-  data.tar.gz: 2d08ddb2575370f0218989f2ca2be53a5b5c50b4
+SHA256:
+  metadata.gz: 8ca5afb9de9e23912a2d514c547ff22d2edaf0a6c440e3663a0ac3fce59933f9
+  data.tar.gz: 7a906efa08a1bd3ee1b33dd02ff9a4350f241042626120278b720b3b3294d013
 SHA512:
-  metadata.gz: f96ea2b62e4d7a555e5c1494af34ee133396ee349c63ce71a6bc1209e0c83eefca0e9fbc95e926862d5f47832e39ef93b83cdeb17213213b85cb26619019d121
-  data.tar.gz: 82a7a369adca203302a4936c6e973bc73a233eaa9639c6350e262909314819fcd06f14797d2161c2196c90a7527e285bcab0e5a5a50db54abbeca2559cb1783e
+  metadata.gz: 482274611996c4e8c2d483088d1bcd8966d6136f849f7b77782d5ac0834556a56682fc2049f8c70da3aa48131bfa71ae13322a8e001252d4744d76787d9c2a23
+  data.tar.gz: 561434d611bc97fe9e30a035db0f3e3b71cbec89dc767f2531e2bf1b03905030364cf1bd833a0c432cad0d184a87a6552ae90430fe03c47f272611799d35cdea

data/lib/sushi_fabric/sushiApp.rb

@@ -1,6 +1,6 @@
 #!/usr/bin/env ruby
 # encoding: utf-8
-# Version = '20190731-142332'
+# Version = '20200703-164839'
 
 require 'csv'
 require 'fileutils'
@@ -227,6 +227,7 @@ class SushiApp
   attr_accessor :mango_run_name
   attr_accessor :input_dataset_bfabric_application_number
   attr_accessor :next_dataset_bfabric_application_number
+  attr_reader :inactivate_nodes
   def initialize
     @gstore_dir = GSTORE_DIR
     @project = nil
@@ -243,6 +244,7 @@ class SushiApp
     @module_source = MODULE_SOURCE
     @modules = []
     #@workflow_manager = workflow_manager_instance||DRbObject.new_with_uri(WORKFLOW_MANAGER)
+    @last_job = true
   end
   def set_input_dataset
     if @dataset_tsv_file
@@ -389,8 +391,13 @@ class SushiApp
   end
   def check_latest_module_version(mod)
     command_out =  %x[ bash -lc "source #{@module_source}; module whatis #{mod} 2>&1" ]
-    latest_mod = command_out.split.first
-    latest_mod = nil if latest_mod == "Failed"
+    latest_mod = nil
+    command_out.split("\n").each do |line|
+      if line =~ /#{mod}/
+        latest_mod = line.split.first
+        break
+      end
+    end
     latest_mod
   end
   def job_header
@@ -414,7 +421,6 @@ class SushiApp
                             modules_with_version = @modules.map{|mod| check_latest_module_version(mod)}
                             modules_with_version.compact!
                             "module add #{modules_with_version.join(' ')}"
-                            #"module add #{@modules.join(' ')}"
                           else
                             ""
                           end
@@ -429,6 +435,7 @@ umask 0002
 SCRATCH_DIR=#{@scratch_dir}
 GSTORE_DIR=#{@gstore_dir}
 INPUT_DATASET=#{@input_dataset_tsv_path}
+LAST_JOB=#{@last_job.to_s.upcase}
 echo "Job runs on `hostname`"
 echo "at $SCRATCH_DIR"
 mkdir $SCRATCH_DIR || exit 1
@@ -632,16 +639,16 @@ rm -rf #{@scratch_dir} || exit 1
   end
   def sample_mode
     selected_samples = Hash[*@params['samples'].split(',').map{|sample_name| [sample_name, true]}.flatten]
-    @dataset_hash.each do |row|
+    @dataset_hash.each_with_index do |row, i|
       @dataset = Hash[*row.map{|key,value| [key.gsub(/\[.+\]/,'').strip, value]}.flatten]
       if selected_samples[@dataset['Name']]
-        ## WRITE THE JOB SCRIPT
         sample_name = @dataset['Name']||@dataset.first
         @job_script = if @dataset_sushi_id and dataset = DataSet.find_by_id(@dataset_sushi_id.to_i)
                         File.join(@job_script_dir, @analysis_category + '_' + sample_name) + '_' + dataset.name.gsub(/\s+/,'_') + '.sh'
                       else
                         File.join(@job_script_dir, @analysis_category + '_' + sample_name) + '.sh'
                       end
+        @last_job = (i == @dataset_hash.length - 1)
         make_job_script
         @job_scripts << @job_script
         @result_dataset << next_dataset
@@ -752,28 +759,14 @@ rm -rf #{@scratch_dir} || exit 1
     if mock
       make_dummy_files
     end
-    copy_inputdataset_parameter_jobscripts
 
-    # job submittion
-    gstore_job_script_paths = []
-    @job_scripts.each_with_index do |job_script, i|
-      if job_id = submit(job_script, mock)
-        @job_ids << job_id
-        print "Submit job #{File.basename(job_script)} job_id=#{job_id}"
-        gstore_job_script_paths << File.join(@gstore_script_dir, File.basename(job_script))
-      end
-    end
-
-    puts
-    print 'job scripts: '
-    p @job_scripts
     print 'result dataset: '
     p @result_dataset
 
     # copy application data to gstore 
     @next_dataset_tsv_path = save_next_dataset_as_tsv
 
-    if !@job_ids.empty? and @dataset_sushi_id and dataset = DataSet.find_by_id(@dataset_sushi_id.to_i)
+    if @dataset_sushi_id and dataset = DataSet.find_by_id(@dataset_sushi_id.to_i)
       data_set_arr = []
       headers = []
       rows = []
@@ -798,20 +791,46 @@ rm -rf #{@scratch_dir} || exit 1
         unless @off_bfabric_registration
           if next_dataset = DataSet.find_by_id(@next_dataset_id)
             next_dataset.register_bfabric(bfabric_application_number: @next_dataset_bfabric_application_number)
+            if next_dataset.workunit_id
+              @job_scripts.each do |job_script|
+                open(job_script, "a") do |out|
+                  out.puts "module load Dev/Python"
+                  out.puts "WORKUNIT_ID=#{next_dataset.workunit_id}"
+                  out.puts "update_resource_size -w $WORKUNIT_ID"
+                end
+              end
+            end
           end
         end
+      end
+    end
+    copy_inputdataset_parameter_jobscripts
+
+    # job submittion
+    gstore_job_script_paths = []
+    @job_scripts.each_with_index do |job_script, i|
+      if job_id = submit(job_script, mock)
+        @job_ids << job_id
+        print "Submit job #{File.basename(job_script)} job_id=#{job_id}"
+        gstore_job_script_paths << File.join(@gstore_script_dir, File.basename(job_script))
+      end
+    end
+
+    puts
+    print 'job scripts: '
+    p @job_scripts
 
-        # save job and dataset relation in Sushi DB
-        job_ids.each_with_index do |job_id, i|
-          new_job = Job.new
-          new_job.submit_job_id = job_id.to_i
-          new_job.script_path = gstore_job_script_paths[i]
-          new_job.next_dataset_id = @next_dataset_id
-          new_job.save
-          new_job.data_set.jobs << new_job
-          new_job.data_set.save
-        end
 
+    unless @job_ids.empty? or NO_ROR
+      # save job and dataset relation in Sushi DB
+      job_ids.each_with_index do |job_id, i|
+        new_job = Job.new
+        new_job.submit_job_id = job_id.to_i
+        new_job.script_path = gstore_job_script_paths[i]
+        new_job.next_dataset_id = @next_dataset_id
+        new_job.save
+        new_job.data_set.jobs << new_job
+        new_job.data_set.save
       end
     end
     copy_nextdataset

data/lib/sushi_fabric/version.rb

@@ -1,3 +1,3 @@
 module SushiFabric
-  VERSION = "0.9.2"
+  VERSION = "0.9.7"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sushi_fabric
 version: !ruby/object:Gem::Version
-  version: 0.9.2
+  version: 0.9.7
 platform: ruby
 authors:
 - Functional Genomics Center Zurich
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-07-31 00:00:00.000000000 Z
+date: 2020-07-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -84,8 +84,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.6.14
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: workflow manager client.