Module: PryHelper::Commands::VD

Defined in:
lib/pry-helper/commands/vd.rb

Class Method Summary collapse

Class Method Details

.get_table_name(name) ⇒ Object 



6
7
8
9
10
11
12
13
14
 
# File 'lib/pry-helper/commands/vd.rb', line 6

def get_table_name(name)
  name = name.to_s
  return name if name =~ /^[a-z]/
  if Object.const_defined?(name)
    klass = Object.const_get(name)
    return klass.table_name if klass < ActiveRecord::Base
  end
  name
end

.table_info(table_name) ⇒ Object 



22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/pry-helper/commands/vd.rb', line 22

def table_info(table_name)
  t = []
  t << ['PK', 'Name', 'SQL Type', 'Ruby Type', 'Limit', 'Precision', 'Scale', 'Default', 'Nullable', 'Comment']
  connection = ::ActiveRecord::Base.connection
  connection.columns(table_name).each do |column|
    pk = if [connection.primary_key(table_name)].flatten.include?(column.name)
           'Y'
         else
           ''
         end
    t << [pk, column.name, column.sql_type,
          column.sql_type_metadata.type, column.sql_type_metadata.limit || '',
          column.sql_type_metadata.precision || '', column.sql_type_metadata.scale || '', column.default || '',
          column.null, column.comment || '']
  end
  t
end

.table_info_vd(table_name) ⇒ Object 



16
17
18
19
20
 
# File 'lib/pry-helper/commands/vd.rb', line 16

def table_info_vd(table_name)
  PryHelper::VD.new do |vd|
    table_info(table_name).each { |row| vd << row }
  end
end