deepl-srt 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fc7aa689e44c0c210dea218cc96ff9d2ab6cf139cf23dbec6b8dd6185f579ab1
-  data.tar.gz: 62e5bf5c573f35b7ab1f0a24ea07f604a30df9693f5522de33d3b6d89a7bbf68
+  metadata.gz: 041c6c0d7cc8a119bca0fca2c85773c1c33bf585c8be7abd44bd90dbfa682392
+  data.tar.gz: 3d54618cf8d7e56d93c4ad19680c13942365609917b5334a87eb5c91392c2f8f
 SHA512:
-  metadata.gz: 1a97eb350ec054b2ba2e636b9942e04d5ed8a48c82dccd06892bdf046afac84727c691468739a0092e9addcbc4df8cb818165257fb02479980c0a369f289f335
-  data.tar.gz: a1ef9dc12ca8f9d7693f70a2d440202992c4cd2ff2251938297a89afc10956d619547d1e775a41ace816741e8a2404d2934c395dc8a354a6c2155dd8748acaa3
+  metadata.gz: ce490c98abfd551bb2d1f39cd0fb6d46d38674f34d5d26de298834737db2ea6c9960671982a0b98a1ea70e0c3057085faf920661caea0de9809f473dec6e0abf
+  data.tar.gz: a3d699287177dd5ed8127953e0b9f745efa8d262dc756071067cc6941ff983d57ea243e9f5b3663fb60beb6a4eb3c5126d026fb74e51fb7956c04f5b398a7fee

data/CHANGELOG.md

@@ -1,4 +1,5 @@
 0.1.0 init release
 0.1.1 improvement in line translation
 0.1.2 split line every 6 words
-0.1.3 fix
+0.1.3 fix
+0.1.4 from line argument

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    deepl-srt (0.1.3)
+    deepl-srt (0.1.4)
 
 GEM
   remote: https://rubygems.org/

data/deepl-srt.gemspec

@@ -9,7 +9,7 @@ Gem::Specification.new do |spec|
   spec.email         = ['lukasz@fuszara.pl']
 
   spec.summary       = 'Subtitles translator with DeepL engine'
-  spec.description   = 'deepl_srt [api_key] [target_lang] [input path] [result_path]'
+  spec.description   = 'deepl_srt [api_key] [target_lang] [input_path] [result_path] [form_line]'
   spec.homepage      = 'https://github.com/lfuszara1/srt-deepl'
   spec.license       = 'MIT'
   spec.required_ruby_version = Gem::Requirement.new('>= 2.4.0')

data/examples/example.output.tmp

@@ -0,0 +1,15 @@
+1
+00:00:00,498 --> 00:00:02,827
+- Oto, co najbardziej lubię w
+jedzeniu i diecie.
+
+2
+00:00:02,827 --> 00:00:06,383
+Wszyscy jemy kilka razy dziennie i
+jesteśmy całkowicie odpowiedzialni za to
+
+3
+00:00:06,383 --> 00:00:09,427
+co trafia na nasz talerz, a
+co z niego schodzi.
+

data/exe/deepl_srt

@@ -0,0 +1,22 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'deepl_srt'
+
+input_params = ARGV
+
+if input_params.empty?
+  puts 'deepl_srt [api_key] [target_lang] [input_path] [result_path] [form_line]'
+else
+  api_key = input_params[0]
+  target_lang = input_params[1]
+  input_path = input_params[2]
+  result_path = input_params[3]
+  if input_params[4].nil?
+    from_line = 0
+  else
+    from_line = input_params[4]
+  end
+
+  DeeplSrt.deepl_srt(api_key, target_lang, input_path, result_path, from_line)
+end

data/lib/deepl_srt.rb

@@ -7,9 +7,8 @@ require_relative './deepl_srt/srt_translate'
 module DeeplSrt
   class Error < StandardError; end
 
-  def self.deepl_srt(api_key, path, target_lang, result_path)
-    srt_translate = SrtTranslate.new(api_key, path, target_lang)
-    result = srt_translate.parse
-    File.open(result_path, 'w') { |file| file.write(result) }
+  def self.deepl_srt(api_key, target_lang, input_path, result_path, from_line)
+    srt_translate = SrtTranslate.new(api_key, target_lang, input_path, result_path)
+    srt_translate.parse(from_line)
   end
 end

data/lib/deepl_srt/deepl_request.rb

@@ -11,7 +11,7 @@ class DeeplRequest
     @api_key = api_key
   end
 
-  def request(text, target_lang)
+  def request(target_lang, text)
     Net::HTTP.post_form(@uri, 'auth_key' => @api_key,
                               'text' => text,
                               'target_lang' => target_lang,

data/lib/deepl_srt/srt_translate.rb

@@ -1,21 +1,25 @@
 # frozen_string_literal: false
 
+require 'fileutils'
 require 'json'
 require 'srt'
 require_relative './deepl_request'
 
 # Class to translate subtitles
 class SrtTranslate
-  def initialize(api_key, path, target_lang)
-    @file = SRT::File.parse(File.new(path))
+  def initialize(api_key, target_lang, input_path, result_path)
+    @file = SRT::File.parse(File.new(input_path))
     @deepl_request = DeeplRequest.new(api_key)
     @target_lang = target_lang
+    FileUtils.rm(result_path) if File.exist?(result_path)
+    FileUtils.touch(result_path)
+    @result_path = result_path
   end
 
-  def parse
+  def parse(from_line)
     @new_content = ''
-    @file.lines.each_with_index do |line, index|
-      @new_content << parse_part(line, index)
+    @file.lines.drop(from_line.to_i).each_with_index do |line, index|
+      @new_content << parse_part(line, index + from_line.to_i)
     end
     @new_content
   end
@@ -24,20 +28,24 @@ class SrtTranslate
 
   def parse_part(line, index)
     str = ''
-    str << "#{index}\n"
+    str << "#{index + 1}\n"
     str << "#{line.time_str}\n"
     json = translate(line.text.join('\n'))
     result = shorter(JSON.parse(json)['translations'][0]['text'])
     str << "#{result}\n\n"
-    str
+    save_line(str)
   end
 
   def translate(text)
-    @deepl_request.request(text, @target_lang).body
+    @deepl_request.request(@target_lang, text).body
   end
 
   def shorter(line)
     short = line.split.each_slice(6).map { |a| a.join ' ' }
     short.join("\n")
   end
+
+  def save_line(line)
+    File.open(@result_path, 'a+') { |file| file.write(line) }
+  end
 end

data/lib/deepl_srt/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DeeplSrt
-  VERSION = '0.1.3'
+  VERSION = '0.1.4'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: deepl-srt
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Łukasz Fuszara
@@ -10,11 +10,11 @@ bindir: exe
 cert_chain: []
 date: 2021-05-28 00:00:00.000000000 Z
 dependencies: []
-description: deepl_srt [api_key] [target_lang] [input path] [result_path]
+description: deepl_srt [api_key] [target_lang] [input_path] [result_path] [form_line]
 email:
 - lukasz@fuszara.pl
 executables:
-- deepl_srt.rb
+- deepl_srt
 extensions: []
 extra_rdoc_files: []
 files:
@@ -32,8 +32,9 @@ files:
 - bin/console
 - bin/setup
 - deepl-srt.gemspec
-- examples/example.srt
-- exe/deepl_srt.rb
+- examples/example.input.srt
+- examples/example.output.tmp
+- exe/deepl_srt
 - lib/deepl_srt.rb
 - lib/deepl_srt/deepl_request.rb
 - lib/deepl_srt/srt_translate.rb

data/exe/deepl_srt.rb

@@ -1,17 +0,0 @@
-#!/usr/bin/env ruby
-# frozen_string_literal: true
-
-require 'deepl_srt'
-
-input_params = ARGV
-
-if input_params.empty?
-  puts 'deepl_srt [api_key] [target_lang] [input path] [result_path]'
-else
-  api_key = input_params[0]
-  target_lang = input_params[1]
-  path = input_params[2]
-  result_path = input_params[3]
-
-  DeeplSrt.deepl_srt(api_key, path, target_lang, result_path)
-end