class Company::Mapping::CompanyCorpus

Public Class Methods

new(path=nil) click to toggle source
Calls superclass method
# File lib/company/mapping/document_utils/company_corpus.rb, line 4
def initialize(path=nil)
  super()
  import_csv path if path
end

Public Instance Methods

add(id, company_name, aliases=[]) click to toggle source
# File lib/company/mapping/document_utils/company_corpus.rb, line 9
def add id, company_name, aliases=[]
  push doc(company_name, id)
  aliases.each_with_index do |company_alias, i|
    push doc(company_alias, "#{id}_#{i}")
  end
end
import_csv(path) click to toggle source

build a corpus from a csv file

# File lib/company/mapping/document_utils/company_corpus.rb, line 17
def import_csv path
  CSV.foreach(path) do |row|
    array = row.first.split(";")
    add array.first, array[1], array[2..-1]
  end
  @corpus
end

Private Instance Methods

doc(content, id) click to toggle source
# File lib/company/mapping/document_utils/company_corpus.rb, line 27
def doc content, id
  alias_doc = TextDocument.new
  alias_doc.contents = content.gsub(",", "").gsub(".", "")
  alias_doc.id = id
  alias_doc
end