lib/openstudio/analysis/server_api.rb in openstudio-analysis-0.2.3 vs lib/openstudio/analysis/server_api.rb in openstudio-analysis-0.3.0

@@ -107,16 +107,34 @@
         end
 
         analysis_ids
       end
 
+      def download_dataframe(analysis_id, save_directory=".")
+        response = @conn.get "/analyses/#{analysis_id}/download_data.rdata"
+        if response.status == 200
+          filename = response['content-disposition'].match(/filename=(\"?)(.+)\1/)[2]
+          puts "File #{filename} already exists, overwriting" if File.exist?("#{save_directory}/#{filename}")
+          File.open("#{save_directory}/#{filename}",'w') {|f| f << response.body}
+        end
+      end
+
+      def download_variables(analysis_id, save_directory=".")
+        response = @conn.get "/analyses/#{analysis_id}/download_variables.rdata"
+        if response.status == 200
+          filename = response['content-disposition'].match(/filename=(\"?)(.+)\1/)[2]
+          puts "File #{filename} already exists, overwriting" if File.exist?("#{save_directory}/#{filename}")
+          File.open("#{save_directory}/#{filename}",'w') {|f| f << response.body}
+        end
+      end
+
       def new_analysis(project_id, options)
         defaults = { analysis_name: nil, reset_uuids: false }
         options = defaults.merge(options)
 
         fail 'No project id passed' if project_id.nil?
         fail 'no formulation passed to new_analysis' unless options[:formulation_file]
-        fail "No formation exists #{options[:formulation_file]}" unless File.exist?(options[:formulation_file])
+        fail "No formulation exists #{options[:formulation_file]}" unless File.exist?(options[:formulation_file])
 
         formulation_json = JSON.parse(File.read(options[:formulation_file]), symbolize_names: true)
 
         # read in the analysis id from the analysis.json file
         analysis_id = nil