gnoso-spreadsheet_report 0.0.4 → 0.0.5

data/lib/spreadsheet_report.rb

@@ -0,0 +1,52 @@
+require 'google_spreadsheet'
+
+class SpreadsheetReport
+  
+  def initialize(username, password, spreadsheet_key, &block)
+    @session = GoogleSpreadsheet.login(username, password)
+    @spreadsheet = @session.spreadsheet_by_key(spreadsheet_key)
+    
+    yield(self)
+  end
+  
+  def worksheet(name, query)
+    worksheet = @spreadsheet.worksheets.find { |ws| ws.title == name }
+    
+    # create the worksheet if we didn't find it
+    if worksheet != nil
+      worksheet.delete
+      worksheet = nil
+    end
+
+    # run the query
+    result = ActiveRecord::Base.connection.select_all(query)
+    
+    # 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)
+    
+    # add the title cells
+    keys.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|
+        worksheet[current_row, index + 1] = row_data[key]
+      end
+      current_row += 1
+      
+      if current_row % 250 == 0
+        worksheet.save
+      end
+    end
+    
+    worksheet.save
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: gnoso-spreadsheet_report
 version: !ruby/object:Gem::Version 
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors: 
 - Gnoso, Inc.
@@ -30,8 +30,8 @@ extensions: []
 
 extra_rdoc_files: []
 
-files: []
-
+files: 
+- lib/spreadsheet_report.rb
 has_rdoc: true
 homepage: http://www.gnoso.com
 licenses: