lib/sushi_fabric/sushiApp.rb in sushi_fabric-0.3.3 vs lib/sushi_fabric/sushiApp.rb in sushi_fabric-0.3.5

@@ -1,8 +1,8 @@
 #!/usr/bin/env ruby
 # encoding: utf-8
-# Version = '20150326-142037'
+# Version = '20150326-150304'
 
 require 'csv'
 require 'fileutils'
 require 'yaml'
 require 'drb/drb'
@@ -199,9 +199,26 @@
       @dataset_hash = []
       @dataset = []
       dataset_tsv.each do |row|
         @dataset_hash << row.to_hash
         @dataset << row.to_hash
+      end
+
+      # save in sushi db
+      data_set_arr = []
+      headers = []
+      rows = []
+      data_set_arr = {'DataSetName'=>File.basename(@dataset_tsv_file).gsub(/.tsv/, ''), 'ProjectNumber'=>@project.gsub(/p/,'')}
+      csv = CSV.readlines(@dataset_tsv_file, :col_sep=>"\t")
+      csv.each do |row|
+        if headers.empty?
+          headers = row
+        else
+          rows << row
+        end
+      end
+      unless NO_ROR
+        @dataset_sushi_id = save_data_set(data_set_arr.to_a.flatten, headers, rows)
       end
     elsif @dataset_sushi_id
       @dataset_hash = []
       @dataset = []
       if dataset = DataSet.find_by_id(@dataset_sushi_id.to_i)