Class: DynamoRecord::TaskHelpers::Cleanup

Inherits:
Object
  • Object
show all
Defined in:
lib/dynamo-record/task_helpers/cleanup.rb

Class Method Summary collapse

Class Method Details

.delete_by_class(filename) ⇒ Object 



11
12
13
14
15
16
 
# File 'lib/dynamo-record/task_helpers/cleanup.rb', line 11

def self.delete_by_class(filename)
  klass = File.basename(filename, '.rb').camelize.constantize
  return unless klass.included_modules.include? DynamoRecord::Model
  Rails.logger.info "Deleting all items in table: #{klass}"
  klass.scan.each(&:delete!)
end

.run ⇒ Object 



4
5
6
7
8
9
 
# File 'lib/dynamo-record/task_helpers/cleanup.rb', line 4

def self.run
  raise 'Task not available on production' if Rails.env.production?
  Dir[Rails.root.join('app/models/*.rb').to_s].each do |filename|
    delete_by_class(filename)
  end
end