creek 2.5.1 → 2.5.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 39ef59b96aa1733768b24a21c8b393247fbdec6d2f9919da92e1329f352524da
-  data.tar.gz: 9b460a1530f18ceaf6d1884b0ac2040f33a88887895fbbe7fea39fa855d8b01c
+  metadata.gz: 1b9680127cd771d8b0840567db333a2a32369952bbeb09eee7364a9dbcdd2284
+  data.tar.gz: e39694d00b142e542457cc585d9b0e027386a8a3fcdb39b0cfe670f6e8d57b81
 SHA512:
-  metadata.gz: 01561d9355ea7f82d1523c887b8e0556f031bf579c36f0ada04b2a9ada5b5c19e10378098764604ae5bf7b18c371d58b787704f1ad84ccafbb93a9905b1e20ec
-  data.tar.gz: 5b9315c34588b2dea639252f188158bcbe6fef1519fd73f6337f7d995be7ac28ba62d35e7cf9ea64ba941c2d6e6b0b3091ca69cded65f06cd943f156998bf8c2
+  metadata.gz: ffa323e9d0f5a9c7976750d4c802a10ca39517b7f2ed388c801c5109aa331a109128dfdfcc5a039e919a2f3e28b9099b9839face7594b1667ccf76cb0bef1bca
+  data.tar.gz: 6c5d6dc7b921d504490502b9e2c37def7112d23bb4a8f6d4a3a7e28d68bc6f297b0ff2864725c2186f302106c40543f4369a54c6884d3a3d908aaf0355539c6f

data/.gitignore

@@ -15,3 +15,6 @@ spec/reports
 test/tmp
 test/version_tmp
 tmp
+
+# Mac finder artifacts
+.DS_Store

data/README.md

@@ -100,6 +100,14 @@ remote_url = 'http://dev-builds.libreoffice.org/tmp/test.xlsx'
 Creek::Book.new remote_url, remote: true
 ```
 
+## Mapping cells with header names
+By default, Creek will map cell names with letter and number(A1, B3 and etc). To be able to get cell values by header column name use ***with_headers*** (can be used only with ***#simple_rows*** method!!!) during creation *(Note: header column is first string of sheet)*
+
+```ruby
+creek = Creek::Book.new file.path, with_headers: true
+```
+
+
 ## Contributing
 
 Contributions are welcomed. You can fork a repository, add your code changes to the forked branch, ensure all existing unit tests pass, create new unit tests which cover your new changes and finally create a pull request.

data/lib/creek.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'creek/version'
 require 'creek/book'
 require 'creek/styles/constants'

data/lib/creek/book.rb

@@ -4,12 +4,11 @@ require 'date'
 require 'open-uri'
 
 module Creek
-
   class Creek::Book
-
     attr_reader :files,
                 :sheets,
-                :shared_strings
+                :shared_strings,
+                :with_headers
 
     DATE_1900 = Date.new(1899, 12, 30).freeze
     DATE_1904 = Date.new(1904, 1, 1).freeze
@@ -23,6 +22,7 @@ module Creek
       path = download_file(path) if options[:remote]
       @files = Zip::File.open(path)
       @shared_strings = SharedStrings.new(self)
+      @with_headers = options.fetch(:with_headers, false)
     end
 
     def sheets
@@ -41,7 +41,17 @@ module Creek
       rels = Nokogiri::XML::Document.parse(rels_doc).css("Relationship")
       @sheets = xml.css(cssPrefix+'sheet').map do |sheet|
         sheetfile = rels.find { |el| sheet.attr("r:id") == el.attr("Id") }.attr("Target")
-        Sheet.new(self, sheet.attr("name"), sheet.attr("sheetid"),  sheet.attr("state"), sheet.attr("visible"), sheet.attr("r:id"), sheetfile)
+        sheet = Sheet.new(
+          self,
+          sheet.attr("name"),
+          sheet.attr("sheetid"),
+          sheet.attr("state"),
+          sheet.attr("visible"),
+          sheet.attr("r:id"),
+          sheetfile
+        )
+        sheet.with_headers = with_headers
+        sheet
       end
     end
 

data/lib/creek/sheet.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'zip/filesystem'
 require 'nokogiri'
 
@@ -5,16 +7,19 @@ module Creek
   class Creek::Sheet
     include Creek::Utils
 
+    HEADERS_ROW_NUMBER = '1'
+
+    attr_accessor :with_headers
     attr_reader :book,
                 :name,
                 :sheetid,
                 :state,
                 :visible,
                 :rid,
-                :index
-
+                :index,
+                :headers
 
-    def initialize book, name, sheetid, state, visible, rid, sheetfile
+    def initialize(book, name, sheetid, state, visible, rid, sheetfile)
       @book = book
       @name = name
       @sheetid = sheetid
@@ -46,7 +51,6 @@ module Creek
       @drawing.images_at(cell) if @images_present
     end
 
-
     ##
     # Provides an Enumerator that returns a hash representing each row.
     # The key of the hash is the column ID and the value is the value of the cell.
@@ -89,35 +93,37 @@ module Creek
         closer = Nokogiri::XML::Reader::TYPE_END_ELEMENT
         Enumerator.new do |y|
           row, cells, cell = nil, {}, nil
-          cell_type  = nil
+          cell_type = nil
           cell_style_idx = nil
           @book.files.file.open(path) do |xml|
             Nokogiri::XML::Reader.from_io(xml).each do |node|
-              if (node.name.eql? 'row') and (node.node_type.eql? opener)
+              if node.name == 'row' && node.node_type == opener
                 row = node.attributes
-                row['cells'] = Hash.new
-                cells = Hash.new
+                row['cells'] = {}
+                cells = {}
                 y << (include_meta_data ? row : cells) if node.self_closing?
-              elsif (node.name.eql? 'row') and (node.node_type.eql? closer)
+              elsif node.name == 'row' && node.node_type == closer
                 processed_cells = fill_in_empty_cells(cells, row['r'], cell, use_simple_rows_format)
+                @headers = processed_cells if row['r'] == HEADERS_ROW_NUMBER
 
                 if @images_present
                   processed_cells.each do |cell_name, cell_value|
                     next unless cell_value.nil?
+
                     processed_cells[cell_name] = images_at(cell_name)
                   end
                 end
 
                 row['cells'] = processed_cells
                 y << (include_meta_data ? row : processed_cells)
-              elsif (node.name.eql? 'c') and (node.node_type.eql? opener)
+              elsif node.name == 'c' && node.node_type == opener
                 cell_type      = node.attributes['t']
                 cell_style_idx = node.attributes['s']
                 cell           = node.attributes['r']
-              elsif (['v', 't'].include? node.name) and (node.node_type.eql? opener)
+              elsif %w[v t].include?(node.name) && node.node_type == opener
                 unless cell.nil?
                   node.read
-                  cells[(use_simple_rows_format ? cell.tr("0-9", "") : cell)] = convert(node.value, cell_type, cell_style_idx)
+                  cells[cell] = convert(node.value, cell_type, cell_style_idx)
                 end
               end
             end
@@ -142,15 +148,13 @@ module Creek
     # The unzipped XML file does not contain any node for empty cells.
     # Empty cells are being padded in using this function
     def fill_in_empty_cells(cells, row_number, last_col, use_simple_rows_format)
-      new_cells = Hash.new
+      new_cells = {}
+      return new_cells if cells.empty?
 
-      unless cells.empty?
-        last_col = last_col.gsub(row_number, '')
-
-        ("A"..last_col).to_a.each do |column|
-          id = use_simple_rows_format ? "#{column}" : "#{column}#{row_number}"
-          new_cells[id] = cells[id]
-        end
+      last_col = last_col.gsub(row_number, '')
+      ('A'..last_col).to_a.each do |column|
+        id = cell_id(column, use_simple_rows_format, row_number)
+        new_cells[id] = cells["#{column}#{row_number}"]
       end
 
       new_cells
@@ -172,5 +176,11 @@ module Creek
       sheet_rels_filepath = expand_to_rels_path(sheet_filepath)
       parse_xml(sheet_rels_filepath).css("Relationship[@Id='#{drawing_rid}']").first.attributes['Target'].value
     end
+
+    def cell_id(column, use_simple_rows_format, row_number = '')
+      return "#{column}#{row_number}" unless use_simple_rows_format
+
+      with_headers && headers ? headers[column] : column
+    end
   end
 end

data/lib/creek/version.rb

@@ -1,3 +1,3 @@
 module Creek
-  VERSION = "2.5.1"
+  VERSION = "2.5.2"
 end

data/spec/sheet_spec.rb

@@ -1,26 +1,26 @@
+# frozen_string_literal: true
+
 require './spec/spec_helper'
 
 describe 'sheet' do
   let(:book_with_images) { Creek::Book.new('spec/fixtures/sample-with-images.xlsx') }
-  let(:book_no_images) { Creek::Book.new('spec/fixtures/sample.xlsx') }
   let(:sheetfile) { 'worksheets/sheet1.xml' }
   let(:sheet_with_images) { Creek::Sheet.new(book_with_images, 'Sheet 1', 1, '', '', '1', sheetfile) }
-  let(:sheet_no_images) { Creek::Sheet.new(book_no_images, 'Sheet 1', 1, '', '', '1', sheetfile) }
 
   def load_cell(rows, cell_name)
-    cell = rows.find { |row| !row[cell_name].nil? }
+    cell = rows.find { |row| row[cell_name] }
     cell[cell_name] if cell
   end
 
   context 'escaped ampersand' do
     let(:book_escaped) { Creek::Book.new('spec/fixtures/escaped.xlsx') }
     it 'does NOT escape ampersand' do
-      expect(book_escaped.sheets[0].rows.to_enum.map(&:values)).to eq([["abc", "def"], ["ghi", "j&k"]])
+      expect(book_escaped.sheets[0].rows.to_enum.map(&:values)).to eq([%w[abc def], %w[ghi j&k]])
     end
 
     let(:book_escaped2) { Creek::Book.new('spec/fixtures/escaped2.xlsx') }
     it 'does escape ampersand' do
-      expect(book_escaped2.sheets[0].rows.to_enum.map(&:values)).to eq([["abc", "def"], ["ghi", "j&k"]])
+      expect(book_escaped2.sheets[0].rows.to_enum.map(&:values)).to eq([%w[abc def], %w[ghi j&k]])
     end
   end
 
@@ -66,6 +66,9 @@ describe 'sheet' do
     end
 
     context 'with excel without images' do
+      let(:book_no_images) { Creek::Book.new('spec/fixtures/sample.xlsx') }
+      let(:sheet_no_images) { Creek::Sheet.new(book_no_images, 'Sheet 1', 1, '', '', '1', sheetfile) }
+
       it 'does not break on with_images' do
         rows = sheet_no_images.with_images.rows.map { |r| r }
         expect(load_cell(rows, 'A10')).to eq(0.15)
@@ -94,4 +97,28 @@ describe 'sheet' do
       expect(image).to eq(nil)
     end
   end
+
+  describe '#simple_rows' do
+    let(:book_with_headers) { Creek::Book.new('spec/fixtures/sample-with-headers.xlsx') }
+    let(:sheet) { Creek::Sheet.new(book_with_headers, 'Sheet 1', 1, '', '', '1', sheetfile) }
+
+    subject { sheet.simple_rows.to_a[1] }
+
+    it 'returns values by letters' do
+      expect(subject['A']).to eq 'value1'
+      expect(subject['B']).to eq 'value2'
+    end
+
+    context 'when enable with_headers property' do
+      before { sheet.with_headers = true }
+
+      subject { sheet.simple_rows.to_a[1] }
+
+      it 'returns values by headers name' do
+        expect(subject['HeaderA']).to eq 'value1'
+        expect(subject['HeaderB']).to eq 'value2'
+        expect(subject['HeaderC']).to eq 'value3'
+      end
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: creek
 version: !ruby/object:Gem::Version
-  version: 2.5.1
+  version: 2.5.2
 platform: ruby
 authors:
 - pythonicrubyist
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-03-08 00:00:00.000000000 Z
+date: 2020-04-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -127,6 +127,7 @@ files:
 - spec/fixtures/invalid.xls
 - spec/fixtures/large_numbers.xlsx
 - spec/fixtures/sample-as-zip.zip
+- spec/fixtures/sample-with-headers.xlsx
 - spec/fixtures/sample-with-images.xlsx
 - spec/fixtures/sample.xlsx
 - spec/fixtures/sample_dates.xlsx
@@ -161,7 +162,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubygems_version: 3.0.8
 signing_key: 
 specification_version: 4
 summary: A Ruby gem for parsing large Excel(xlsx and xlsm) files.
@@ -173,6 +174,7 @@ test_files:
 - spec/fixtures/invalid.xls
 - spec/fixtures/large_numbers.xlsx
 - spec/fixtures/sample-as-zip.zip
+- spec/fixtures/sample-with-headers.xlsx
 - spec/fixtures/sample-with-images.xlsx
 - spec/fixtures/sample.xlsx
 - spec/fixtures/sample_dates.xlsx