lib/pgcli/rails/dbconsole.rb in pgcli-rails-0.2.1 vs lib/pgcli/rails/dbconsole.rb in pgcli-rails-0.3.0

@@ -1,5 +1,9 @@
-require "rails/commands/dbconsole"
+begin
+  require "rails/commands/dbconsole"
+rescue LoadError
+  require "rails/commands/dbconsole/dbconsole_command"
+end
 
 module Pgcli
   module Rails
     # Subclass the built-in Rails DBConsole to override "psql" with "pgcli".
     class DBConsole < ::Rails::DBConsole