embulk-filter-amazon_rekognition 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: be495b0edfb083fe1529919d862ebf95317c6e70
+  data.tar.gz: 7f02b004fa5cc4a4bce2bde54c6b21e0e60d43b5
+SHA512:
+  metadata.gz: cd786c0d0959240d6e08ea8a91c318069139b99f095bcea32e6329c590942555efc9f7d61035fa0e6a46b5eeb89e0a16065de6bc20b376a5385571e895f39d5a
+  data.tar.gz: ae9f96da47ac79b3adf751a1202bf34fa4147e9d092bc34703deb208e162837c93898423e24090ba5bb30b8ae166fffe9800148654362719d2214e2e18ed18c8

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,35 @@
+# Amazon Rekognition filter plugin for Embulk
+
+* [Amazon Rekognition \| AWS](https://aws.amazon.com/jp/rekognition/)
+* [Amazon Rekognition – Image Detection and Recognition Powered by Deep Learning \| AWS Blog](https://aws.amazon.com/jp/blogs/aws/amazon-rekognition-image-detection-and-recognition-powered-by-deep-learning/)
+* [Class: Aws::Rekognition::Client — AWS SDK for Ruby V2](https://docs.aws.amazon.com/sdkforruby/api/Aws/Rekognition/Client.html)
+
+## Overview
+
+* **Plugin type**: filter
+
+## Configuration
+
+- **api_type**: api_type. detect_faces or detect_labels.(string)
+- **out_key_name**: out_key_name(string)
+- **image_path_key_name**: image_path_key_name(string)
+- **delay**: delay(integer, default: 0)
+- **aws_access_key_id**: aws_access_key_id(string, default: nil)
+- **aws_region**: aws_region(string, default: 'us-east-1')
+
+## Example
+
+```yaml
+filters:
+  - type: amazon_rekognition
+    api_type: detect_labels
+    out_key_name: amazon_rekognition_info
+    image_path_key_name: image_path
+```
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

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

data/embulk-filter-amazon_rekognition.gemspec

@@ -0,0 +1,20 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-filter-amazon_rekognition"
+  spec.version       = "0.1.0"
+  spec.authors       = ["toyama0919"]
+  spec.summary       = "Amazon Rekognition filter plugin for Embulk"
+  spec.description   = "Amazon Rekognition"
+  spec.email         = ["toyama0919@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/toyama0919/embulk-filter-amazon_rekognition"
+
+  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 'aws-sdk-rekognition', ['1.0.0.rc3']
+  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/amazon_rekognition.rb

@@ -0,0 +1,92 @@
+require 'aws-sdk-rekognition'
+
+module Embulk
+  module Filter
+    class AmazonRekognition < FilterPlugin
+      Plugin.register_filter("amazon_rekognition", self)
+      ENABLE_API = ['detect_faces', 'detect_labels']
+
+      def self.transaction(config, in_schema, &control)
+        task = {
+          "api_type" => config.param("api_type", :string),
+          "out_key_name" => config.param("out_key_name", :string),
+          "image_path_key_name" => config.param("image_path_key_name", :string),
+          "delay" => config.param("delay", :integer, default: 0),
+          "aws_access_key_id" => config.param("aws_access_key_id", :string, default: nil),
+          "aws_secret_access_key" => config.param("aws_secret_access_key", :string, default: nil),
+          "aws_region" => config.param("aws_region", :string, default: 'us-east-1'),
+        }
+
+        unless ENABLE_API.include?(task['api_type'])
+          raise ConfigError.new "Not support api => [#{task['api_type']}]"
+        end
+
+        Embulk.logger.info "api => [#{task['api_type']}]"
+
+        add_columns = [
+          Column.new(nil, task["out_key_name"], :json)
+        ]
+
+        out_columns = in_schema + add_columns
+
+        yield(task, out_columns)
+      end
+
+      def init
+        @image_path_key_name = task['image_path_key_name']
+        @delay = task['delay']
+        @api_type = task['api_type'].to_sym
+        @client = Aws::Rekognition::Client.new(
+          access_key_id: task['aws_access_key_id'],
+          secret_access_key: task['aws_secret_access_key'],
+          region: task['aws_region']
+        )
+      end
+
+      def close
+      end
+
+      def add(page)
+        page.each do |record|
+          hash = Hash[in_schema.names.zip(record)]
+          image_path = hash[@image_path_key_name]
+          Embulk.logger.info "Amazon Rekognition #{@api_type} processing.. #{image_path}"
+          page_builder.add(hash.values + [get_response(image_path)])
+          sleep @delay
+        end
+      end
+
+      def finish
+        page_builder.finish
+      end
+
+      def get_response(image_path)
+        image_params = get_image_params(image_path)
+        resp = @client.send(
+                 @api_type,
+                 { image: image_params }
+               )
+        resp.to_h
+      rescue => e
+        Embulk.logger.warn "#{image_path}\n#{e.message}\n#{e.backtrace.join("\n")}"
+        return { error_message: e.message }
+      end
+
+      def get_image_params(image_path)
+        if image_path =~ /s3\:\/\//
+          s3_uri = URI.parse(image_path)
+          return {
+              s3_object: {
+                bucket: s3_uri.host, 
+                name: s3_uri.path, 
+              }
+            }
+        elsif image_path =~ /https?\:\/\//
+          return { bytes: Net::HTTP.get_response(URI.parse(image_path)).body }
+        else
+          return { bytes: File.read(image_path) }
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,108 @@
+--- !ruby/object:Gem::Specification
+name: embulk-filter-amazon_rekognition
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- toyama0919
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2017-02-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.0.0.rc3
+  name: aws-sdk-rekognition
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.0.0.rc3
+- !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: Amazon Rekognition
+email:
+- toyama0919@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-filter-amazon_rekognition.gemspec
+- lib/embulk/filter/amazon_rekognition.rb
+homepage: https://github.com/toyama0919/embulk-filter-amazon_rekognition
+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: Amazon Rekognition filter plugin for Embulk
+test_files: []