Class: CompletionKit::Dataset

Inherits:
ApplicationRecord show all
Includes:
Taggable
Defined in:
app/models/completion_kit/dataset.rb

Constant Summary

Constants inherited from ApplicationRecord

ApplicationRecord::TenantScopedUniquenessValidator

Instance Method Summary collapse

Methods included from Taggable

#tag_names, #tag_names=

Instance Method Details

#as_json(options = {}) ⇒ Object 



18
19
20
21
22
23
24
 
# File 'app/models/completion_kit/dataset.rb', line 18

def as_json(options = {})
  {
    id: id, name: name, csv_data: csv_data,
    created_at: created_at, updated_at: updated_at,
    tags: tags.as_json
  }
end

#csv_data_within_size_limit ⇒ Object 



11
12
13
14
15
16
 
# File 'app/models/completion_kit/dataset.rb', line 11

def csv_data_within_size_limit
  return if csv_data.blank?
  return if csv_data.bytesize <= CompletionKit.config.max_upload_bytes

  errors.add(:csv_data, "is too large (limit #{CompletionKit.config.max_upload_bytes / (1024 * 1024)} MB)")
end

#headers ⇒ Object 



39
40
41
42
43
44
45
46
 
# File 'app/models/completion_kit/dataset.rb', line 39

def headers
  return [] if csv_data.blank?

  require "csv"
  ::CSV.parse(csv_data.lines.first.to_s).first.to_a.map(&:to_s).map(&:strip)
rescue ::CSV::MalformedCSVError
  []
end

#row_count ⇒ Object 



26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'app/models/completion_kit/dataset.rb', line 26

def row_count
  return 0 if csv_data.blank?

  require "csv"
  @row_count ||= begin
    count = 0
    ::CSV.new(csv_data, headers: true).each { count += 1 }
    count
  rescue ::CSV::MalformedCSVError
    0
  end
end