llm_translator 0.0.1 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4ba8b71b3fbec395856782c1d39551cd6c0aae45f929acea782bebff7725b402
-  data.tar.gz: 336658ba8016e367826971801cd1ec80f5c0ccac741cbe8bced4ad15b6d137ce
+  metadata.gz: 2eaa70543efe06ebcff7a1eca548dc36e8ab0983a7d3b180b5953fec12c2c71f
+  data.tar.gz: 249fc33eb499585ac53bb95bbd1c4156c6c562d411752befd34c3946df0fd5f4
 SHA512:
-  metadata.gz: 18a70a5da904a92d758923f935fc0678dd8a8ea8bce9cd84c21c00c43bc766bfb096fb8bc04c3723c348e908f6d0f9a61e6834706cca0a9da4b1a83309f565d5
-  data.tar.gz: 163cb0170971b850c49df6a3a1231e623be192ba8211cafd7a0dc804c82ee81c9c39a45f6ae6a67fa42bdec78a5f8d415f4e4fbb481f060488dfbd5669d010ba
+  metadata.gz: 390b2f53bb562d431eb203aa007a1a46dbb9ad2a079abaa116de33007769eac1dadc30576650e411057ff0699d6dd19dff5aee27ab97647c225ce9b83e66b7a8
+  data.tar.gz: 423480766fcc286f93840540fc75bead95bcbb6a7a2cf7ded757faeb0c55393bbdc8c0e777abd33ade79457e9c1ab5ea6cfa1cea107c44c537df39f2039589a3

data/bin/llm_translator

@@ -1,15 +1,5 @@
 #!/usr/bin/env ruby
-
 require 'llm_translator'
 
-config = {
-  :appkey => '0b53e81a10846a3a',
-  :appscr => 'JLKxQO22jBft6w3PgLBh5as5iNJ1w5wS'
-}
-
-translator = LlmTranslator.new(config[:appkey], config[:appscr])
-if ARGV[0] == '-f'
-  puts(translator.translate_mix_w File::readlines(ARGV[1]).join)
-else
-  puts(translator.translate_mix_w ARGV[0])
-end
+llm = LlmTranslator.new nil 
+pp llm.command(ARGV)

data/lib/llm_translator.rb

@@ -3,114 +3,218 @@ require 'digest'
 require 'net/http'
 require 'securerandom'
 require 'json'
+require 'optparse'
+
+LANG_CODE = %|
+中文	zh-CHS
+中文繁体	zh-CHT
+英文	en
+日文	ja
+韩文	ko
+法文	fr
+西班牙文	es
+葡萄牙文	pt
+意大利文	it
+俄文	ru
+越南文	vi
+德文	de
+阿拉伯文	ar
+印尼文	id
+南非荷兰语	af
+波斯尼亚语	bs
+保加利亚语	bg
+粤语	yue
+加泰隆语	ca
+克罗地亚语	hr
+捷克语	cs
+丹麦语	da
+荷兰语	nl
+爱沙尼亚语	et
+斐济语	fj
+芬兰语	fi
+希腊语	el
+海地克里奥尔语	ht
+希伯来语	he
+印地语	hi
+白苗语	mww
+匈牙利语	hu
+斯瓦希里语	sw
+克林贡语	tlh
+拉脱维亚语	lv
+立陶宛语	lt
+马来语	ms
+马耳他语	mt
+挪威语	no
+波斯语	fa
+波兰语	pl
+克雷塔罗奥托米语	otq
+罗马尼亚语	ro
+塞尔维亚语(西里尔文)	sr-Cyrl
+塞尔维亚语(拉丁文)	sr-Latn
+斯洛伐克语	sk
+斯洛文尼亚语	sl
+瑞典语	sv
+塔希提语	ty
+泰语	th
+汤加语	to
+土耳其语	tr
+乌克兰语	uk
+乌尔都语	ur
+威尔士语	cy
+尤卡坦玛雅语	yua
+阿尔巴尼亚语	sq
+阿姆哈拉语	am
+亚美尼亚语	hy
+阿塞拜疆语	az
+孟加拉语	bn
+巴斯克语	eu
+白俄罗斯语	be
+宿务语	ceb
+科西嘉语	co
+世界语	eo
+菲律宾语	tl
+弗里西语	fy
+加利西亚语	gl
+格鲁吉亚语	ka
+古吉拉特语	gu
+豪萨语	ha
+夏威夷语	haw
+冰岛语	is
+伊博语	ig
+爱尔兰语	ga
+爪哇语	jw
+卡纳达语	kn
+哈萨克语	kk
+高棉语	km
+库尔德语	ku
+柯尔克孜语	ky
+老挝语	lo
+拉丁语	la
+卢森堡语	lb
+马其顿语	mk
+马尔加什语	mg
+马拉雅拉姆语	ml
+毛利语	mi
+马拉地语	mr
+蒙古语	mn
+缅甸语	my
+尼泊尔语	ne
+齐切瓦语	ny
+普什图语	ps
+旁遮普语	pa
+萨摩亚语	sm
+苏格兰盖尔语	gd
+塞索托语	st
+修纳语	sn
+信德语	sd
+僧伽罗语	si
+索马里语	so
+巽他语	su
+塔吉克语	tg
+泰米尔语	ta
+泰卢固语	te
+乌兹别克语	uz
+南非科萨语	xh
+意第绪语	yi
+约鲁巴语	yo
+南非祖鲁语	zu|
 
 class LlmTranslator
-  attr_accessor :q, :from, :to, :debug
-  T_URL_PATH = "https://openapi.youdao.com/api"
-  EN_REGEX = /\w+/
-  ZH_REGEX = /[\u4e00-\u9fa5]+/
-  MIX_EN_ZH_REGEX = /\w+|[^\w]+/
-  MIX_ZH_EN_REGEX = /[\u4e00-\u9fa5]+|[^\u4e00-\u9fa5]+/
-  ZH_SYMBOL = 'zh-CHS'
-  EN_SYMBOL = 'en'
 
-  def initialize(appkey, appscr, sign_type='v3')
-    @appkey = appkey
-    @appscr = appscr
-    @sign_type = sign_type
-    @dict_s_buffer = {}
-    @dict_w_buffer = {}
-    @debug = false
-  end
-  
-  def translate_parsed(q, from=nil, to=nil, ext=nil, voice=nil, strict=nil, vocabld=nil)
-    json = translate_request q, from, to
-    begin
-      JSON.parse json
-    rescue => exception
-      puts exception
-      puts json if @debug
-    end
-  end
+  CONFIG_PATH = File.expand_path('~/.llm_translator')
+  attr_accessor :text,
+                :history,
+                :auto, # 自动中英互译
+                :detail,  # 详细翻译
+                :from, # 源语言
+                :to    # 目标语言
 
-  def translate_get_w(q, from=nil, to=nil, ext=nil, voice=nil, strict=nil, vocabld=nil)
-    return @dict_w_buffer[q] if !@dict_w_buffer[q].nil?
-    res = translate_parsed(q, from, to)
-    begin
-      @dict_w_buffer[q] = res['translation'][0]
-    rescue => exception
-      puts "havn't result for #{q}"
-      puts res.to_json if @debug
+  def initialize(text)
+    if !Dir.exists?(CONFIG_PATH)
+      Dir.mkdir(CONFIG_PATH)
     end
+    @text   = text
+    @history= File.open("#{CONFIG_PATH}/history.csv", 'a+')
+    @url    = 'https://openapi.youdao.com/api'
+    @appkey = '0b53e81a10846a3a'
+    @appscr = 'JLKxQO22jBft6w3PgLBh5as5iNJ1w5wS'
+    @sign_type = 'v3'
+    @auto  = true
+    @detail= false
   end
-
-  def translate_get_web_info(q, from=nil, to=nil, ext=nil, voice=nil, strict=nil, vocabld=nil)
-    return @dict_s_buffer[q] if !@dict_s_buffer[q].nil?
-    res = translate_parsed(q, from, to)
-    begin
-      @dict_s_buffer[q] = res['web']
-    rescue => exception
-      puts "havn't result for #{q}"
-      puts res.to_json if @debug
-    end
-  end
-
-  def translate_mix_w(q)
-    if (from == EN_SYMBOL)
-      q.scan(MIX_EN_ZH_REGEX).map { |w| w.match?(EN_REGEX) ? snake_style(translate_get_w(w)) : w }.join
-    else
-      q.scan(MIX_ZH_EN_REGEX).map { |w| w.match?(ZH_REGEX) ? snake_style(translate_get_w(w)) : w }.join
-    end
-  end  
-
-  def translate_request(q, from=nil, to=nil, ext=nil, voice=nil, strict=nil, vocabld=nil)
-    @curtime = Time.now.to_i.to_s
-    @q = q
-    @from = from
-    @to = to
-    auto_from_to q if @from.nil? || @to.nil?
-    @salt=SecureRandom::uuid
+  
+  def to_result
+    salt=SecureRandom::uuid
+    curtime = Time.now.utc.to_i.to_s
+    input = @text.length > 20 ? @text[0..10] + @text.length.to_s + @text[-10..-1] : @text
     params = {
-      q: @q,
-      from: @from,
-      to: @to,
-      appKey: @appkey,
-      salt: @salt,
-      sign: LlmTranslator::sign(@appkey, @appscr, LlmTranslator::input(q), @curtime, @salt),
+      q:        @text,
+      from:     @from,
+      to:       @to,
+      appKey:   @appkey,
+      salt:     salt,
+      sign:     Digest::SHA256.hexdigest(@appkey + input + salt + curtime + @appscr),
       signType: @sign_type,
-      curtime: @curtime
+      curtime:  curtime,
+      strict:   !@auto
     }
-    @uri = URI(T_URL_PATH)
-    res = Net::HTTP.post_form(@uri, params)
-    res.body if res.is_a?(Net::HTTPSuccess)
+    JSON.parse Net::HTTP.post_form(URI(@url), params).body
   end
 
-  def self.sign(appkey, appscr, input, curtime=Time.now.utc.to_s, salt=SecureRandom::uuid)
-    Digest::SHA256.hexdigest(appkey + input + salt + curtime + appscr)
-  end
+  def run
+    result = to_result()
+    history.write("#{text} #{result['translation'][0]} #{Time.now}\n") if @history
+    return result if @detail
 
-  def self.input(q)
-    input = q.length > 20 ? q[0,10] + q.length.to_s + q.reverse[0, 10].reverse : q
+    result['translation'][0]
   end
 
-  private
-  def auto_from_to(q)
-    if q.match?ZH_REGEX
-      @from = ZH_SYMBOL
-      @to = EN_SYMBOL
-    else
-      @from = EN_SYMBOL
-      @to = ZH_SYMBOL
-    end
-  end
+  def command(args)
+    OptionParser.new do |parser|
+      parser.banner = "用法: llm_translator [options]"
+      parser.separator ""
+      parser.separator "指定选项:"
 
-  def camel_style(s)
-    s = s.split(' ').map! {|w| w.capitalize}.join
-    s[0] = s[0].downcase
-    s
-  end
+      parser.on('-d', '--detail', '显示详细翻译结果,默认false')      do
+        self.detail = true
+      end
+      parser.on('-a', '--auto', '严格按照指定from和to进行，默认false') do
+        self.auto   = true
+      end
+
+      parser.on('-i TEXT', '--input=TEXT', '待翻译文本, 必填项') do |text|
+        self.text = text
+      end
+      parser.on('-f FROM', '--from=FROM', '源语言')      do |from|
+        self.from = from
+      end
+      parser.on('-t TO', '--to=TO', '目标语言')          do |to|
+        self.to = to
+      end
+
+      parser.separator ""
+      parser.separator "通用选项:"
+      parser.on_tail("-h", "--help", "帮助信息") do
+        puts parser
+        exit
+      end
+      parser.on_tail("--lang", "语言编码信息") do
+        puts LANG_CODE
+        exit
+      end
+      parser.on_tail("--version", "版本信息") do
+        puts '0.0.1'
+        exit
+      end
+
+    end.parse!(args)
+
+    if self.text.nil?
+      puts '请输入文本'
+      puts '-i, --input=TEXT                 待翻译文本, 必填项'
+      exit
+    end
 
-  def snake_style(s)
-    s.split(' ').map {|w| w.downcase}.join('_')
+    run
   end
-end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_translator
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.4
 platform: ruby
 authors:
 - Liuliming
@@ -39,7 +39,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
+rubygems_version: 3.2.3
 signing_key: 
 specification_version: 4
 summary: a simple translator!