spreadsheet_report 0.0.5 → 0.0.6

data/lib/spreadsheet_report.rb

@@ -9,7 +9,11 @@ class SpreadsheetReport
     yield(self)
   end
   
-  def worksheet(name, query)
+  # Writes a report to a worksheet. By default, if cols is left nil, the report
+  # will base cols on the results of the query (but it's a hash, so no ordering
+  # is guaranteed). If cols is given, only those columns are reported, and the
+  # spreadsheet columns are populated in the order that they are given.
+  def worksheet(name, query, cols = nil)
     worksheet = @spreadsheet.worksheets.find { |ws| ws.title == name }
     
     # create the worksheet if we didn't find it
@@ -20,24 +24,23 @@ class SpreadsheetReport
 
     # run the query
     result = ActiveRecord::Base.connection.select_all(query)
+
+    # figure out the column names being used
+    if cols.nil? && result.size > 0
+      cols = result[0].keys      
+    end
     
     # resize the worksheet
-    rows = result.size + 1
-    keys = []
-    if result.size > 0
-      keys = result[0].keys
-    end
-    cols = keys.size
-    worksheet = @spreadsheet.add_worksheet(name, result.size + 1, keys.size)
+    worksheet = @spreadsheet.add_worksheet(name, result.size + 1, cols.size)
     
     # add the title cells
-    keys.each_with_index do |key, index|
+    cols.each_with_index do |key, index|
       worksheet[1, index + 1] = key
     end
     
     current_row = 2
     result.each do |row_data|
-      keys.each_with_index do |key, index|
+      cols.each_with_index do |key, index|
         worksheet[current_row, index + 1] = row_data[key]
       end
       current_row += 1
@@ -48,5 +51,6 @@ class SpreadsheetReport
     end
     
     worksheet.save
+    worksheet.synchronize
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: spreadsheet_report
 version: !ruby/object:Gem::Version 
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors: 
 - Gnoso, Inc.
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-09-25 00:00:00 -04:00
+date: 2009-09-28 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency