Sha256: 07dbd02cdaf5fbd78ead3aa71ef4b8f6e235962d81922f605f62159cceaeda32
Contents?: true
Size: 1.56 KB
Versions: 15
Compression:
Stored size: 1.56 KB
Contents
require 'active_model'
module RailsDb
class Table
include Connection
attr_reader :name, :data
delegate :count, to: :data
delegate :paginate, to: :data
delegate :limit, to: :data
delegate :order, to: :data
delegate :asc, to: :data
delegate :desc, to: :data
def initialize(table_name)
throw 'Access Denied' unless RailsDb::Database.accessible_tables.include?(table_name)
@name = table_name
@data = RailsDb::TableData.new(self)
self
end
def to_csv
CSV.generate do |csv|
csv << column_names
data.data.rows.each do |row|
csv << row
end
end
end
def indexes
RailsDb::Database.indexes(name)
end
def truncate
RailsDb::Database.truncate(name)
end
def primary_key
RailsDb::Database.primary_key(name)
end
def delete(id)
RailsDb::Database.delete(name, primary_key, id)
end
def create_model(table_name, &block)
begin
klass = Class.new(ActiveRecord::Base) do
def self.model_name
ActiveModel::Name.new(self, nil, table_name)
end
self.table_name = table_name
self.inheritance_column = nil
end
klass.count # verify that it works, if not load other, hack
rescue
klass = ActiveRecord::Base.descendants.detect { |c| c.table_name == table_name }
end
klass.class_eval(&block) if block_given?
klass
end
def as_model
@model ||= create_model(name)
end
end # module
end
Version data entries
15 entries across 15 versions & 1 rubygems