embulk-parser-roo-excel 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 64fc22e2d5dbfb23ecfe3ed1c79052eaf6e50c78
+  data.tar.gz: 1562aedacddcecbd1fcbd4a50309f23110f7f03d
+SHA512:
+  metadata.gz: 23f9f8c19103d5ca363eed6ec2d5bcf2a8c317c06b00778021820ea3b5df18f81a17023ec87021b56aff929147d03410156a770d5e1f9a9b09ac3be417b96d22
+  data.tar.gz: 3e540d730f174ba917d5df6f5168dab6de6030070a3a5f87f4d0230e6b0b4a810faf80f3570efd9048a0f36046a1c90ec8de6c352c7be4141ad91dca6b9c39a3

data/.gitignore

@@ -0,0 +1,5 @@
+*~
+/pkg/
+/tmp/
+/.bundle/
+/Gemfile.lock

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org/'
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,38 @@
+# Roo Excel parser plugin for Embulk
+
+Read Microsoft Excel(xlsx) files from input plugins.
+
+## Overview
+
+* **Plugin type**: parser
+* **Guess supported**: no
+
+## Configuration
+
+- **skip_header_lines**: Skip this number of lines first. Set 1 if the file has header line. (integer, default: 0)
+- **sheet**: the name of sheet (string, default: null (first sheet))
+
+## Example
+
+```yaml
+in:
+  type: any file input plugin type
+  parser:
+    type: roo-excel
+    skip_header_lines: 1 # first row is header.
+    sheet: "beatles"
+
+```
+
+(If guess supported) you don't have to write `parser:` section in the configuration file. After writing `in:` section, you can let embulk guess `parser:` section using this command:
+
+```
+$ embulk gem install embulk-parser-roo-excel
+$ embulk guess -g roo-excel config.yml -o guessed.yml
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

@@ -0,0 +1,3 @@
+require "bundler/gem_tasks"
+
+task default: :build

data/embulk-parser-roo-excel.gemspec

@@ -0,0 +1,18 @@
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-parser-roo-excel"
+  spec.version       = "0.0.1"
+  spec.authors       = ["Hiroyuki Sato"]
+  spec.summary       = "Roo Excel parser plugin for Embulk"
+  spec.description   = "Parses Excel files(xlsx) read by other file input plugins."
+  spec.email         = ["hiroysato@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/hiroyuki-sato/embulk-parser-roo-excel"
+
+  spec.files         = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
+  spec.test_files    = spec.files.grep(%r{^(test|spec)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency 'bundler', ['~> 1.0']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+  spec.add_dependency 'roo', ['~> 2.0.1']
+end

data/lib/embulk/guess/roo-excel.rb

@@ -0,0 +1,63 @@
+module Embulk
+  module Guess
+
+    # TODO implement guess plugin to make this command work:
+    #      $ embulk guess -g "roo-excel" partial-config.yml
+    #
+    #      Depending on the file format the plugin uses, you can use choose
+    #      one of binary guess (GuessPlugin), text guess (TextGuessPlugin),
+    #      or line guess (LineGuessPlugin).
+
+    #require "embulk/parser/roo-excel.rb"
+
+    #class RooExcelParserGuessPlugin < GuessPlugin
+    #  Plugin.register_guess("roo-excel", self)
+    #
+    #  def guess(config, sample_buffer)
+    #    if sample_buffer[0,2] == GZIP_HEADER
+    #      guessed = {}
+    #      guessed["type"] = "roo-excel"
+    #      guessed["property1"] = "guessed-value"
+    #      return {"parser" => guessed}
+    #    else
+    #      return {}
+    #    end
+    #  end
+    #end
+
+    #class RooExcelParserGuessPlugin < TextGuessPlugin
+    #  Plugin.register_guess("roo-excel", self)
+    #
+    #  def guess_text(config, sample_text)
+    #    js = JSON.parse(sample_text) rescue nil
+    #    if js && js["mykeyword"] == "keyword"
+    #      guessed = {}
+    #      guessed["type"] = "roo-excel"
+    #      guessed["property1"] = "guessed-value"
+    #      return {"parser" => guessed}
+    #    else
+    #      return {}
+    #    end
+    #  end
+    #end
+
+    #class RooExcelParserGuessPlugin < LineGuessPlugin
+    #  Plugin.register_guess("roo-excel", self)
+    #
+    #  def guess_lines(config, sample_lines)
+    #    all_line_matched = sample_lines.all? do |line|
+    #      line =~ /mypattern/
+    #    end
+    #    if all_line_matched
+    #      guessed = {}
+    #      guessed["type"] = "roo-excel"
+    #      guessed["property1"] = "guessed-value"
+    #      return {"parser" => guessed}
+    #    else
+    #      return {}
+    #    end
+    #  end
+    #end
+
+  end
+end

data/lib/embulk/parser/roo-excel.rb

@@ -0,0 +1,105 @@
+require 'roo'
+module Embulk
+  module Parser
+
+    class RooExcelParserPlugin < ParserPlugin
+      Plugin.register_parser("roo-excel", self)
+
+      def self.transaction(config, &control)
+        # configuration code:
+        task = {
+          "columns"   => config.param("columns", :array),
+          "sheet"     => config.param("sheet",   :string, default: nil),
+          "skip_header_lines" => config.param("skip_header_lines",:integer, default:0),
+        }
+
+        if( task['skip_header_lines'] < 0 )
+          raise ArgumentError, "skip_header_line does not allow negative number"
+        end
+
+        columns = []
+        task['columns'].each_with_index do |c,i|
+          columns << Column.new(i, c['name'], c['type'].to_sym)
+        end
+
+        yield(task, columns)
+      end
+
+      def init
+        # initialization code:
+        @sheet   = task["sheet"]
+        @columns = task["columns"]
+        @data_pos  = task["skip_header_lines"] + 1
+      end
+
+      def run(file_input)
+        while file = file_input.next_file
+
+          begin
+            xlsx = Roo::Excelx.new(StringIO.new(file.read))
+            if( @sheet )
+              xlsx.default_sheet = @sheet
+            else
+              xlsx.default_sheet = xlsx.sheets.first
+            end
+            last_row = xlsx.last_row
+            if ( last_row.nil? or  last_row - @data_pos <= 0 )
+              puts "No data. skip this file"
+              next
+            end
+
+            ncol = @columns.size
+            @data_pos.upto(last_row) do |row|
+              data = []
+              1.upto(ncol) do |col|
+                column = @columns[col-1]
+                data << convert_cell(column,xlsx,row,col)
+              end
+              @page_builder.add(data)
+            end
+          rescue ArgumentError
+            puts $!
+            puts $!.backtrace
+            puts "Can't open data file"
+          rescue
+            raise
+          end
+        end
+        page_builder.finish
+      end
+
+      # MEMO roo celltype
+      # returns the type of a cell: * :float * :string, * :date * :percentage * :formula * :time * :datetime.
+      #
+      def convert_cell(column,xlsx,nrow,ncol)
+        d = xlsx.cell(nrow,ncol)
+        type = column['type'] || 'string'
+        case type
+        when 'long'
+          d.to_i
+        when 'double'
+          d.to_f
+        when 'string'
+          d.to_s
+        when 'timestamp'
+          convert_time(d)
+        else # TODO
+          d.to_s
+        end
+      end
+
+      def convert_time(t)
+        if( t.kind_of?(Date) or t.kind_of?(DateTime) )
+          t.to_time
+        elsif( t.kind_of?(Time) )
+          t
+        elsif( t.kind_of?(String) )
+          Time.parse(t)
+        else
+          raise ArgumentError,"Can't convert time:#{t}"
+        end
+      end
+
+    end
+  end
+end

metadata

@@ -0,0 +1,95 @@
+--- !ruby/object:Gem::Specification
+name: embulk-parser-roo-excel
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Hiroyuki Sato
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-06-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: roo
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.1
+description: Parses Excel files(xlsx) read by other file input plugins.
+email:
+- hiroysato@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-parser-roo-excel.gemspec
+- lib/embulk/guess/roo-excel.rb
+- lib/embulk/parser/roo-excel.rb
+homepage: https://github.com/hiroyuki-sato/embulk-parser-roo-excel
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Roo Excel parser plugin for Embulk
+test_files: []
+has_rdoc: