class Aio::Base::Toolkit::Table
Prints text in a tablized format. Pretty lame at the moment, but whatever. 经过修改增加了对中文字体格式的支持 by Elin
Public Class Methods
Initializes a text table instance using the supplied properties. The Table
class supports the following hash attributes:
Header
The string to display as a heading above the table. If none is specified, no header will be displayed.
HeaderIndent
The amount of space to indent the header. The default is zero.
Columns
The array of columns that will exist within the table.
Rows
The array of rows that will exist.
Width
The maximum width of the table in characters.
Indent
The number of characters to indent the table.
CellPad
The number of characters to put between each horizontal cell.
Prefix
The text to prefix before the table.
Postfix
The text to affix to the end of the table.
Sortindex
The column to sort the table on, -1 disables sorting.
# File lib/aio/base/toolkit/table.rb, line 59 def initialize(opts = {}) self.header = opts['Header'] self.headeri = opts['HeaderIndent'] || 0 self.columns = opts['Columns'] || [] # updated below if we got a "Rows" option self.rows = [] self.width = opts['Width'] || 80 self.indent = opts['Indent'] || 0 self.cellpad = opts['CellPad'] || 2 self.prefix = opts['Prefix'] || '' self.postfix = opts['Postfix'] || '' self.colprops = [] self.scterm = /#{opts['SearchTerm']}/mi if opts['SearchTerm'] self.sort_index = opts['SortIndex'] || 0 self.sort_order = opts['SortOrder'] || :forward # Default column properties self.columns.length.times { |idx| self.colprops[idx] = {} self.colprops[idx]['MaxWidth'] = self.columns[idx].length + ch_num(self.columns[idx]) } # ensure all our internal state gets updated with the given rows by # using add_row instead of just adding them to self.rows. See #3825. opts['Rows'].each { |row| add_row(row) } if opts['Rows'] # Merge in options if (opts['ColProps']) opts['ColProps'].each_key { |col| idx = self.columns.index(col) if (idx) self.colprops[idx].merge!(opts['ColProps'][col]) end } end end
Build table from CSV dump
# File lib/aio/base/toolkit/table.rb, line 262 def self.new_from_csv(csv) # Read in or keep data, get CSV or die if csv.is_a?(String) csv = File.file?(csv) ? CSV.read(csv) : CSV.parse(csv) end # Adjust for skew if csv.first == ["Keys", "Values"] csv.shift # drop marker cols = [] rows = [] csv.each do |row| cols << row.shift rows << row end tbl = self.new('Columns' => cols) rows.in_groups_of(cols.count) {|r| tbl << r.flatten} else tbl = self.new('Columns' => csv.shift) while !csv.empty? do tbl << csv.shift end end return tbl end
Public Instance Methods
Adds a row using the supplied fields.
# File lib/aio/base/toolkit/table.rb, line 163 def <<(fields) add_row(fields) end
# File lib/aio/base/toolkit/table.rb, line 287 def [](*col_names) tbl = self.class.new('Indent' => self.indent, 'Header' => self.header, 'Columns' => col_names) indexes = [] col_names.each do |col_name| index = self.columns.index(col_name) raise RuntimeError, "Invalid column name #{col_name}" if index.nil? indexes << index end self.rows.each do |old_row| new_row = [] indexes.map {|i| new_row << old_row[i]} tbl << new_row end return tbl end
Adds a horizontal line.
# File lib/aio/base/toolkit/table.rb, line 226 def add_hr rows << '__hr__' end
Adds a row with the supplied fields.
# File lib/aio/base/toolkit/table.rb, line 170 def add_row(fields = []) if fields.length != self.columns.length raise RuntimeError, 'Invalid number of columns!' end fields.each_with_index { |field, idx| # Remove whitespace and ensure String format field = field.to_s.strip field_length = field.to_s.length + ch_num(field.to_s) if (colprops[idx]['MaxWidth'] < field_length) old = colprops[idx]['MaxWidth'] colprops[idx]['MaxWidth'] = field_length end } rows << fields end
Returns new sub-table with headers and rows maching column names submitted
Flips table 90 degrees left
# File lib/aio/base/toolkit/table.rb, line 236 def drop_left tbl = self.class.new( 'Columns' => Array.new(self.rows.count+1,' '), 'Header' => self.header, 'Indent' => self.indent) (self.columns.count+1).times do |ti| row = self.rows.map {|r| r[ti]}.unshift(self.columns[ti]).flatten # insert our col|row break. kind of hackish row[1] = "| #{row[1]}" unless row.all? {|e| e.nil? || e.empty?} tbl << row end return tbl end
# File lib/aio/base/toolkit/table.rb, line 187 def ip_cmp(a, b) begin a = ::IPAddr.new(a.to_s) b = ::IPAddr.new(b.to_s) return 1 if a.ipv6? && b.ipv4? return -1 if a.ipv4? && b.ipv6? a <=> b rescue ::IPAddr::Error nil end end
Prints the contents of the table.
# File lib/aio/base/toolkit/table.rb, line 156 def print puts to_s end
Sorts the rows based on the supplied index of sub-arrays If the supplied index is an IPv4 address, handle it differently, but avoid actually resolving domain names.
# File lib/aio/base/toolkit/table.rb, line 204 def sort_rows(index = sort_index, order = sort_order) return if index == -1 return unless rows rows.sort! do |a,b| if a[index].nil? cmp = -1 elsif b[index].nil? cmp = 1 elsif a[index] =~ /^[0-9]+$/ and b[index] =~ /^[0-9]+$/ cmp = a[index].to_i <=> b[index].to_i elsif (cmp = ip_cmp(a[index], b[index])) != nil else cmp = a[index] <=> b[index] # assumes otherwise comparable. end cmp ||= 0 order == :forward ? cmp : -cmp end end
Converts table contents to a csv
# File lib/aio/base/toolkit/table.rb, line 126 def to_csv str = '' str << ( columns.join(",") + "\n" ) rows.each { |row| next if is_hr(row) || !row_visible(row) str << ( row.map{|x| x = x.to_s x.gsub(/[\r\n]/, ' ').gsub(/\s+/, ' ').gsub('"', '""') }.map{|x| "\"#{x}\"" }.join(",") + "\n" ) } str end
Converts table contents to a string.
# File lib/aio/base/toolkit/table.rb, line 103 def to_s str = prefix.dup str << header_to_s || '' str << columns_to_s || '' str << hr_to_s || '' sort_rows rows.each { |row| if (is_hr(row)) str << hr_to_s else str << row_to_s(row) if row_visible(row) end } str << postfix return str end
# File lib/aio/base/toolkit/table.rb, line 250 def valid_ip?(value) begin ::IPAddr.new value true rescue ::IPAddr::Error false end end
Protected Instance Methods
# File lib/aio/base/toolkit/table.rb, line 431 def ch_num(str) s = str.dup s.size-s.gsub(/[\u4e00-\u9fa5]/,'').size end
Returns if a row should be visible or not
# File lib/aio/base/toolkit/table.rb, line 322 def row_visible(row) return true if self.scterm.nil? row_to_s(row).match(self.scterm) end