takenoko 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1bb80ebb114ac3ee473ee833d63d9436dbf25ea9
-  data.tar.gz: 76dc1e87a834c27e407dff4180b5a0ea8f2590cf
+  metadata.gz: 9d15b5e30e05ec7639868f1061f6c04b340e837c
+  data.tar.gz: 7e31a25c1fbe2c5d850e59098bda414f5f473271
 SHA512:
-  metadata.gz: 7510fcbbbc44c414a0a5e0c3b25139be5ee11b340ee6b66e30a6592018fcf96696e3a7aac28b55931817c639adc48af709412dbf8473f3f6801d8f5b928f371f
-  data.tar.gz: 0fb1c6bfce088a274b00840b10d4fd6d14e0dc539f14a7435cdd994e2b2a36ad0e5d382dcc545613e413cd8b3b9b7cd1ad0ebecaf49bb8760d894f84cd60d854
+  metadata.gz: 4456cdcf48a2bf37b961b655ff9ce561c2c0fdf6cd2a3cfa708377a2d07b9086efb60c3ee303aa2feee1ece66b09d57e326240523d390d1a83925b197b2cca87
+  data.tar.gz: 7d9b34e206e3edb024c5e6c44174c9067656a3c397c9d8dcc6da8bd9ab4972420583d29193950507a1aba0700f7c0a969e930f2cc92db6262466aae7821fb6b2

data/lib/takenoko.rb

@@ -48,10 +48,10 @@ module Takenoko
   mattr_accessor :folder_id
   @@folder_id = nil
 
-
   require 'takenoko/exporter'
   require 'takenoko/google_client'
   require 'takenoko/attach_helper'
+  require 'takenoko/mapping_generator'
 
   def config
     yield self
@@ -62,14 +62,8 @@ module Takenoko
     return @@google_client
   end
 
-  def check_config
-    raise "Must specify mapping_file or sheet_id" unless (@@mapping_file || @@sheet_id)
-    raise "file not found:#{@@mapping_file}" if @@mapping_file && !::File.exist?(@@mapping_file)
-    return true
-  end
-
   def mapping_config
-    (@@always_reload && generate_base_config) || @@mapping_config || generate_base_config
+    MappingGenerator.generate
   end
 
   def table_config(table_name)
@@ -110,72 +104,6 @@ module Takenoko
     end
   end
 
-  def generate_base_config
-    return unless check_config
-
-    conf = HashWithIndifferentAccess.new
-
-    if @@sheet_id
-      sheet_conf = HashWithIndifferentAccess.new({tables: {}})
-      google_client.spreadsheet.worksheets.each do |ws|
-        next if ws.title.match(/\s*#.*/)
-        sheet_conf[:tables][ws.title] = {
-          worksheet_id: ws.gid,
-          worksheet: ws.title
-        }
-      end
-      conf.deep_merge!(sheet_conf)
-    end
-
-    if @@mapping_file
-      file_conf = YAML.load_file(@@mapping_file).with_indifferent_access
-      raise "tables not exists" if file_conf[:tables].blank?
-      conf.deep_merge!(file_conf)
-    end
-
-    conf[:tables].compact!
-    conf[:tables].each do |t, v|
-      unless conf[:tables][t]
-        conf[:tables][t] = Hash.new
-      end
-      table = conf[:tables][t]
-      raise "#{sheet_id} cannot be blank" unless table[:sheet_id] ||= @@sheet_id
-      table[:worksheet] = t if table[:worksheet].blank? && table[:worksheet_id].blank?
-      table[:find_column] = table[:find_column] || :id
-      table_name = table[:table_name] = t.pluralize if table[:table_name].blank?
-      table[:class_name] = table[:class_name] || (table_name && table_name.singularize.camelize) || table[:table_name].singularize.camelize
-      [
-        :allow_overwrite,
-        :truncate_all_data,
-        :file_extension,
-        :export_file_location,
-        :enable_postprocess,
-        :postprocess_class,
-        :folder_id
-      ].each do |f|
-        table[f] = class_variable_get("@@" + f.to_s) unless table.key?(f)
-      end
-
-      if table[:enable_postprocess] && table[:postprocess_class].blank?
-        table[:postprocess_class] = table[:class_name]
-      end
-
-      raise "Not support file extension: #{table[:file_extension]}" unless SUPPORTED_FILE_EXT.include?(table[:file_extension])
-
-      table[:download_location] = table[:download_location] || (@@download_location  && (@@download_location +"/"+table[:table_name]))
-      if attach_files = table[:attach_files].present?
-        attach_files = table[:attach_files]
-        attach_files = attach_files.map do |col|
-          raise "column_name must be set" unless col[:column_name]
-          col[:download_location] = col[:download_location].present? ? table[:download_location] + "/" + col[:download_location] : table[:download_location]
-          raise "folder_id should be set" unless col[:folder_id] = col[:folder_id] || table[:folder_id]
-        end
-      end
-    end
-    @@mapping_config = conf
-  end
-  alias reload_config! generate_base_config
-
   class Railtie < Rails::Railtie
     railtie_name :takenoko
 

data/lib/takenoko/mapping_generator.rb

@@ -0,0 +1,113 @@
+module Takenoko
+  module MappingGenerator
+    extend self
+    @@mapping_conf = nil
+    ALLOWED_OVERWRITE_FIELD = [
+      :allow_overwrite,
+      :truncate_all_data,
+      :file_extension,
+      :export_file_location,
+      :enable_postprocess,
+      :postprocess_class,
+      :folder_id
+    ]
+
+    FILTER = [
+      [:set_worksheet, ->(name, table) {
+        table[:worksheet] = name if table[:worksheet].blank? && table[:worksheet_id].blank?
+      }],
+
+      [:set_sheetid, ->(name, table){
+        raise "#{sheet_id} cannot be blank" unless table[:sheet_id] ||= Takenoko.sheet_id
+      }],
+
+      [:set_find_column, ->(name, table){
+        table[:find_column] ||= :id
+      }],
+
+      [:set_tablename, ->(name, table){
+        table[:table_name] = name.pluralize if table[:table_name].blank?
+      }],
+
+      [:set_class_name, ->(name, table){
+        table[:class_name] ||= table[:table_name] && table[:table_name].singularize.camelize
+      }],
+
+      [:overwrite_global_config, ->(name, table){
+        ALLOWED_OVERWRITE_FIELD.each do |f|
+          table[f] = Takenoko.public_send(f) unless table.key?(f)
+        end
+      }],
+
+      [:set_postprocess_class, ->(name, table){
+        table[:postprocess_class] ||= table[:class_name]
+      }],
+
+      [:check_file_extension, ->(name, table){
+        raise "Not support file extension: #{table[:file_extension]}" unless SUPPORTED_FILE_EXT.include?(table[:file_extension])
+      }],
+
+      [:set_download_location, ->(name, table){
+        table[:download_location] ||= Takenoko.download_location && (Takenoko.download_location + "/" + table[:table_name])
+      }],
+
+      [:set_attach_files, ->(name, table){
+        if (attach_files = table[:attach_files]).present?
+          attach_files = attach_files.map do |col|
+            raise "column_name must be set" unless col[:column_name]
+            raise "folder_id should be set" unless col[:folder_id] ||= table[:folder_id]
+            col[:download_location] ||= col[:download_location].present? ? table[:download_location] + "/" + col[:download_location] : table[:download_location]
+          end
+        end
+      }],
+    ]
+
+    def generate
+      return @@mapping_conf if Takenoko.allow_overwrite && @@mapping_conf
+      @@mapping_conf = base_config
+      @@mapping_conf[:tables].each do |name, table|
+        FILTER.each do |f|
+          f[1].call(name, table)
+        end
+      end
+      return @@mapping_conf
+    end
+
+    private
+
+    def check_config
+      raise "Must specify mapping_file or sheet_id" unless (Takenoko.mapping_file || Takenoko.sheet_id)
+      raise "file not found:#{Takenoko.mapping_file}" if Takenoko.mapping_file && !::File.exist?(Takenoko.mapping_file)
+      return true
+    end
+
+    def spread_sheet_config
+      sheet_conf = HashWithIndifferentAccess.new({tables: {}})
+      return sheet_conf unless Takenoko.sheet_id
+      Takenoko.google_client.spreadsheet.worksheets.each do |ws|
+        next if ws.title.match(/\s*#.*/)
+        sheet_conf[:tables][ws.title] = {
+          worksheet_id: ws.gid,
+          worksheet: ws.title
+        }
+      end
+      sheet_conf
+    end
+
+    def mapping_file_config
+      return {} unless Takenoko.mapping_file
+      file_conf = YAML.load_file(Takenoko.mapping_file).with_indifferent_access
+      raise "tables not exists" if file_conf[:tables].blank?
+      file_conf
+    end
+
+    def base_config
+      check_config
+      spread_sheet_config
+      conf = spread_sheet_config.deep_merge(mapping_file_config).with_indifferent_access
+      conf[:tables].compact!
+      conf
+    end
+
+  end
+end

data/lib/takenoko/version.rb

@@ -1,3 +1,3 @@
 module Takenoko
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: takenoko
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - KhiemNS
@@ -83,11 +83,12 @@ files:
 - lib/takenoko/attach_helper.rb
 - lib/takenoko/exporter.rb
 - lib/takenoko/google_client.rb
+- lib/takenoko/mapping_generator.rb
 - lib/takenoko/version.rb
 - lib/tasks/takenoko.rake
 - test/takenoko_test.rb
 - test/test_helper.rb
-homepage: https://github.com/khiemns54/takenoko/tree/release/0.2.0
+homepage: https://github.com/khiemns54/takenoko/tree/release/0.2.1
 licenses: []
 metadata: {}
 post_install_message: