Class: TableImporter::Excel

Inherits:
Source
  • Object
show all
Defined in:
lib/table_importer/excel.rb

Constant Summary

Constants inherited from Source

Source::SEPARATORS

Instance Method Summary collapse

Methods inherited from Source

#clean_chunks, #default_headers, #get_column_separator, #get_record_separator, #get_sep_count, #sort_separators

Constructor Details

#initialize(data) ⇒ Excel

Returns a new instance of Excel.



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/table_importer/excel.rb', line 5

def initialize(data)
  begin
    @type = File.extname(data[:content]) == ".xls" ? "xls" : "xlsx"
    @file_path = data[:content].path
    @headers_present = data[:headers_present]
    @file = get_file
    @compulsory_headers = data[:compulsory_headers]
    @delete_empty_columns = (File.size(@file_path) < 100000)
    @mapping = !data[:user_headers].blank? ? data[:user_headers] : data[:headers]
    raise TableImporter::EmptyFileImportError.new if !@file.first_row
    if !data[:headers].nil?
      @headers = data[:headers]
    else
      @headers = @headers_present ? @file.row(1).map.with_index { |header, index| header.present? ? header.to_sym : "column_#{index}"} : default_headers
    end
  rescue NoMethodError
    raise TableImporter::HeaderMismatchError.new
  end
end

Instance Method Details

#convert_headers ⇒ Object 



69
70
71
72
73
74
75
76
77
78
79
 
# File 'lib/table_importer/excel.rb', line 69

def convert_headers
  new_headers = @headers_present ? @file.row(1) : default_headers
  new_headers = default_headers(new_headers.count)
  return new_headers unless @mapping
  @mapping.each do |key, value|
    if value.to_i.to_s == value.to_s
      new_headers[value.to_i] = key.to_sym
    end
  end
  new_headers
end

#get_chunks(chunk_size) ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
92
 
# File 'lib/table_importer/excel.rb', line 81

def get_chunks(chunk_size)
  @headers = convert_headers
  @last_row ||= @file.last_row
  chunks = []
  start_point = @headers_present ? 2 : 1
  while chunks.count <= @last_row/chunk_size
    chunks << get_lines(start_point, chunk_size)
    start_point += chunk_size
  end
  chunks.last << Hash[@headers.zip(@file.row(@last_row))]
  clean_chunks(chunks, @compulsory_headers)
end

#get_file ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/table_importer/excel.rb', line 29

def get_file
  begin
    if @type == "xls"
      Roo::Excel.new(@file_path).sheet(0)
    elsif @type == "xlsx"
      Roo::Excelx.new(@file_path).sheet(0)
    end
  rescue TypeError
    raise TableImporter::IncorrectFileError.new
  end
end

#get_headers ⇒ Object 



25
26
27
 
# File 'lib/table_importer/excel.rb', line 25

def get_headers
  @headers
end

#get_lines(start, number_of_lines) ⇒ Object 



59
60
61
62
63
64
65
66
67
 
# File 'lib/table_importer/excel.rb', line 59

def get_lines(start, number_of_lines)
  @last_row ||= @file.last_row
  finish = [@last_row, start + number_of_lines].min
  mapped_lines = []
  (start...finish).each do |row_number|
    mapped_lines << Hash[@headers.zip(@file.row(row_number))]
  end
  mapped_lines
end

#get_preview_lines(start_point = 0, end_point = 10) ⇒ Object 



45
46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/table_importer/excel.rb', line 45

def get_preview_lines(start_point = 0, end_point = 10)
  begin
    @headers = @mapping.present? && @mapping != false ? convert_headers : @headers
    lines = clean_chunks([get_lines(start_point, end_point)], @compulsory_headers)[0][:lines]
    if lines.first.nil?
      get_preview_lines(start_point+10, end_point+10)
    else
      lines[0..8]
    end
  rescue SystemStackError, NoMethodError
    raise TableImporter::EmptyFileImportError.new
  end
end

#get_type ⇒ Object 



41
42
43
 
# File 'lib/table_importer/excel.rb', line 41

def get_type
  "xls"
end