Sha256: 0a88d5dd5804d1e840a2d2fbf8804abab6af25de16b3db92152a7af3f686c5bc

Contents?: true

Size: 1.32 KB

Versions: 9

Compression:

Stored size: 1.32 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)
      klass = Class.new(ActiveRecord::Base) do
        def self.model_name
          ActiveModel::Name.new(self, nil, table_name)
        end
        self.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

9 entries across 9 versions & 1 rubygems

Version Path
rails_db-1.3.4 lib/rails_db/table.rb
rails_db-1.3.3 lib/rails_db/table.rb
rails_db-1.3.2 lib/rails_db/table.rb
rails_db-1.3.1 lib/rails_db/table.rb
rails_db-1.3 lib/rails_db/table.rb
rails_db-1.1.1 lib/rails_db/table.rb
rails_db-1.1 lib/rails_db/table.rb
rails_db-1.0 lib/rails_db/table.rb
rails_db-0.9.9 lib/rails_db/table.rb