table_analysis 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 13a1fb68d047cfe87c585371d029bbcb3d3fb073c8107ef8fcba5000c80ddb88
-  data.tar.gz: 30c80555dd72d8fa583c677ddd9ac8e38c69116e10d9ee6c124edabc417ecff9
+  metadata.gz: e9bf918d52ccc29e84ca90d77bd5e0f46946d4fc960e22bd81d6552459fbf236
+  data.tar.gz: 0a07dd387dc71a76ee60b40d64e501eac515cbb3e0db6ec1a2de57f980694baa
 SHA512:
-  metadata.gz: 2eebc65a935e80aa0840484e147a9da852839eee74cdde0d96c086a8393858cce46859566e61595dba349a263791f285b3f3584bb2b08950035cc1f0a02a4676
-  data.tar.gz: 377ec082df94eaea375331b04aff2529aacd91fa34235a4baf29df5d6dd3a328e342b0da2e91051ec7d9150b578caa0b0d363023ba97c446500f3070435a8cb1
+  metadata.gz: 7adf1b44623efce822185d8b45b6d2f0cb9d8ae0da74d39a99fa74ddb621e454276096740cbd0353829e8d7ef2744a03d48bfc18e7b974720ac2e917ae23a578
+  data.tar.gz: 7683f2c480e3d956a9c61901f3cef32922323c6c080c2291178222a08f4df430436eae50123e2ef604f3f3ffdec6d871954373ca3838b619a65021ec7bb74e8f

data/demo.rb

@@ -2,4 +2,4 @@ require './lib/table_analysis.rb'
 
 doc = File.read('file/demo2.html')
 
-TableAnalysis::Main.generator(doc, 1)
+TableAnalysis::Main.generator(doc, 2, 1)

data/file/demo2.html

@@ -10,6 +10,14 @@
 <col style="width:139.5pt;min-width:139.5pt;"/>
 <col style="width:105pt;min-width:105pt;"/>
 </colgroup>
+<tr style="height:21.8pt">
+  <td valign="middle" style="vertical-align:middle;border-top-style:solid;border-top-width:0.75pt;border-bottom-style:solid;border-bottom-width:0.75pt;border-left-style:solid;border-left-width:0.75pt;border-right-style:solid;border-right-width:0.75pt;">
+    <p class="smt_tblC">1</p>
+  </td>
+  <td valign="middle" style="vertical-align:middle;border-top-style:solid;border-top-width:0.75pt;border-bottom-style:solid;border-bottom-width:0.75pt;border-left-style:solid;border-left-width:0.75pt;border-right-style:solid;border-right-width:0.75pt;">
+    <p class="smt_tblC">2</p>
+  </td>
+</tr>
 <tr style="height:21.8pt">
 <td valign="middle" style="vertical-align:middle;border-top-style:solid;border-top-width:0.75pt;border-bottom-style:solid;border-bottom-width:0.75pt;border-left-style:solid;border-left-width:0.75pt;border-right-style:solid;border-right-width:0.75pt;">
 <p class="smt_tblC">契約会社名</p>

data/lib/table_analysis/version.rb

@@ -1,3 +1,3 @@
 module TableAnalysis
-  VERSION = "0.1.2"
+  VERSION = '0.1.4'.freeze
 end

data/lib/table_analysis.rb

@@ -3,23 +3,23 @@ require_relative 'table_analysis/version'
 require 'nokogiri'
 module TableAnalysis
   class Main
-    def self.generator(html, *selected_cols)
+    def self.generator(html, selected_row, *selected_cols)
       selected_cols = selected_cols.flatten 
       doc = Nokogiri::HTML(html, nil, 'utf-8')
       # 多个table,仅处理第一个
       table = doc.xpath('//table')[0]
-      return false if table.size = 0
+      return false if table.nil?
       header_content_tds = []
       body_content_tds = []
       body_tr_size = 0
       table.xpath('./tr').each_with_index do |tr, tr_index|
-        if tr_index == 0
+        if tr_index == selected_row.to_i - 1
           tr.xpath('./td').each do |td|
             header_name = td.content
             colspan = td.attribute('colspan')&.value
             header_content_tds << [header_name, colspan]
           end
-        else
+        elsif tr_index >= selected_row.to_i
           body_tr_size += 1
           tr.xpath('./td').each_with_index do |td, td_index|
             rowspan = td.attribute('rowspan')&.value 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: table_analysis
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - mico_xiaozhen@sina.com