Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
# To change this license header, choose License Headers in Project Properties.
# To change this template file, choose Tools | Templates
# and open the template in the editor.
import beatbox
import xml.etree.ElementTree as ET
import traceback
import json
__author__ = "Matias Medina"
__date__ = "$11/07/2016 04:02:59 PM$"
class XmlEngine:
# Create a map of attachments by id
def getMapAttachments(self, parentIds, service):
# Query to Salesforce to get the attachments
query_result = service.query("SELECT Id, ParentId, Body FROM Attachment WHERE ParentId IN (" + parentIds + ")")
records = query_result['records']
total_records = query_result['size']
query_locator = query_result['queryLocator']
# Map id to xml string
mapParentIdToXml = {}
# Loop through, pulling the next 500 and appending it to your records dict
while query_result['done'] is False and len(records) < total_records:
query_result = service.queryMore(query_locator)
query_locator = query_result['queryLocator']
records = records + query_result['records']
# Build the map
for att in records:
if att.Body != None:
mapParentIdToXml[att.ParentId] = att.Body.decode('base64')
return mapParentIdToXml
# Read the xml and get the data needed to update Salesforce records
def getDemandSavings(self, xmlStr):
result = ""
node = ""
try:
# Get the root node
root = ET.fromstring(xmlStr)
# Go through the xml nodes to get needed node
for child in root:
if "Project" in child.tag:
for secondChild in child:
if "ProjectDetails" in secondChild.tag:
for thirdChild in secondChild:
if "EnergySavingsInfo" in thirdChild.tag:
for fourthChild in thirdChild:
if "DemandSavings" in fourthChild.tag:
node = fourthChild.text
if node != "":
result = float(node)
except:
print "Malformed XML, missing or invalid DemandSavings"
return result
# Loop through a list of records and update them
def updateObject(self, recordsToUpdate, fieldToUpdate, fieldValue, objectType, service):
updateList = []
for id in recordsToUpdate:
objectUpdate = {"Id": id, fieldToUpdate: fieldValue, "type": objectType}
updateList.append(objectUpdate)
try:
service.update(objectUpdate)
except:
print "failed"
def runScript(self):
# Initialize the beatbox api to talk to Salesforce
service = beatbox.PythonClient()
# Read credentials to log in Salesforce
with open('settings.json') as data_file:
projectSettings = json.load(data_file)
service.serverUrl = str(projectSettings["SF_SERVER_URL"])
service.login(str(projectSettings["SF_ORG_USER"]), str(projectSettings["SF_ORG_PASSWORD"]))
# The "parentId" needs to be replaced by the id of the record that have attached the xml file
# you can pass a list of string if you want to work with more than one record
xmlByIdMap = self.getMapAttachments("parentId", service);
# Get the attachment stored at the map. Replace "appFileId" with the record id that contains the attachment
# that you you want to parse
xmlString = xmlByIdMap["appFileId"]
# Get the value needed from xml file. This value will be used to update an especific record in Salesforce
demandSavings = self.getDemandSavings(xmlString)
# Replace recordsToUpdate by the list of records to update with the parsed value
# Replace fieldToUpdate by the name of the field that you want to update. Eg: Demand_Savings__c
# Replace objectType by the object that you want to update. Eg: MyCustomObject__c
self.updateObject("recordsToUpdate", "fieldToUpdate", demandSavings, "objectType", service)
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.