Class: Gini::Api::Document

Inherits:
Object
  • Object
show all
Defined in:
lib/gini-api/document.rb

Overview

Contains document related data from uploaded or fetched document

Defined Under Namespace

Classes: Extractions, Layout

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(api, location, from_data = nil) ⇒ Document

Instantiate a new Gini::Api::Document object from URL

Parameters:

  • api (Gini::Api::Client) —

    Gini::Api::Client object

  • location (String) —

    Document URL

  • from_data (Hash) (defaults to: nil) —

    Hash with doc data (from search for example)



16
17
18
19
20
21
 
# File 'lib/gini-api/document.rb', line 16

def initialize(api, location, from_data = nil)
  @api      = api
  @location = location

  update(from_data)
end

Instance Attribute Details

#duration ⇒ Object

Returns the value of attribute duration.



8
9
10
 
# File 'lib/gini-api/document.rb', line 8

def duration
  @duration
end

Instance Method Details

#completed? ⇒ Boolean

Indicate if the document has been processed

Returns:

  • (Boolean) —

    true if progress == PENDING



71
72
73
 
# File 'lib/gini-api/document.rb', line 71

def completed?
  @progress != 'PENDING'
end

#extractions ⇒ Gini::Api::Document::Extractions

Initialize extractions from @_links and return Gini::Api::Extractions object

Returns:



106
107
108
 
# File 'lib/gini-api/document.rb', line 106

def extractions
  @extractions ||= Gini::Api::Document::Extractions.new(@api, @_links[:extractions])
end

#layout ⇒ Gini::Api::Document::Layout

Initialize layout from @_links and return Gini::Api::Layout object

Returns:



114
115
116
 
# File 'lib/gini-api/document.rb', line 114

def layout
  @layout ||= Gini::Api::Document::Layout.new(@api, @_links[:layout])
end

#pages ⇒ Object

Override @pages instance variable. Removes key :pageNumber, key :images and starts by index 0. Page 1 becomes index 0



121
122
123
 
# File 'lib/gini-api/document.rb', line 121

def pages
  @pages.map { |page| page[:images] }
end

#poll(interval, &block) ⇒ Object

Poll document progress and return when state equals COMPLETED Known states are PENDING, COMPLETED and ERROR

Parameters:

  • interval (Float) —

    API polling interval



58
59
60
61
62
63
64
65
 
# File 'lib/gini-api/document.rb', line 58

def poll(interval, &block)
  until @progress =~ /(COMPLETED|ERROR)/ do
    update
    yield self if block_given?
    sleep(interval)
  end
  nil
end

#processed ⇒ data

Get processed document

Returns:

  • (data) —

    The binary representation of the processed document (pdf, jpg, png, …)



87
88
89
90
91
92
93
94
95
96
97
98
99
100
 
# File 'lib/gini-api/document.rb', line 87

def processed
  response = @api.request(
    :get,
    @_links[:processed],
    headers: { accept: 'application/octet-stream' }
  )
  unless response.status == 200
    raise Gini::Api::DocumentError.new(
      "Failed to fetch processed document (code=#{response.status})",
      response
    )
  end
  response.body
end

#submit_feedback(label, value) ⇒ Object

Submit feedback on extraction label

Parameters:

  • label (String) —

    Extraction label to submit feedback on

  • value (String) —

    The new value for the given label



130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
 
# File 'lib/gini-api/document.rb', line 130

def submit_feedback(label, value)
  unless extractions.send(label.to_sym)
    raise Gini::Api::DocumentError.new("Unknown label #{label}: Not found")
  end
  response = @api.request(
    :put,
    "#{@_links[:extractions]}/#{label}",
    headers: { 'content-type' => @api.version_header[:accept] },
    body: { value: value }.to_json
  )
  unless response.status == 204
    raise Gini::Api::DocumentError.new(
      "Failed to submit feedback for label #{label} (code=#{response.status})",
      response
    )
  end
end

#successful? ⇒ Boolean

Was the document processed successfully?

Returns:

  • (Boolean) —

    true/false based on @progress



79
80
81
 
# File 'lib/gini-api/document.rb', line 79

def successful?
  @progress == 'COMPLETED'
end

#update(from_data = nil) ⇒ Object

Fetch document resource and populate instance variables

Parameters:

  • from_data (Hash) (defaults to: nil) —

    Ruby hash with doc data



27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/gini-api/document.rb', line 27

def update(from_data = nil)
  data = {}

  if from_data.nil?
    response = @api.request(:get, @location)
    unless response.status == 200
      raise Gini::Api::DocumentError.new(
        "Failed to fetch document data (code=#{response.status})",
        response
      )
    end
    data = response.parsed
  else
    data = from_data
  end

  data.each do |k, v|
    instance_variable_set("@#{k}", v)

    # We skip pages as it's rewritted by method pages()
    next if k == :pages

    self.class.send(:attr_reader, k)
  end
end