Execute rake tasks including arguments using capistrano (based on: https://github.com/sheharyarn/capistrano-rake/blob/master/lib/capistrano/tasks/invoke.rake)

invoke_args.rake

require 'shellwords'

namespace :invoke do

  # Defalut to :app roles
  set :rake_roles, :app unless fetch(:rake_roles)

  desc "Execute a rake task on a remote server (cap invoke:rake TASK=db:migrate)"
  task :rake do
    if ENV['TASK']
      on roles(fetch(:rake_roles)) do
        within current_path do
          with rails_env: fetch(:rails_env) do
            execute 'bundle', 'exec', 'rake', Shellwords.escape(ENV['TASK'])
          end
        end
      end
    else
      puts "\n\nFailed! You need to specify the 'TASK' parameter!",
           "Usage: cap <stage> invoke:rake TASK=your:task"
    end
  end
end

Example: cap production invoke:rake TASK=some:customtask[argument1,argument2]