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 



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

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

#headers ⇒ Object 



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

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 



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

def row_count
  return 0 if csv_data.blank?

  require "csv"
  ::CSV.parse(csv_data, headers: true).length
rescue ::CSV::MalformedCSVError
  0
end