arxiv-references 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 572c835aeb2f46e94a5ef6a14344d1ba5d69cc7b
-  data.tar.gz: ac5a480450dd8aaf65b700cacf9d1974498ff99c
+  metadata.gz: 225161a6b0503ca774a846ad804eb205144d6c5a
+  data.tar.gz: c48bee9f466545bc878471e022bce09e936a0b05
 SHA512:
-  metadata.gz: 6ca33ba794a7af28fe4504e201dd2d328bb82c524ae892d41057fdb17697981638067518beaef9a0251910bfb3435bf3c8caaa3b51c0b369b551e01b2db3ee37
-  data.tar.gz: c3dd5c93e659ccc2e5abee5ab9abab605d7673fb5c914afcebf10be4f8b5186aaacf0cbb427a867d5305597ca4e689ca768e34e29d363910bc9d06397345b6ec
+  metadata.gz: ee917aeaf67117fa8a9041574262e4c939908dd95a362c5530ca86c97400d4776c329b12e8f0c7ca1c1c62ed9ce56feaccfea2e95d032d8af0f4e1b2f3e739d8
+  data.tar.gz: 7e20169f3ed005131be57a33d3e4563a430121f553974b3e2dc4e5241b071200c6da8cfb4ecdfba7108b943b03e9a1ab926ed51a53105f2dca29c19b046ea0bc

data/lib/arxiv/references/ArxivReferences.rb

@@ -11,18 +11,23 @@ module ArxivReferences
   class CLI < Thor
     include ArxivUtil
     desc 'url', 'Extract references from arxiv URL'
+    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
     def url(urlName)
-      puts ArxivUtil.fetchFromUrl(urlName)
+      puts ArxivUtil.fetchFromUrl(urlName, work_dir)
     end
 
     desc 'id', 'Extract references from Arxiv id'
+    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
     def arxivid(idName)
-      puts ArxivUtil.fetchFromArxivId(idName)
+      work_dir = options['work_dir'].nil? ? '/tmp' : options['work_dir']
+      puts ArxivUtil.fetchFromArxivId(idName, work_dir)
     end
 
     desc 'pdfurl', 'Extract references from pdf URL'
+    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
     def pdfurl(pdfUrlName)
-      puts ArxivUtil.fetchFromPdfUrl(pdfUrlName)
+      work_dir = options['work_dir'].nil? ? '/tmp' : options['work_dir']
+      puts ArxivUtil.fetchFromPdfUrl(pdfUrlName, work_dir)
     end
   end
 end

data/lib/arxiv/references/myUtil.rb

@@ -15,24 +15,24 @@ module ArxivUtil
     return Digest::SHA256.hexdigest Time.now.strftime("%F %H:%M:%S")
   end
 
-  def self.makeDir(id)
-    Dir.mkdir("/tmp/#{id}") 
+  def self.makeDir(id, work_dir)
+    Dir.mkdir("#{work_dir}/#{id}") 
   end
 
-  def self.removeDir(id)
-    FileUtils.rm_rf("/tmp/#{id}")
+  def self.removeDir(id, work_dir)
+    FileUtils.rm_rf("#{work_dir}/#{id}")
   end
 
-  def self.makeFile(id)
-    return "/tmp/#{id}/output.pdf"
+  def self.makeFile(id, work_dir)
+    return "#{work_dir}/#{id}/output.pdf"
   end
 
-  def self. getK2Pdf(id)
-    return "/tmp/#{id}/output_k2opt.pdf"
+  def self. getK2Pdf(id, work_dir)
+    return "#{work_dir}/#{id}/output_k2opt.pdf"
   end
 
 
-  def self.fetchFromUrl(urlName)
+  def self.fetchFromUrl(urlName, work_dir)
     puts "fetch => #{urlName}"
     charset = nil
     html = open(urlName) do |f|
@@ -46,19 +46,19 @@ module ArxivUtil
     result[:authors] = page.xpath('//*[@id="abs"]/div[2]/div[2]/a').text
     result[:abstruct] = page.xpath('//*[@id="abs"]/div[2]/blockquote').text
     result[:pdfurl] = "#{BASE_URL}#{page.xpath('//*[@id="abs"]/div[1]/div[1]/ul/li[1]/a').attr('href').value}"
-    result[:references] = fetchFromPdfUrl(result[:pdfurl]) 
+    result[:references] = fetchFromPdfUrl(result[:pdfurl], work_dir) 
     return result.to_json
   end
 
-  def self.fetchFromArxivId(id)
+  def self.fetchFromArxivId(id, work_dir)
     target_url = "#{BASE_URL}/abs/#{id}" 
-    fetchFromUrl(target_url)
+    fetchFromUrl(target_url, work_dir)
   end
 
-  def self.fetchFromPdfUrl(pdfUrl)
+  def self.fetchFromPdfUrl(pdfUrl, work_dir)
     job_id = makeId
-    makeDir(job_id)
-    file_name = makeFile(job_id)
+    makeDir(job_id, work_dir)
+    file_name = makeFile(job_id, work_dir)
 
     open(file_name, 'wb') do |o|
       open(pdfUrl) do |data|
@@ -78,7 +78,7 @@ module ArxivUtil
         break unless res.index('written').nil?
       end
     end
-    executed_pdf = getK2Pdf(job_id)
+    executed_pdf = getK2Pdf(job_id, work_dir)
     reader = PDF::Reader.new(executed_pdf)
     page_no = reader.
       pages.
@@ -106,7 +106,7 @@ module ArxivUtil
       select{|i|
       i.length > 5
     }
-    removeDir(job_id)
+    removeDir(job_id, work_dir)
     return references
   end
 end

data/lib/arxiv/references/version.rb

@@ -1,5 +1,5 @@
 module Arxiv
   module References
-    VERSION = "0.1.4"
+    VERSION = "0.1.5"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: arxiv-references
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Takahiro Nishimura