embulk-filter-google_natural_language_api 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d4ad2a7fd3873b1b7e810bd19799f3cc4953648c
+  data.tar.gz: ff1f308f8ff6267cb232c8290ce0adfeaaaf8c4d
+SHA512:
+  metadata.gz: 80ad5396e358eff5d6a18519fd1b410a97ee9651ef8fbc2c23ff4a829a64d1d10812c5a2672f5bd6435192acc2b13391facd3385c251f583d0dcc133a31e3efe
+  data.tar.gz: 840d7581b04d4208f9545070cbdb5be93c330c6ee0e788348ef1b648f2fa2cb93b8de394f0467a724d8ae8b83b1c03de92754d1275afa96e489e392682897c24

data/.gitignore

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

data/.ruby-version

@@ -0,0 +1 @@
+jruby-9.1.5.0

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,37 @@
+# Google Natural Language Api filter plugin for Embulk
+
+Google Natural Language Api filter plugin for Embulk.
+
+## Overview
+
+* **Plugin type**: filter
+
+## Configuration
+
+- **api**: api type. analyzeSentiment or analyzeEntities or analyzeSyntax. (string, required)
+- **out_key_name_suffix**: out_key_name_suffix (string, required)
+- **key_names**: key_names (array, required)
+- **delay**: delay (integer, default: 0)
+- **google_api_key**: google_api_key (string, default: ENV['GOOGLE_API_KEY'])
+- **language**: language. default is auto detect. (string, default: nil)
+
+
+## Example
+
+```yaml
+filters:
+  - type: google_natural_language_api
+    api: analyzeSentiment
+    out_key_name_suffix: _parsed
+    language: en
+    key_names:
+      - title
+      - message
+```
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/bin/.gitignore

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

data/bin/.ruby-version

@@ -0,0 +1 @@
+jruby-9.1.5.0

data/bin/Gemfile

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

data/bin/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/bin/README.md

@@ -0,0 +1,37 @@
+# Google Natural Language Api filter plugin for Embulk
+
+Google Natural Language Api filter plugin for Embulk.
+
+## Overview
+
+* **Plugin type**: filter
+
+## Configuration
+
+- **api**: api type. analyzeSentiment or analyzeEntities or analyzeSyntax. (string, required)
+- **out_key_name_suffix**: out_key_name_suffix (string, required)
+- **key_names**: key_names (array, required)
+- **delay**: delay (integer, default: 0)
+- **google_api_key**: google_api_key (string, default: ENV['GOOGLE_API_KEY'])
+- **language**: language. default is auto detect. (string, default: nil)
+
+
+## Example
+
+```yaml
+filters:
+  - type: google_natural_language_api
+    api: analyzeSentiment
+    out_key_name_suffix: _parsed
+    language: en
+    key_names:
+      - title
+      - message
+```
+
+
+## Build
+
+```
+$ rake
+```

data/bin/Rakefile

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

data/bin/embulk-filter-google_natural_language_api.gemspec

@@ -0,0 +1,20 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-filter-google_natural_language_api"
+  spec.version       = "0.1.0"
+  spec.authors       = ["toyama0919"]
+  spec.summary       = "Google Natural Language Api filter plugin for Embulk"
+  spec.description   = "Google Natural Language Api filter plugin for Embulk"
+  spec.email         = ["toyama0919@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/toyama0919/embulk-filter-google_natural_language_api"
+
+  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_dependency 'jruby-openssl'
+  spec.add_development_dependency 'embulk', ['>= 0.8.15']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/bin/lib/embulk/filter/google_natural_language_api.rb

@@ -0,0 +1,74 @@
+require "json"
+require "net/http"
+require "uri"
+require "openssl"
+
+module Embulk
+  module Filter
+
+    class GoogleNaturalLanguageApi < FilterPlugin
+      Plugin.register_filter("google_natural_language_api", self)
+      ENDPOINT_PREFIX = "https://language.googleapis.com/v1/documents"
+
+      def self.transaction(config, in_schema, &control)
+        task = {
+          "api" => config.param("api", :string),
+          "out_key_name_suffix" => config.param("out_key_name_suffix", :string),
+          "key_names" => config.param("key_names", :array),
+          "delay" => config.param("delay", :integer, default: 0),
+          "google_api_key" => config.param("google_api_key", :string, default: ENV['GOOGLE_API_KEY']),
+          "language" => config.param("language", :string, default: nil),
+        }
+
+        add_columns = task["key_names"].map { |key_name|
+          Column.new(nil, key_name + task["out_key_name_suffix"], :json)
+        }
+
+        out_columns = in_schema + add_columns
+
+        yield(task, out_columns)
+      end
+
+      def init
+        @uri = URI.parse("#{ENDPOINT_PREFIX}:#{task['api']}?key=#{task['google_api_key']}")
+        @http = Net::HTTP.new(@uri.host, @uri.port)
+        @http.use_ssl = true
+        @http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        @request = Net::HTTP::Post.new(@uri.request_uri, initheader = {'Content-Type' =>'application/json'})
+        @key_names = task['key_names']
+        @out_key_name_suffix = task['out_key_name_suffix']
+        @delay = task['delay']
+        @body = {
+          "document" => {
+            "type" => "PLAIN_TEXT",
+          }
+        }
+        @body["document"]["language"] = task['language'] if task['language']
+      end
+
+      def close
+      end
+
+      def add(page)
+        page.each do |record|
+          hash = Hash[in_schema.names.zip(record)]
+          @key_names.each do |key_name|
+            @body["document"]["content"] = hash[key_name]
+            @request.body = @body.to_json
+
+            @http.start do |h|
+              response = h.request(@request)
+              hash[key_name + @out_key_name_suffix] = JSON.parse(response.body)
+            end
+          end
+          page_builder.add(hash.values)
+          sleep @delay
+        end
+      end
+
+      def finish
+        page_builder.finish
+      end
+    end
+  end
+end

data/embulk-filter-google_natural_language_api.gemspec

@@ -0,0 +1,20 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-filter-google_natural_language_api"
+  spec.version       = "0.1.0"
+  spec.authors       = ["toyama0919"]
+  spec.summary       = "Google Natural Language Api filter plugin for Embulk"
+  spec.description   = "Google Natural Language Api filter plugin for Embulk"
+  spec.email         = ["toyama0919@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/toyama0919/embulk-filter-google_natural_language_api"
+
+  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_dependency 'jruby-openssl'
+  spec.add_development_dependency 'embulk', ['>= 0.8.15']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/filter/google_natural_language_api.rb

@@ -0,0 +1,74 @@
+require "json"
+require "net/http"
+require "uri"
+require "openssl"
+
+module Embulk
+  module Filter
+
+    class GoogleNaturalLanguageApi < FilterPlugin
+      Plugin.register_filter("google_natural_language_api", self)
+      ENDPOINT_PREFIX = "https://language.googleapis.com/v1/documents"
+
+      def self.transaction(config, in_schema, &control)
+        task = {
+          "api" => config.param("api", :string),
+          "out_key_name_suffix" => config.param("out_key_name_suffix", :string),
+          "key_names" => config.param("key_names", :array),
+          "delay" => config.param("delay", :integer, default: 0),
+          "google_api_key" => config.param("google_api_key", :string, default: ENV['GOOGLE_API_KEY']),
+          "language" => config.param("language", :string, default: nil),
+        }
+
+        add_columns = task["key_names"].map { |key_name|
+          Column.new(nil, key_name + task["out_key_name_suffix"], :json)
+        }
+
+        out_columns = in_schema + add_columns
+
+        yield(task, out_columns)
+      end
+
+      def init
+        @uri = URI.parse("#{ENDPOINT_PREFIX}:#{task['api']}?key=#{task['google_api_key']}")
+        @http = Net::HTTP.new(@uri.host, @uri.port)
+        @http.use_ssl = true
+        @http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        @request = Net::HTTP::Post.new(@uri.request_uri, initheader = {'Content-Type' =>'application/json'})
+        @key_names = task['key_names']
+        @out_key_name_suffix = task['out_key_name_suffix']
+        @delay = task['delay']
+        @body = {
+          "document" => {
+            "type" => "PLAIN_TEXT",
+          }
+        }
+        @body["document"]["language"] = task['language'] if task['language']
+      end
+
+      def close
+      end
+
+      def add(page)
+        page.each do |record|
+          hash = Hash[in_schema.names.zip(record)]
+          @key_names.each do |key_name|
+            @body["document"]["content"] = hash[key_name]
+            @request.body = @body.to_json
+
+            @http.start do |h|
+              response = h.request(@request)
+              hash[key_name + @out_key_name_suffix] = JSON.parse(response.body)
+            end
+          end
+          page_builder.add(hash.values)
+          sleep @delay
+        end
+      end
+
+      def finish
+        page_builder.finish
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,116 @@
+--- !ruby/object:Gem::Specification
+name: embulk-filter-google_natural_language_api
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- toyama0919
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2017-01-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: jruby-openssl
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.15
+  name: embulk
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.15
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  name: bundler
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  name: rake
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Google Natural Language Api filter plugin for Embulk
+email:
+- toyama0919@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/.gitignore
+- bin/.ruby-version
+- bin/Gemfile
+- bin/LICENSE.txt
+- bin/README.md
+- bin/Rakefile
+- bin/embulk-filter-google_natural_language_api.gemspec
+- bin/lib/embulk/filter/google_natural_language_api.rb
+- embulk-filter-google_natural_language_api.gemspec
+- lib/embulk/filter/google_natural_language_api.rb
+homepage: https://github.com/toyama0919/embulk-filter-google_natural_language_api
+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.6.6
+signing_key:
+specification_version: 4
+summary: Google Natural Language Api filter plugin for Embulk
+test_files: []