seamusabshere-gdocs_bootstrap 0.0.2 → 0.1.0

data/README.rdoc

@@ -1,14 +1,23 @@
 = gdocs_bootstrap
 
   class FuelType
-    gdoc_bootstrap :url => 'http://spreadsheets.google.com/pub?key=p70r3FHguhimIdBKyVz3iPA&output=csv&gid=0'
+    gdocs_bootstrap :url => 'http://spreadsheets.google.com/pub?key=p70r3FHguhimIdBKyVz3iPA'
   end
 
-This lets you bootstrap your ActiveRecord models with Google Docs spreadsheets.
+This will bootstrap your ActiveRecord model with sheet 1 of the Google Docs spreadsheet.
 
   >> FuelType.bootstrap!
   => true
 
+You can specify other sheets (starts at 1):
+
+class FuelType
+  gdocs_bootstrap :url => 'http://spreadsheets.google.com/pub?key=p70r3FHguhimIdBKyVz3iPA',
+                 :sheet => 1
+end
+
+The gem will always set request the CSV version of the document in question.
+
 == Spreadsheet structure
 
 The first column is always the key.
@@ -31,6 +40,8 @@ would generate
   a = FuelType.find_or_create_by_name('fuel oil')
   a.update_attributes(:name => 'coal', :emission_factor => '22.51', :units => 'lbs/gallon')
 
+
+
 == Copyright
 
 Copyright (c) 2009 Seamus Abshere. See LICENSE for details.

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
 :major: 0
-:minor: 0
-:patch: 2
+:minor: 1
+:patch: 0

data/lib/gdocs_bootstrap.rb

@@ -4,7 +4,7 @@ require 'open-uri'
 require 'active_support'
 
 class GdocsBootstrap
-  attr_accessor :url, :target_class_name, :key_sprintf # set in options
+  attr_accessor :url, :target_class_name, :key_sprintf, :sheet # set in options
   attr_accessor :key_name, :headers, :dictionary, :width
   
   def bootstrap!
@@ -19,7 +19,8 @@ class GdocsBootstrap
   private
     
   def initialize(options = {})
-    @url = options[:url]
+    @sheet = interpret_sheet(options[:sheet])
+    @url = interpret_url(options[:url])
     @target_class_name = options[:target_class_name]
     @key_sprintf = options[:key_sprintf] || '%s'
   end
@@ -42,6 +43,16 @@ class GdocsBootstrap
     end
   end
   
+  def interpret_sheet(sheet)
+    sheet.to_i >= 1 ? sheet.to_i - 1 : 0
+  end
+  
+  def interpret_url(url)
+    url = url.gsub(/\&(gid|output)=.*(\&|$)/, '')
+    url << "&output=csv&gid=#{sheet}"
+    url
+  end
+  
   def interpret_key(k)
     k = k.to_s.strip
     k = k.to_i if /\%[0-9\.]*d/.match(key_sprintf)

data/test/gdocs_bootstrap_test.rb

@@ -12,7 +12,7 @@ class GdocsBootstrapTest < Test::Unit::TestCase
   end
   
   def test_should_have_url
-    assert_equal @bootstrap.url, GOOGLE_DOC_URL
+    assert_equal @bootstrap.url, GOOGLE_DOC_URL + '&output=csv&gid=0'
   end
   
   def test_should_have_target_class_name

data/test/test_helper.rb

@@ -62,4 +62,4 @@ GdocsBootstrapTestHelper::Initializer.start
 
 class Scrum < ActiveRecord::Base; end
 
-GOOGLE_DOC_URL = 'http://spreadsheets.google.com/pub?key=p70r3FHguhimIdBKyVz3iPA&output=csv&gid=0'
+GOOGLE_DOC_URL = 'http://spreadsheets.google.com/pub?key=p70r3FHguhimIdBKyVz3iPA'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: seamusabshere-gdocs_bootstrap
 version: !ruby/object:Gem::Version 
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors: 
 - Seamus Abshere