def lambda_handler(event, context):
logger.info("event: %s" % json.dumps(event))
event_record = event['Records'][0]
bucket = event_record['s3']['bucket']['name']
logger.debug("bucket: %s" % bucket)
key = urllib.parse.unquote(event_record['s3']['object']['key'])
logger.debug("key: %s" % key)
call_traces_raw = s3.get_object(Bucket=bucket, Key=key)["Body"].read().decode()
sf = Salesforce(version=get_arg(os.environ, "SF_VERSION"),
consumer_key=get_arg(os.environ, "SF_CONSUMER_KEY"),
consumer_secret=get_arg(os.environ, "SF_CONSUMER_SECRET"),
username=get_arg(os.environ, "SF_USERNAME"),
password=get_arg(os.environ, "SF_PASSWORD") + get_arg(os.environ, "SF_ACCESS_TOKEN"))
if get_arg(os.environ, "SF_PRODUCTION").lower() == "true":
sf.set_production()
sf.sign_in()
for call_trace_raw in call_traces_raw.split("\n"):
logger.debug("call trace: %s" % call_trace_raw)
if len(call_trace_raw) == 0:
continue
call_trace = json.loads(call_trace_raw)
for func in [process_transcribe, process_task_update]:
try:
func(sf, call_trace)
except Exception as ex:
logger.error(traceback.print_exc())
s3.delete_object(Bucket=bucket, Key=key)
logger.info("done")
def lambda_handler(event, context):
logger.info("event: %s" % json.dumps(event))
sf = Salesforce(version=get_arg(os.environ, "SF_VERSION"),
consumer_key=get_arg(os.environ, "SF_CONSUMER_KEY"),
consumer_secret=get_arg(os.environ, "SF_CONSUMER_SECRET"),
username=get_arg(os.environ, "SF_USERNAME"),
password=get_arg(os.environ, "SF_PASSWORD") + get_arg(os.environ, "SF_ACCESS_TOKEN"))
if get_arg(os.environ, "SF_PRODUCTION").lower() == "true":
sf.set_production()
sf.sign_in()
resp = create(sf=sf, **event['Details']['Parameters'])
logger.info("result: %s" % resp)
return resp
def lambda_handler(event, context):
logger.info("event: %s" % json.dumps(event))
event_record = event['Records'][0]
bucket = event_record['s3']['bucket']['name']
logger.debug("bucket: %s" % bucket)
key = urllib.parse.unquote(event_record['s3']['object']['key'])
logger.debug("key: %s" % key)
data = s3.get_object(Bucket=bucket, Key=key)["Body"].read().decode()
sf = Salesforce(version=get_arg(os.environ, "SF_VERSION"),
consumer_key=get_arg(os.environ, "SF_CONSUMER_KEY"),
consumer_secret=get_arg(os.environ, "SF_CONSUMER_SECRET"),
username=get_arg(os.environ, "SF_USERNAME"),
password=get_arg(os.environ, "SF_PASSWORD") + get_arg(os.environ, "SF_ACCESS_TOKEN"))
if get_arg(os.environ, "SF_PRODUCTION").lower() == "true":
sf.set_production()
sf.sign_in()
for record in csv.DictReader(data.split("\n")):
sf.create("ACT_HistoricalReportData__c", prepare_agent_record(record, event_record['eventTime']))
logger.info("done")
class SalesforceTest(unittest.TestCase):
def setUp(self):
self.version = "v5"
self.consumer_key = "consumer_key"
self.consumer_secret = "consumer_secret"
self.username = "******"
self.password = "******"
self.access_token = "access_token"
self.sf = Salesforce(self.version, self.consumer_key,
self.consumer_secret, self.username,
self.password)
def test_sign_in(self):
data = {
'access_token': self.access_token,
'instance_url': 'https://instanceid.salesforce.com'
}
resp = unittest.mock.Mock()
resp.json.return_value = data
self.sf.request = unittest.mock.Mock()
self.sf.request.post.return_value = resp
self.sf.sign_in()
self.assertEqual(data['access_token'], self.sf.access_token)
self.assertEqual(data['instance_url'], self.sf.host)
self.sf.request.post.called_once()
_, kwargs = self.sf.request.post.call_args
self.assertTrue("client_id" in kwargs['params'])
self.assertEqual(self.consumer_key, kwargs['params']['client_id'])
self.assertTrue("client_secret" in kwargs['params'])
self.assertEqual(self.consumer_secret,
kwargs['params']['client_secret'])
self.assertTrue("username" in kwargs['params'])
self.assertEqual(self.username, kwargs['params']['username'])
self.assertTrue(self.password in kwargs['params'])
self.assertEqual(self.password, kwargs['params']['password'])
self.assertTrue('Authorization' in self.sf.headers)
self.assertTrue(
self.sf.access_token in self.sf.headers['Authorization'])
self.assertTrue("Content-Type" in self.sf.headers)
self.assertEqual("application/json", self.sf.headers["Content-Type"])
def test_query(self):
self.sf.headers = {
'Authorization': 'Bearer %s' % self.access_token,
'Content-Type': 'application/json'
}
self.sf.host = "https://instancid.salesforce.com"
data_original = {
"done":
True,
"totalSize":
2,
"records": [{
"attributes": {
"type":
"Account",
"url":
"/services/data/v20.0/sobjects/Account/001D000000IRFmaIAH"
},
"Name": "Test 1"
}, {
"attributes": {
"type":
"Account",
"url":
"/services/data/v20.0/sobjects/Account/001D000000IomazIAB"
},
"Name": "Test 2"
}]
}
resp = unittest.mock.Mock()
resp.json.return_value = data_original
self.sf.request.get = unittest.mock.Mock(return_value=resp)
query = "SELECT Name FROM Account"
data = self.sf.query(query)
self.assertEqual(data_original['records'], data)
_, kwargs = self.sf.request.get.call_args
self.assertEqual(
self.sf.host + "/services/data/" + self.version + "/query",
kwargs['url'])
self.assertTrue('params' in kwargs)
self.assertEqual(query, kwargs['params']['q'])
def test_search(self):
self.sf.headers = {
'Authorization': 'Bearer %s' % self.access_token,
'Content-Type': 'application/json'
}
self.sf.host = "https://instancid.salesforce.com"
data_original = {
"searchRecords": [{
"attributes": {
"type":
"Account",
"url":
"/services/data/v35.0/sobjects/Account/001D000000IqhSLIAZ"
},
"Id": "001D000000IqhSLIAZ",
}, {
"attributes": {
"type":
"Account",
"url":
"/services/data/v35.0/sobjects/Account/001D000000IomazIAB"
},
"Id": "001D000000IomazIAB"
}]
}
resp = unittest.mock.Mock()
resp.json.return_value = data_original
self.sf.request.get = unittest.mock.Mock(return_value=resp)
query = "FIND Acme"
data = self.sf.search(query)
self.assertEqual(data_original['searchRecords'], data)
_, kwargs = self.sf.request.get.call_args
self.assertEqual(
self.sf.host + "/services/data/" + self.version + "/search",
kwargs['url'])
self.assertTrue('params' in kwargs)
self.assertEqual(query, kwargs['params']['q'])
def test_update(self):
self.sf.headers = {
'Authorization': 'Bearer %s' % self.access_token,
'Content-Type': 'application/json'
}
self.sf.host = "https://instancid.salesforce.com"
self.sf.request = unittest.mock.Mock(return_value=(None, None))
sobject = "Case"
data = {'Name': 'Pete'}
sojb_id = "abc"
self.sf.update(sobject, sojb_id, data)
_, kwargs = self.sf.request.patch.call_args
self.assertEqual(self.sf.host + "/services/data/v5/sobjects/Case/abc",
kwargs['url'])
self.assertEqual(data, kwargs['data'])
self.assertEqual(self.sf.headers, kwargs['headers'])
def test_update_by_external(self):
self.sf.headers = {
'Authorization': 'Bearer %s' % self.access_token,
'Content-Type': 'application/json'
}
self.sf.host = "https://instancid.salesforce.com"
self.sf.request = unittest.mock.Mock(return_value=None)
sobject = "Case"
sfield = "call_id__c"
data = {'Name': 'Pete'}
sojb_id = "abc"
self.sf.update_by_external(sobject, sfield, sojb_id, data)
_, kwargs = self.sf.request.patch.call_args
self.assertEqual(
self.sf.host + "/services/data/v5/sobjects/Case/call_id__c/abc",
kwargs['url'])
self.assertEqual(data, kwargs['data'])
self.assertEqual(self.sf.headers, kwargs['headers'])
def test_create(self):
self.sf.headers = {
'Authorization': 'Bearer %s' % self.access_token,
'Content-Type': 'application/json'
}
self.sf.host = "https://instancid.salesforce.com"
resp_data = {"id": "001D000000IqhSLIAZ", "errors": [], "success": True}
resp = unittest.mock.Mock()
resp.json.return_value = resp_data
self.sf.request.post = unittest.mock.Mock(return_value=resp)
sobj = "Account"
data = {'Name': 'Pete'}
new_id = self.sf.create(sobj, data)
self.assertEqual(resp_data['id'], new_id)
_, kwargs = self.sf.request.post.call_args
self.assertTrue('url' in kwargs)
self.assertEqual(self.sf.host + "/services/data/v5/sobjects/Account",
kwargs['url'])
self.assertTrue('data' in kwargs)
self.assertEqual(data, kwargs['data'])
self.assertEqual(self.sf.headers, kwargs['headers'])
def test_delete(self):
self.sf.headers = {
'Authorization': 'Bearer %s' % self.access_token,
'Content-Type': 'application/json'
}
self.sf.host = "https://instancid.salesforce.com"
resp = unittest.mock.Mock()
resp.json.return_value = "create_data"
self.sf.request.delete = unittest.mock.Mock(return_value=resp)
sobj = "Account"
sobj_id = "abc"
self.sf.delete(sobj, sobj_id)
_, kwargs = self.sf.request.delete.call_args
self.assertTrue('url' in kwargs)
self.assertEqual(
self.sf.host + "/services/data/v5/sobjects/Account/abc",
kwargs['url'])
self.assertEqual(self.sf.headers, kwargs['headers'])
def test_set_production(self):
current_login_host = self.sf.login_host
self.sf.set_production()
self.assertNotEqual(current_login_host, self.sf.login_host)
@unittest.skip("Configuration required")
def test_end_to_end(self):
import os
import sf_config
import uuid
sf = Salesforce(version=os.environ["SF_VERSION"],
consumer_key=os.environ["SF_CONSUMER_KEY"],
consumer_secret=os.environ["SF_CONSUMER_SECRET"],
username=os.environ["SF_USERNAME"],
password=os.environ["SF_PASSWORD"])
sf.sign_in()
name = str(uuid.uuid4())
sf.create("Account", {'Name': name})
acc = sf.query("SELECT Id FROM Account WHERE Name='%s'" % name)[0]
sf.update("Account",
sobj_id=acc['Id'],
data={"Description": "hello world"})
sf.delete("Account", acc['Id'])