spreadsheet_to_json 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 18657291c808206aca46ce741a081cf1ecdc704c
-  data.tar.gz: d98ef0e39f714f8ba6bccf2507608c5ebe0f5372
+  metadata.gz: a42faa6b5d9367ed8f14491b01e595507ff58ed5
+  data.tar.gz: 8f70bf37a7a2dfd709183b3bff44852583fc1c27
 SHA512:
-  metadata.gz: 1e408913b1861712639a8649797661dda88cc403eb478d48cd6e2c8ffe8f12c0fe5a92c0868aaa71b3a581943a2a92f87b176b4f1ea59807ba189fb8542d593b
-  data.tar.gz: 430fa204dcceb6a6f741c56fbcfc4f91c6a462ed1b90ca36c8710515dda9c8bdfa817ad332b1d739533df5c53250d525e710fc145527f68bd16360c6e2fd585d
+  metadata.gz: 6868a8ef44a9bca74de6aa9525cc0dff40a91d3032284280d4fb3870b22c719aa32a720968b052538aa652262499ddedc8ace45d4a889cd2e24c6b13eab20e6f
+  data.tar.gz: 4c4e0ebb74f77d2b29924194b9741e39cbf4b68767a99f5c71d57758d818c6fb075bc57fb3f382cb2bdb150103d8d13fc63f305948d5721ca8cc9112baf04cf8

data/lib/json_to_spreadsheet.rb

@@ -0,0 +1,30 @@
+require 'google_drive'
+require 'yaml'
+require 'json'
+require './lib/spreadsheet'
+
+class JsonToSpreadsheet
+  def initialize(spreadsheet)
+    @spreadsheet = spreadsheet
+  end 
+
+  def convert_json_to_sheet(json_file_path, sheet_name)
+    arrays = json_to_hasharray(File.read(json_file_path))
+    puts (arrays.class)
+    if arrays.count <= 0
+      puts "The JSON file is empty"
+      exit
+    end
+    worksheet = @spreadsheet.get_sheet_by_name(sheet_name)
+    worksheet.list.keys = arrays[0].keys
+    arrays.each do |each|
+      worksheet.list.push(each)
+    end
+    worksheet.save
+  end
+
+  private
+  def json_to_hasharray(json_content)
+    JSON.parse(json_content)
+  end
+end

data/lib/spreadsheet.rb

@@ -0,0 +1,18 @@
+require 'google_drive'
+require './lib/util/auth'
+
+class Spreadsheet
+  def initialize(config_path)
+    session = GoogleDrive.login_with_oauth(Auth.get_auth_token(config_path))
+    @instance = session.spreadsheet_by_key(Auth.get_settings_from_yml(config_path)['spreadsheet_key'])
+  end
+
+  def get_sheet_by_id(worksheet_id=0)
+    @instance.worksheets[worksheet_id]
+  end
+
+  def get_sheet_by_name(worksheet_name)
+    @instance.worksheet_by_title(worksheet_name)
+  end
+end
+

data/lib/spreadsheet_to_json.rb

@@ -2,103 +2,72 @@
 require 'google_drive'
 require 'yaml'
 require 'json'
+require './lib/spreadsheet'
 
-module SpreadsheetToJson
-  class << self
-    def convert_rows_by_sheet_id(config_path, worksheet_id, rows, keys_row_num)
-      rows_to_json(
-        get_sheet_by_id(worksheet_id, config_path),
-        rows,
-        keys_row_num
-      )
-    end
-
-    def convert_rows_by_sheet_name(config_path, worksheet_name, rows, keys_row_num)
-      rows_to_json(
-        get_sheet_by_name(worksheet_name, config_path),
-        rows,
-        keys_row_num
-      )
-    end
-
-    def convert_a_worksheet_by_name(config_path, worksheet_name, keys_row_num) 
-      worksheet = get_sheet_by_name(worksheet_name, config_path)
-      rows_to_json(
-        worksheet,
-        [*keys_row_num+1..worksheet.num_rows],
-        keys_row_num
-      )
-    end
+class SpreadsheetToJson
+  def initialize(spreadsheet)
+    @spreadsheet = spreadsheet
+  end
 
-    def convert_a_worksheet_by_id(config_path, worksheet_id, keys_row_num) 
-      worksheet = get_sheet_by_id(worksheet_id, config_path)
-      rows_to_json(
-        worksheet,
-        [*keys_row_num+1..worksheet.num_rows],
-        keys_row_num
-      )
-    end
+  def convert_rows_by_sheet_id(worksheet_id, rows, keys_row_num)
+    rows_to_json(
+      @spreadsheet.get_sheet_by_id(worksheet_id),
+      rows,
+      keys_row_num
+    )
+  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
+  def convert_rows_by_sheet_name(worksheet_name, rows, keys_row_num)
+    rows_to_json(
+      @spreadsheet.get_sheet_by_name(worksheet_name),
+      rows,
+      keys_row_num
+    )
+  end
 
-    private
-    def rows_to_json(spreadsheet, rows, keys_row_num)
-      rows.collect {|row_num| construct_hash(spreadsheet, keys_row_num-2, row_num-2)}.to_json
-    end
+  def convert_a_worksheet_by_name(worksheet_name, keys_row_num) 
+    worksheet = @spreadsheet.get_sheet_by_name(worksheet_name)
+    rows_to_json(
+      worksheet,
+      [*keys_row_num+1..worksheet.num_rows],
+      keys_row_num
+    )
+  end
 
-    def get_sheet_by_id(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 convert_a_worksheet_by_id(worksheet_id, keys_row_num) 
+    worksheet = @spreadsheet.get_sheet_by_id(worksheet_id)
+    rows_to_json(
+      worksheet,
+      [*keys_row_num+1..worksheet.num_rows],
+      keys_row_num
+    )
+  end
 
-    def get_sheet_by_name(worksheet_name, config_path)
-      session = GoogleDrive.login_with_oauth(init_auth_token(config_path))
-      session.spreadsheet_by_key(get_settings_from_yml(config_path)['spreadsheet_key']).
-        worksheet_by_title(worksheet_name)
+  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
 
-    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'],
-        site: config['auth']['site'],
-        token_url: config['auth']['token_url'],
-        authorize_url: config['auth']['token_url']
-      )
-      auth_token = OAuth2::AccessToken.from_hash(
-        client,
-        {
-          :refresh_token => config['auth']['refresh_token'],
-          :expires_at => 3600
-        }
-      )
-      auth_token.refresh!.token
-    end
+  private
+  def rows_to_json(worksheet, rows, keys_row_num)
+    rows.collect {|row_num| construct_hash(worksheet, keys_row_num-2, row_num-2)}.to_json
+  end
 
-    def get_settings_from_yml(path)
-      File.open(path) { |file| YAML.load(file) }
+  def construct_hash(worksheet, keys_row_num, row=1)
+    if row < 1
+      puts "illeagal row num"
+      exit
     end
 
-    def construct_hash(spreadsheet, keys_row_num, row=1)
-      if row < 1
-        puts "illeagal row num"
-      end
-
-      begin
-        json_keys = spreadsheet.list[keys_row_num].to_hash.values
-      rescue NoMethodError
-        puts "worksheet_id is not exist"
-        exit
-      end
-      json_values = spreadsheet.list[row].to_hash.values
-      Hash[json_keys.zip(json_values)]
+    begin
+      json_keys = worksheet.list[keys_row_num].to_hash.values
+    rescue NoMethodError
+      puts "worksheet_id is not exist"
+      exit
     end
+    json_values = worksheet.list[row].to_hash.values
+    Hash[json_keys.zip(json_values)]
   end
 end

data/lib/util/auth.rb

@@ -0,0 +1,28 @@
+require 'google_drive'
+
+class Auth
+  class << self
+    def get_auth_token(config_path)
+      config = get_settings_from_yml(config_path)
+      client = OAuth2::Client.new(
+        config['auth']['client_id'],
+        config['auth']['client_secret'],
+        site: config['auth']['site'],
+        token_url: config['auth']['token_url'],
+        authorize_url: config['auth']['token_url']
+      )
+      auth_token = OAuth2::AccessToken.from_hash(
+        client,
+        {
+          :refresh_token => config['auth']['refresh_token'],
+          :expires_at => 3600
+        }
+      )
+      auth_token.refresh!.token
+    end
+
+    def get_settings_from_yml(path)
+      File.open(path) { |file| YAML.load(file) }
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: spreadsheet_to_json
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Xu Jianyong
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-19 00:00:00.000000000 Z
+date: 2015-06-20 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
@@ -18,7 +18,10 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/json_to_spreadsheet.rb
+- lib/spreadsheet.rb
 - lib/spreadsheet_to_json.rb
+- lib/util/auth.rb
 homepage: https://github.com/jackxu/google_spreadsheet_to_json
 licenses:
 - MIT