embulk-filter-google_translate_api 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9a7226d317141e558edc1c87560d2bb04beef7de
-  data.tar.gz: c30c74eb6838a2dff7142d552dfe80d38029ed0c
+  metadata.gz: 1262edf05cdf86f48da5cf116d64f08b3aaebea5
+  data.tar.gz: 193cefceda03cb422b022b00a7f3de7d70547a7f
 SHA512:
-  metadata.gz: f050d824b14b0e8892e23f2de4f4bfd9e11765589508abd177b78aa1dad64d95c7ea6b2bc28969aefd5169154466788d83a49b2467c7a634c41b52b4c2cc26d6
-  data.tar.gz: 468b418d545d133147f6f7f22822eae9596e92ed4095d4350ea31a2532a8f75f13c7ecc5d02e8ce86e4891f06137cfe0cbb4e7711b9f92d8ea8dda4e31f2637f
+  metadata.gz: a1cbeb0c236d8008dc725ac0fd12f8df8c4cd3e4541a41270dac43d8bc703b566ce412c234b5071e393c4683f02457f5a27dd16a6cecd9f34afb321dafe577fd
+  data.tar.gz: ca85d7abc4a200f715a23c346eaec0d7f642398b14458bde17b3c5930cf553aa31ed38daabdf8818abb83a4c87910b42071cfc763afca5887d308076a7c99489

data/README.md

@@ -14,6 +14,7 @@ see support language. [Google Language Codes \- tomihasa](https://sites.google.c
 - **out_key_name_suffix**: translated target key names suffix (string, required)
 - **source_lang**: source language (string, default: `null`)
 - **target_lang**: target language (string, required)
+- **model**: if premium edition can use. nmt(neural machine translation) or base. (string, default: `null`)
 - **sleep**: delay per record, define milliseconds. (integer, default: 0)
 - **google_api_key**: google_api_key. support environment variable. please `export GOOGLE_API_KEY`(string, default: `null`)
 

data/build.gradle

@@ -14,7 +14,7 @@ configurations {
     provided
 }
 
-version = "0.1.1"
+version = "0.2.0"
 
 sourceCompatibility = 1.7
 targetCompatibility = 1.7

data/src/main/java/org/embulk/filter/google_translate_api/GoogleTranslateApiFilterPlugin.java

@@ -33,6 +33,10 @@ public class GoogleTranslateApiFilterPlugin implements FilterPlugin
         @Config("target_lang")
         public String getTargetLang();
 
+        @Config("model")
+        @ConfigDefault("null")
+        public Optional<String> getModel();
+
         @Config("google_api_key")
         @ConfigDefault("null")
         public Optional<String> getGoogleApiKey();

data/src/main/java/org/embulk/filter/google_translate_api/GoogleTranslateApiPageOutput.java

@@ -2,6 +2,7 @@ package org.embulk.filter.google_translate_api;
 
 import java.util.List;
 
+import org.apache.commons.lang3.ArrayUtils;
 import org.embulk.config.TaskSource;
 import org.embulk.filter.google_translate_api.GoogleTranslateApiFilterPlugin.PluginTask;
 import org.embulk.spi.Column;
@@ -32,6 +33,7 @@ public class GoogleTranslateApiPageOutput implements PageOutput
     private final PageBuilder builder;
     private final Translate translate;
     private final TranslateOption srcLang;
+    private final TranslateOption model;
     
     private static final Logger logger = Exec.getLogger(GoogleTranslateApiFilterPlugin.class);
 
@@ -48,6 +50,7 @@ public class GoogleTranslateApiPageOutput implements PageOutput
         this.translate = createTranslateService();
 
         this.srcLang = task.getSourceLang().isPresent() ? TranslateOption.sourceLanguage(task.getSourceLang().get()) : null;
+        this.model = task.getModel().isPresent() ? TranslateOption.model(task.getModel().get()) : null;
     }
 
     @Override
@@ -90,7 +93,14 @@ public class GoogleTranslateApiPageOutput implements PageOutput
      * @return
      */
     private List<Translation> translate(List<String> texts) {
-        return (srcLang == null) ? translate.translate(texts) : translate.translate(texts, srcLang);
+        TranslateOption[] translateOptions = new TranslateOption[] {};
+        if (srcLang != null) {
+            translateOptions = ArrayUtils.add(translateOptions, srcLang);
+        }
+        if (model != null) {
+            translateOptions = ArrayUtils.add(translateOptions, model);
+        }
+        return (translateOptions.length == 0) ? translate.translate(texts) : translate.translate(texts, translateOptions);
     }
 
     /**

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: embulk-filter-google_translate_api
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - toyama0919
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2017-01-16 00:00:00.000000000 Z
+date: 2017-01-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   requirement: !ruby/object:Gem::Requirement
@@ -63,7 +63,7 @@ files:
 - classpath/auto-value-1.2.jar
 - classpath/commons-codec-1.3.jar
 - classpath/commons-logging-1.1.1.jar
-- classpath/embulk-filter-google_translate_api-0.1.1.jar
+- classpath/embulk-filter-google_translate_api-0.2.0.jar
 - classpath/gax-0.0.25.jar
 - classpath/google-api-client-1.21.0.jar
 - classpath/google-api-client-appengine-1.21.0.jar