embulk-filter-azure_translator_api 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: afa13d2b165a2a901773a492d11e0ef786245125
+  data.tar.gz: 4b8d2da4781f4bfcb20f66527d69f666522225ef
+SHA512:
+  metadata.gz: 0b0da7a63c4d1c61cec6bbec439c1de470c8ad84a662ac384897a91ee0b1ed9a2ea5dfeff24caba3a0fedb3d8a0e8fa33d9810bcb565749c21a8505881ee2105
+  data.tar.gz: 598e775bab15d78c62a3f5e8c27521fce63f9b2baf74bbb7b6b42a3780abf6fccb47d2c90af2ac265a1a517346d1c5bd74a9d4c38cd54873525cce8e88509f8d

data/.gitignore

@@ -0,0 +1,5 @@
+*~
+/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,43 @@
+# Azure Translator Api filter plugin for Embulk
+
+Write short description here and embulk-filter-azure_translator_api.gemspec file.
+
+## Documentation
+* [seamlessly](https://www.microsoft.com/cognitive-services/en-us/translator-api/documentation/Text/overview)
+* [Microsoft Cognitive Services](http://docs.microsofttranslator.com/text-translate.html)
+
+## Overview
+
+* **Plugin type**: filter
+
+## Configuration
+- **api_type**: api_type (string),
+- **out_key_name_suffix**: out_key_name_suffix (string),
+- **key_names**: key_names (array),
+- **params**: params (hash, default: {}),
+- **delay**: delay (integer, default: 0),
+- **subscription_key**: subscription_key (string),
+- **content_type**: content_type (string, default: nil),
+- **category**: category (string, default: nil),
+- **to**: to (string),
+- **from**: from (string, default: nil)
+
+## Example
+
+```yaml
+filters:
+  - type: azure_translator_api
+    key_names:
+      - string_column
+      - string_column2
+    out_key_name_suffix: translated
+    subscription_key: XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
+    api_type: Translate
+    to: ja
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-filter-azure_translator_api.gemspec

@@ -0,0 +1,21 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-filter-azure_translator_api"
+  spec.version       = "0.1.0"
+  spec.authors       = ["toyama0919"]
+  spec.summary       = "Azure Translator Api filter plugin for Embulk"
+  spec.description   = "Azure Translator Api"
+  spec.email         = ["toyama0919@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/toyama0919/embulk-filter-azure_translator_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 'rest-client'
+  spec.add_dependency 'crack'
+  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/azure_translator_api.rb

@@ -0,0 +1,73 @@
+require_relative  'azure_translator_api/azure_translator_client'
+
+module Embulk
+  module Filter
+
+    class AzureTranslatorApi < FilterPlugin
+      Plugin.register_filter("azure_translator_api", self)
+
+      def self.transaction(config, in_schema, &control)
+        task = {
+          "api_type" => config.param("api_type", :string),
+          "out_key_name_suffix" => config.param("out_key_name_suffix", :string),
+          "key_names" => config.param("key_names", :array),
+          "params" => config.param("params", :hash, default: {}),
+          "delay" => config.param("delay", :integer, default: 0),
+          "subscription_key" => config.param("subscription_key", :string),
+          "content_type" => config.param("content_type", :string, default: nil),
+          "category" => config.param("category", :string, default: nil),
+          "to" => config.param("to", :string),
+          "from" => config.param("from", :string, default: nil)
+        }
+
+        add_columns = task['key_names'].map do |key_name|
+          Column.new(nil, key_name + task["out_key_name_suffix"], :string)
+        end
+
+        out_columns = in_schema + add_columns
+        task['authorization_token'] = AzureTranslatorClient.get_authorization_token(task["subscription_key"])
+
+        yield(task, out_columns)
+      end
+
+      def init
+        @delay = task['delay']
+        @key_names = task['key_names']
+        @out_key_name_suffix = task['out_key_name_suffix']
+
+        params = {
+          'to' => task['to'],
+        }
+        params['from'] = task['from'] if task['from']
+        params['content_type'] = task['content_type'] if task['content_type']
+        params['category'] = task['category'] if task['category']
+        @client = AzureTranslatorClient.new(
+          params: params,
+          authorization_token: task['authorization_token'],
+          api_type: task['api_type']
+        )
+      end
+
+      def close
+      end
+
+      def add(page)
+        records = page.map do |record|
+          Hash[in_schema.names.zip(record)]
+        end
+
+        records.each do |record|
+          @key_names.each do |key_name|
+            record[key_name + @out_key_name_suffix] = @client.translate_text(record[key_name])
+          end
+          page_builder.add(record.values)
+          sleep @delay
+        end
+      end
+
+      def finish
+        page_builder.finish
+      end
+    end
+  end
+end

data/lib/embulk/filter/azure_translator_api/azure_translator_client.rb

@@ -0,0 +1,47 @@
+require 'crack/xml'
+require 'rest-client'
+
+module Embulk
+  module Filter
+    class AzureTranslatorApi < FilterPlugin
+      class AzureTranslatorClient
+        ENDPOINT_PREFIX = "https://api.microsofttranslator.com/v2/http.svc"
+        AUTH_ENDPOINT_PREFIX = "https://api.cognitive.microsoft.com/sts/v1.0/issueToken"
+
+        def self.get_authorization_token(subscription_key)
+          RestClient.post AUTH_ENDPOINT_PREFIX, "", {
+            params: { 'Subscription-Key' => subscription_key }
+          }
+        end
+
+        def initialize(params: , authorization_token: , api_type:)
+          uri_string = "#{ENDPOINT_PREFIX}/#{api_type}"
+          @resource = RestClient::Resource.new uri_string
+          @params = params
+          @authorization_token = authorization_token
+        end
+
+        def translate_text(text)
+          translate(text)['string']
+        end
+
+        def translate(text)
+          request(text)
+        end
+
+        def request(text)
+          request_param = { text: text }.merge(@params)
+          Embulk.logger.debug("request_param => #{request_param}")
+          ::Crack::XML.parse(
+            @resource.get(
+              {
+                Authorization: "Bearer #{@authorization_token}",
+                params: request_param
+              }
+            )
+          )
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,123 @@
+--- !ruby/object:Gem::Specification
+name: embulk-filter-azure_translator_api
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- toyama0919
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2017-01-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: rest-client
+  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'
+  name: crack
+  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: Azure Translator Api
+email:
+- toyama0919@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-filter-azure_translator_api.gemspec
+- lib/embulk/filter/azure_translator_api.rb
+- lib/embulk/filter/azure_translator_api/azure_translator_client.rb
+homepage: https://github.com/toyama0919/embulk-filter-azure_translator_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: Azure Translator Api filter plugin for Embulk
+test_files: []