export_to 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4359e4a3d0f73c554990815260ffb54104fea537
-  data.tar.gz: a4c3f34ea3cc77b40f8fc68ff87b9355edc8057c
+  metadata.gz: 6457e74385201c5b900c9ccd219348ae675853f6
+  data.tar.gz: b9595214bc9ad0228fbc90830130b80c050e2496
 SHA512:
-  metadata.gz: 2a765688b85e6ea915968606f06ecb14d110a6307e5fdef7347d63b17fca5f6bf670d04fa6924a1ad1dad321eac789dc55f0a8f297e736e7d51adeb7f49d9d83
-  data.tar.gz: 8264fd6d0902cd63e6a76232e5262e8233e1c3022bde5ef357090d870c1d3aa5ad46a0008caf2bfe53aeaa0450ebb16575c40b907fdcae7f70efa25f2e2cc436
+  metadata.gz: 832043736ed5cb9f89b45fac1e2b2a768ca80cabc716c5b1e9b8c6797b7e2066fc5697253eb71d4d5b91b957f091bbe55c4c2ef7fcfd68bb932677522d5ac531
+  data.tar.gz: 4d74f611a3f2b8962d8ae9f4c00ebc360b7eb99b1a07dc1c2e4678c32b140e47b347b3dbfb258f7e1d35cf776442e4c5ca86cac72f9d6688a33a79f634c3cb02

data/Gemfile.lock

@@ -3,10 +3,9 @@ PATH
   specs:
     export_to (0.1.0)
       activesupport
+      fast_excel
       iconv
-      roo
       spreadsheet
-      write_xlsx
 
 GEM
   remote: https://rubygems.org/
@@ -17,30 +16,21 @@ GEM
       minitest (~> 5.1)
       tzinfo (~> 1.1)
     concurrent-ruby (1.0.5)
+    fast_excel (0.2.5)
+      ffi (> 1.9, < 2)
+    ffi (1.9.25)
     i18n (1.1.1)
       concurrent-ruby (~> 1.0)
     iconv (1.0.5)
-    mini_portile2 (2.3.0)
     minitest (5.11.3)
-    nokogiri (1.8.5)
-      mini_portile2 (~> 2.3.0)
     rake (10.5.0)
-    roo (2.7.1)
-      nokogiri (~> 1)
-      rubyzip (~> 1.1, < 2.0.0)
     ruby-ole (1.2.12.1)
-    rubyzip (1.2.2)
     shoulda-context (1.2.2)
     spreadsheet (1.1.8)
       ruby-ole (>= 1.0)
     thread_safe (0.3.6)
     tzinfo (1.2.5)
       thread_safe (~> 0.1)
-    write_xlsx (0.85.5)
-      rubyzip (>= 1.0.0)
-      zip-zip
-    zip-zip (0.3)
-      rubyzip (>= 1.0.0)
 
 PLATFORMS
   ruby

data/export_to.gemspec

@@ -28,9 +28,8 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "minitest", "~> 5.0"
   spec.add_development_dependency "shoulda-context"
 
-  spec.add_dependency "write_xlsx"
+  spec.add_dependency "fast_excel"
   spec.add_dependency "spreadsheet"
-  spec.add_dependency "roo"
   spec.add_dependency "iconv"
   spec.add_dependency "activesupport"
 end

data/lib/export_to/base.rb

@@ -2,39 +2,52 @@ module ExportTo
   class Base < Struct.new(:records)
     class_attribute :head_titles
     class_attribute :body_keys
+    class_attribute :body_column_proc
     class_attribute :presenter_klass
     class_attribute :join_relation
+    class_attribute :each_proc, :each_method
+    class_attribute :xlsx_file_path, :xlsx_file_name
 
     # 預設 presenter
     self.presenter_klass = Presenter
+    self.each_method = :each
+    self.xlsx_file_path = '/tmp'
+    self.xlsx_file_name = '/file'
 
     attr_accessor :object
 
     def to_csv
       CSV.generate do |csv|
-        each do |columns, order, i|
+        rows! do |columns, model, x|
           csv << columns
         end
       end
     end
 
     # 新版 Excel
-    def to_xlsx
-      io = StringIO.new
-      book = WriteXLSX.new(io)
-      sheet = book.add_worksheet
+    def to_xlsx(file_path=nil, file_name=nil)
+      file_path ||= self.class.xlsx_file_path
+      file_name ||= self.class.xlsx_file_name
+      path = to_xlsx_file(file_path, file_name)
+      # TODO: 讀取檔案回傳
+      File.open(path, 'rb') { |f| f.read }
+    end
 
-      # 表格頭 (粗體紅字)
-      format_head = book.add_format.tap { |f| f.set_bold }.tap { |f| f.set_color('red') }
+    # 新版 Excel (outpuut: path)
+    def to_xlsx_file(file_path="tmp", file_name="export")
+      path = "#{file_path}/#{file_name}_#{Time.now.to_i}_#{SecureRandom.hex}.xlsx"
+      workbook = FastExcel.open(path, constant_memory: true)
+      worksheet = workbook.add_worksheet("Default")
 
-      each do |columns, order, x|
-        columns.each_with_index do |column, y|
-          sheet.write(x, y, column, format_head)
-        end
+      bold = workbook.bold_cell_format
+      worksheet.set_row(0, FastExcel::DEF_COL_WIDTH, bold)
+
+      rows! do |columns, model, x|
+        worksheet.write_row(x, columns)
       end
 
-      book.close
-      io.string
+      workbook.close
+      path
     end
 
     # 舊版 Excel
@@ -42,9 +55,10 @@ module ExportTo
       book = Spreadsheet::Workbook.new
       sheet = book.create_worksheet
 
-      each do |columns, order, x|
+      rows! do |columns, model, x|
         sheet.row(x).concat(columns)
       end
+
       spreadsheet = StringIO.new
       book.write(spreadsheet)
       spreadsheet.string
@@ -52,39 +66,77 @@ module ExportTo
 
     protected
 
-    def each
+    def rows
+      data = []
+      rows! do |columns, model, x|
+        data.push(columns)
+      end
+      data
+    end
+
+    def rows!
       i = 0
+
       yield(self.class.head_titles, nil, i)
 
       join_relation = self.class.join_relation
-      records.each do |record|
-        run_records = if join_relation && record.send(join_relation).nil? == false
+
+      each_models do |record, x|
+        run_records = if join_relation.present? && record.send(join_relation).present?
           record.send(join_relation)
         else
           [ record ]
         end
 
         # 指定目前 order 讓 subclass 可以 override
-        run_records.each do |run_record|
+        run_records.each_with_index do |run_record, y|
           i = i + 1
-          self.object = if run_record != record
-            self.class.presenter_klass.new(record, run_record)
+          object = if run_record != record
+            self.class.presenter_klass.new(record, run_record, x, y)
           else
-            self.class.presenter_klass.new(run_record)
+            self.class.presenter_klass.new(run_record, nil, x, y)
           end
-          yield(columns!, run_record, i)
+
+          each_proc.call(object) if self.class.each_proc.present?
+
+          columns = fetch_columns!(object)
+
+          yield(columns, run_record, i)
+        end
+      end
+    end
+
+    def each_models(&block)
+      case self.class.each_method
+      when :each
+        records.each.with_index(&block)
+      when :find_in_batches
+        find_in_batches(&block)
+      end
+    end
+
+    def find_in_batches
+      i = 0
+      records.find_in_batches do |group|
+        group.each do |model|
+          yield(model, i)
+          i = i + 1
         end
       end
     end
 
     class << self
+
       protected
 
-      def set(title, key)
+      def set(title, key, &block)
         self.head_titles ||= []
         self.body_keys ||= []
+        self.body_column_proc ||= []
+
         self.head_titles.push(title)
         self.body_keys.push(key)
+        self.body_column_proc.push(block)
       end
 
       def presenter(&block)
@@ -103,14 +155,25 @@ module ExportTo
       def joins(relation)
         self.join_relation = relation
       end
+
+      def each_with(&block)
+        self.each_proc = block
+      end
     end
 
     private
 
-    def columns!
-      self.class.body_keys.map do |key|
-        data = object.send(key)
-        data = data.gsub("\n", " ").gsub("\r", " ") if data.is_a?(String) 
+    def fetch_columns!(object)
+      self.class.body_keys.map.with_index do |key, i|
+        data = case key
+        when String
+          key
+        when Symbol
+          (object.send(key) || "")
+        end
+
+        column_proc = self.class.body_column_proc[i]
+        data = self.class.body_column_proc[i].call(data) if column_proc.present?
         data
       end
     end

data/lib/export_to/presenter.rb

@@ -1,15 +1,15 @@
 module ExportTo
   class Presenter
-    attr_accessor :model, :relation
+    attr_accessor :model, :relation, :x, :y
     NoAttributeError = Class.new(NoMethodError)
 
-    def initialize(model, relation=nil)
-      self.model = model
-      self.relation = relation
+    def initialize(model, relation=nil, x, y)
+      self.model, self.relation = model, relation
+      self.x, self.y = x, y
     end
 
     def relation?
-      relation != nil
+      relation.present?
     end
 
     def method_missing(m, *args, &block)

data/lib/export_to/version.rb

@@ -1,3 +1,3 @@
 module ExportTo
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/export_to.rb

@@ -1,6 +1,8 @@
-require 'active_support/core_ext/class/attribute'
-require 'write_xlsx'
-require 'roo'
+require 'active_support/core_ext/object'
+require 'active_support/core_ext/class'
+require 'active_support'
+require 'fast_excel'
+require 'spreadsheet'
 require 'csv'
 require 'iconv'
 require 'export_to/version'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: export_to
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - eddie
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-10-21 00:00:00.000000000 Z
+date: 2018-12-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -67,7 +67,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: write_xlsx
+  name: fast_excel
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -94,20 +94,6 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-- !ruby/object:Gem::Dependency
-  name: roo
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
 - !ruby/object:Gem::Dependency
   name: iconv
   requirement: !ruby/object:Gem::Requirement