rails console で m select * from users; とかできるようにするスニペット

.pryrc

class Pry
  class Command::MySQLQueryCommand < Pry::ClassCommand
    match(/m\s+(.*)/)
    description "execute mysql command line"
    banner <<-'BANNER'
      Usage: m SQLCOMMAND
      ex) m select * from users;
    BANNER

    @@config = YAML.load_file("config/database.yml")
    def process(cmd)
      env_config = @@config[ Rails.env ]
      config_options = {
        "username" => "-u",
        "database" => "--database",
        "password" => "-P",
        "port"     => "--port",
        "host"     => "-h",
      }
      option_string = ""
      config_options.each do |k, v|
        val = env_config[ k ]
        next if val.nil?
        option_string += " #{v} #{val}"
      end
      system("mysql #{option_string} -e '#{cmd}'")
    end
  end
  Pry::Commands.add_command(Pry::Command::MySQLQueryCommand)
end