我创建了以下脚本来擦除mysql数据库(并重置每个表的主键)。我想知道我应该如何重构它,以及如何从ActiveSupport中引入复数形式。
代码:
MODEL_DIR = File.expand_path("app/models")
Dir.chdir(MODEL_DIR)
files = Dir.glob(File.join("**", "*.rb"))
files.map! do |file|
file[0..-4] + "s"
end
print "This will WIPE your database. Continue? (y/n): "
if $stdin.gets.chomp.downcase == "y"
files.each do |f|
puts "Wiping #{f}.."
ActiveRecord::Base.connection.execute "TRUNCATE TABLE #{f};"
end
else
puts "Terminating script..."
end
我的逻辑是,models目录中没有.rb和复数的每个文件都表示数据库中的一个表,所以这就是我获得与该应用程序相关的表列表的方式。
我用以下命令运行它:rails runner script/cleandb.rb
应该如何重构它,以及如何引入复数?
基于Rails约定,您应该能够用以下代码以更安全的方式实现这一点(例如,如果您的模型有特定的表名前缀或表名):
print "This will WIPE your database. Continue? (y/n): "
if $stdin.gets.chomp.downcase == "y"
# iterate over all model definition files
Dir["#{Rails.root}/app/models/**/*.rb"].map do |model_filename|
# get the file base_name
model_file_basename = File.basename(model_filename,File.extname(model_filename))
# get the model class from the file basename
model_class = model_file_basename.camelize.constantize
# ask the model (ActiveRecord::Base subclass) to give you its table_name
table_name = model_class.table_name
# wipe the table
puts "Wiping table #{table_name}.."
ActiveRecord::Base.connection.execute "TRUNCATE TABLE #{table_name};"
end
else
puts "Terminating script..."
end
请参阅表名称文档:http://apidock.com/rails/ActiveRecord/ModelSchema/ClassMethods/table_name