batch.rb in activeforce-3.0.0

- old
+ new
@@ -5,77 +5,77 @@
 module Salesforce
   module Bulk
     class Batch
       Result = Struct.new :id, :success, :created, :error
       include ::Salesforce::Attributes
-      
+
       attr_accessor :id, :job, :number_records_failed, :number_records_processed, :state, :state_message, :total_time_processed, :filename, :csv
 
       include Blockenspiel::DSL
-      
+
       def initialize(job)
         self.job = job
         self.filename = temporary_csv_file
         self.csv = CSVLib.open(self.filename, 'w+')
         self.csv << csv_header
       end
-            
+
       def record(record)
         if record.is_a?(Hash)
           self.csv << ordered_values(record)
         else
           self.csv << ordered_values(record.attributes)
         end
       end
-      
+
       def ordered_values(record)
         job.csv_columns.map do |col|
           raw_value = record[col.name.to_sym]
           Column.to_csv_value Column.typecast(col.type, raw_value)
         end
       end
-        
+
       def create!
         self.csv.close
         response = ::Salesforce.connection.async_post("job/#{job.id}/batch", File.read(self.filename), :format => :xml, :content_type => 'text/csv')
         assign_attributes!(response)
       end
-      
+
       def update_status
         return state if completed?
         response = ::Salesforce.connection.async_get("job/#{job.id}/batch/#{id}", :format => :xml)
         self.state = response[:state]
       end
-      
+
       def results
         parse_csv_results ::Salesforce.connection.async_get("job/#{job.id}/batch/#{id}/result")
       end
-      
+
       [ :queued, :in_progress, :completed, :failed, :not_processed ].each do |status|
         define_method "#{status}?" do
           self.state == status.to_s.titleize
         end
       end
-      
+
       def temporary_csv_file
-        if Object.const_defined?(:Rails) && Rails.root.present?
+        if Object.const_defined?(:Rails) && Rails.respond_to?(:root) && Rails.root.present?
           Rails.root.join('tmp', 'files', "#{ Time.now.to_i}#{rand(10000)}.csv")
         else
           File.join("/tmp/#{ Time.now.to_i}#{rand(10000)}.csv")
         end
       end
-      
+
       private
-      
+
       def parse_csv_results(results)
          parsed_results = CSVLib.parse(results)
          parsed_results[1..-1].map { |row| Result.new(*row) }
       end
-      
+
       def csv_header
         self.job.csv_columns.map(&:original_name)
       end
-      
-      
+
+
     end
   end
 end