spreadsheet_to_json 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f6ea1f08ddb691d4358cc326c599e938c7156ee8
-  data.tar.gz: a73b2304166583529583afbeac2eb2187d1d9a40
+  metadata.gz: 20579b86b447e68124afe0d5c2b155565022174d
+  data.tar.gz: 0d8a1b0c58519ddf3c0ed77b97fe168932e8d622
 SHA512:
-  metadata.gz: 52b084a9cd9ef1d174c6c2d5dbc2fc7335fc91d017a38079912865621e02c69cd572d2ba362897687abf5c2e228ebcf0bd815edbb4ce62580b86871c6d393a9e
-  data.tar.gz: a908ba349004bc4413965c3219ae660a4f7445a7d3a2baefb520f4017d902408d714ecdf9c2723999e3a22e181d39a1fa7e8aaaf82ad8c955825db01abca0034
+  metadata.gz: 3bab23c043dfa357b2f6a58367cd44fc0745377e4f9af7774a32c8682aa2d2e89fcac49d0c87d7d753b56cd63f3d151cff0ff3d87e8584543a319b7996572da7
+  data.tar.gz: 3692f977b977f922b24aea7ffe5d5796d62841c8a22b14eb2dd2f6e2ad1079f3574c443af1934e18a0bcbb613fa4531c2ba11ca873b2d6372390f4d705fafd33

data/lib/spreadsheet_to_json.rb

@@ -5,19 +5,27 @@ require 'json'
 
 module SpreadsheetToJson
   class << self
-    def convert_to_json(worksheet_id=0, rows=[*2..get_spreadsheet.num_rows-2], keys_row_num)
-      spreadsheet = get_spreadsheet(worksheet_id)
+    def convert_to_json(config_path, worksheet_id=0, rows=[*2..get_spreadsheet.num_rows-2], keys_row_num)
+      spreadsheet = get_spreadsheet(worksheet_id, config_path)
       rows.collect {|row_num| construct_hash(spreadsheet, keys_row_num-2, row_num-2)}.to_json
     end
 
+    def output_json_file(content, output_path)
+      unless File.exist?(output_path)
+        File.new(output_path, 'w+')
+      end
+      File.open(output_path, 'w+') {|file| file.write(content) }
+    end
+
     private
-    def get_spreadsheet(worksheet_id=0)
-      session = GoogleDrive.login_with_oauth(init_auth_token)
-      session.spreadsheet_by_key(get_settings_from_yml['spreadsheet_key']).worksheets[worksheet_id]
+    def get_spreadsheet(worksheet_id=0, config_path)
+      session = GoogleDrive.login_with_oauth(init_auth_token(config_path))
+      session.spreadsheet_by_key(get_settings_from_yml(config_path)['spreadsheet_key']).
+        worksheets[worksheet_id]
     end
 
-    def init_auth_token
-      config = get_settings_from_yml
+    def init_auth_token(config_path)
+      config = get_settings_from_yml(config_path)
       client = OAuth2::Client.new(
         config['auth']['client_id'],
         config['auth']['client_secret'],
@@ -35,8 +43,8 @@ module SpreadsheetToJson
       auth_token.refresh!.token
     end
 
-    def get_settings_from_yml
-      File.open("./spreadsheet_access.yml") { |file| YAML.load(file) }
+    def get_settings_from_yml(path)
+      File.open(path) { |file| YAML.load(file) }
     end
 
     def construct_hash(spreadsheet, keys_row_num, row=1)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: spreadsheet_to_json
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Xu Jianyong
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-16 00:00:00.000000000 Z
+date: 2015-06-17 00:00:00.000000000 Z
 dependencies: []
 description: |-
   A gem to help you convert your google spreadsheet data to json format. With this gem, you can