Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
CLI - thor file
require 'databasedotcom'
class Utils < Thor
desc "query SOQL", "runs a soql query and displays the value of each record's 'name' field"
method_option :config_file, :type => :string, :default => "databasedotcom.yml",
:aliases => "-c", :desc => "The name of the file containing the connection parameters."
def query(soql)
client = authenticate(options[:config_file])
# execute the soql and iterate over the results to output the name
client.query("#{soql}").each do |r|
puts r.Name
end
end
desc "export SOQL FIELDS FILE", "runs a soql query and exports the specified
comma separated list of fields to a comma separated file"
method_option :config_file, :type => :string, :default => "databasedotcom.yml",
:aliases => "-c", :desc => "The name of the file containing the connection parameters."
def export(soql, fields, file)
client = authenticate(options[:config_file])
# query for records
records = client.query("#{soql}")
# open the file to write (probably local directory)
File.open(file, 'w') do |f|
# interate over the records
records.each do |r|
# create a single line with all field values specified
line = ''
fields.split(',').each do |field|
line += "#{eval("r.#{field}")},"
end
# write each line to the csv file
f.puts "#{line}\n"
end
end
end
desc "describe OBJECT", "displays the describe info for a particular object"
method_option :config_file, :type => :string, :default => "databasedotcom.yml",
:aliases => "-c", :desc => "The name of the file containing the connection parameters."
def describe(object)
client = authenticate(options[:config_file])
# call describe on the object by name
sobject = client.describe_sobject(object)
# output the results -- not very useful (frowny face)
puts sobject
end
desc "get_token", "retreives an access token"
method_option :config_file, :type => :string, :default => "databasedotcom.yml",
:aliases => "-c", :desc => "The name of the file containing the connection parameters."
def get_token
client = authenticate(options[:config_file]).oauth_token
puts "Access token: #{client}"
end
desc "show_config", "display the salesforce connection properties"
method_option :config_file, :type => :string, :default => "databasedotcom.yml",
:aliases => "-c", :desc => "The name of the file containing the connection parameters."
def show_config
config = YAML.load_file(options[:config_file])
puts config
end
private
def authenticate(file_name)
# load the configuration file with connection parameters
config = YAML.load_file(file_name)
# init the databasedotcom gem with the specified yml config file
client = Databasedotcom::Client.new(file_name)
# pass the credentials to authenticate
client.authenticate :username => config['username'], :password => config['password']
return client
end
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.