texttable 1.1.3 → 1.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: daf8fb1088ce9570617aea39c1cbbfa3b4b5819af7013e26c2e4680d1ec2ceb1
-  data.tar.gz: 343e38b5ff4ed2cb20076af22bcfaedb00fd70702dcc8654b4169736b87a4768
+  metadata.gz: 79e3c725b088a02d42d7096862e6087d090ba057b66e28297bf796597da59f16
+  data.tar.gz: 7a07a8a529036fe884d27290576c530b29c990a606866f09d8aa24526a7e1661
 SHA512:
-  metadata.gz: dfbd834c7999c0c666c39043f2529b31fa4077292aec331541744cb6fc22a841380914435b75c3462647896969d4b8eb6846e72ebdf6cb3878c761f0ac8005a1
-  data.tar.gz: 8b401aab5b7df8d12aa883a7c0932cf5e4713889d5a3f63aa59416cfb3e3c15eb9029257819e9e46aa62e7f23a547239cd2e21e52a35ab91990240c950bdd279
+  metadata.gz: 5ac2b2ece811fc38432cec0ad1315c211030b9effef406d7e715eff94243c5a4535ab6935dab23e0b926f23d70082daa55000253ad6d3a7597050498c6800cf8
+  data.tar.gz: 30345f8c02e6ec6169bb82e06ba19be129e6d74c4edb254ff308a989e2fe70cb6a6a0c422e8b87e16e6a5d8f991aad25950a4475a4a4f5cb36c6be4b3f415c48

data/.ruby-version

@@ -1 +1 @@
-2.5
+2.6.5

data/lib/texttable.rb

@@ -2,15 +2,41 @@ class TextTable
   attr_accessor :values, :rows
 
   class << self
-    def csv(src, sep=',', encoding: nil)
+    def csv(src, sep=',', encoding: nil, **kw)
       require 'csv'
       new CSV.read(src || ARGF, {
         col_sep:  sep,
         encoding: encoding ? encoding + ":UTF-8" : nil,
+        **kw
       }.compact)
     end
-    def tsv(*args); csv(args.shift, "\t", *args); end
-    def psv(*args); csv(args.shift, "|" , *args); end
+    def tsv(*args, **kw); csv(args.shift, "\t", *args, **kw); end
+    def psv(*args, **kw); csv(args.shift, "|" , *args, **kw); end
+
+    def add(*args)
+      new.add(*args)
+    end
+
+    def load(data, delim="\t", headers=true)
+      case data
+      when String
+        if data.include?(delim) # string
+          text = data
+        elsif File.exist?(data) # filename
+          text = File.read(data)
+        end
+      when File, ARGF
+        text = data.read
+      end
+
+      text or raise "unable to load #{data.inspect}"
+      rows = text.split(/\r?\n/).map {|line| line.split(delim).map {|part| part.strip}}
+      info = new
+      rows.shift.each_with_index {|col, i| info.index!(col || i) } if headers
+      info.rows = rows
+      info.row(0)
+      info
+    end
   end
 
   def initialize(*args)
@@ -21,9 +47,8 @@ class TextTable
     rows        = []       if !rows || !rows[0].is_a?(Array) || rows[0].empty?
     @cols = Hash.new {|h,k| h[k] = h.size}
     @rows = rows
-    @values = nil
-    @row = 0
-    cols.each {|col| index!(col) }
+    row(0)
+    cols.each_with_index {|col, i| index!(col || i) }
   end
 
   def index(field, auto=false)
@@ -42,6 +67,14 @@ class TextTable
     index(field, true)
   end
 
+  def lookup!(field)
+    @rows or raise "no rows defined"
+    index = index(field)
+    lookup = {}
+    @rows.each_with_index {|cols, i| lookup[cols[index]] = i}
+    lookup
+  end
+
   def convert_key(key)
     key.
       gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2').
@@ -86,6 +119,10 @@ class TextTable
     @rows.each_with_index {|_, row| yield(row(row)) }
   end
 
+  def each_pair
+    @cols.each {|col, pos| yield col, @values[pos] }
+  end
+
   def [](field, val=nil)
     index = index(field)
     value = vals[index] if index
@@ -114,13 +151,15 @@ class TextTable
     obj = [obj, *args] if args.size > 0
     @values = @rows[@row = @rows.size] = []
     case obj
-      when Hash  then obj.each {|k, v| @values[@cols[k]] = v}
+      when Hash  then obj.each {|k, v| @values[index(k.to_s, true)] = v }
       when Array then @values.replace(obj)
       else raise "unable to add #{obj.class} objects"
     end
     self
   end
 
+  alias :<< :add
+
   def show(*)
     self
   end
@@ -149,20 +188,27 @@ class TextTable
     self
   end
 
-  def csv(sep=',', encoding: nil)
+  def as_json(obj = defined?(ConfigHash) ? +{} : {})
+    (@rows || []).map {|r| r.each_with_index.inject(obj) {|h, (v, c)| h[fields[c]] = v; h }}
+  end
+
+  def csv(sep=',', encoding: nil, **kw)
     require 'csv'
-    csv = {}
-    csv[:encoding] = encoding + ":UTF-8" if encoding
-    csv[:col_sep ] = sep
-    csv = CSV.new($stdout, csv)
+    out = kw.key?(:out) ? (kw.delete(:out) || "").dup : nil
+    csv = CSV.new(out || $stdout, {
+      col_sep:     sep,
+      encoding:    encoding ? encoding + ":UTF-8" : nil,
+      quote_empty: false, #!# TODO: make this an option
+      **kw
+    })
     csv << @cols.keys
     @rows.each {|vals| csv << vals}
-    nil
+    out
   end
-  def tsv; csv("\t"); end
-  def psv; csv("|" ); end
+  def tsv(**kw); csv("\t", **kw); end
+  def psv(**kw); csv("|" , **kw); end
 
-  def sql(table='table', quote: false, timestamps: false, verb: 'insert')
+  def sql(table='table', quote: false, timestamps: false, verb: 'insert', out: nil)
     q = quote ? '`' : ''
     flip = @cols.invert
     @rows.each do |vals|
@@ -172,9 +218,12 @@ class TextTable
         list
       end
       list.push('created_at=now(), updated_at=now()') if timestamps
-      puts "#{verb} into #{q}#{table}#{q} set #{list * ', '};" if !list.empty?
+      if !list.empty?
+        line = "#{verb} into #{q}#{table}#{q} set #{list * ', '};"
+        out ? (out << line) : puts(line)
+      end
     end
-    nil
+    out
   end
 end
 

data/texttable.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name        = "texttable"
-  s.version     = "1.1.3"
+  s.version     = "1.1.9"
   s.author      = "Steve Shreeve"
   s.email       = "steve.shreeve@gmail.com"
   s.summary     = "An easy way to work with rows and columns as simple tables"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: texttable
 version: !ruby/object:Gem::Version
-  version: 1.1.3
+  version: 1.1.9
 platform: ruby
 authors:
 - Steve Shreeve
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-06-10 00:00:00.000000000 Z
+date: 2021-04-26 00:00:00.000000000 Z
 dependencies: []
 description: This gem will auto-size based on column widths.
 email: steve.shreeve@gmail.com
@@ -26,7 +26,7 @@ homepage: https://github.com/shreeve/texttable
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -41,8 +41,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
-signing_key: 
+rubygems_version: 3.2.16
+signing_key:
 specification_version: 4
 summary: An easy way to work with rows and columns as simple tables
 test_files: []