Created
June 3, 2020 19:42
-
-
Save visaolive/aaff9849494b972895ad3b8212a2b9ed to your computer and use it in GitHub Desktop.
GCPPSCaseSalesforce.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from simple_salesforce import Salesforce, SalesforceLogin | |
| from simple_salesforce import SFType | |
| from google.cloud import secretmanager | |
| import base64 | |
| import requests | |
| import json | |
| import os | |
| def main(event, context): | |
| print('event payload {}!'.format(event)) | |
| sftype_object = os.environ["sftype_object"] | |
| try: | |
| if 'data' in event: | |
| message_data = base64.b64decode(event['data']).decode('utf-8') | |
| print('event payload - message_data {}!'.format(message_data)) | |
| request_json = json.loads(message_data) | |
| print('event payload - request_json {}!'.format(request_json)) | |
| else: | |
| raise ValueError('data key missing in Pub/Sub message.') | |
| #print('event payload - request_json {}!'.format(request_json)) | |
| #retrieve salesforce session and instance reference | |
| session_id, instance = sf_login() | |
| record = SFType(sftype_object,session_id,instance) | |
| #send payload to Salesforce API | |
| record.create(request_json) | |
| #parse response from Salesforce API | |
| record_submit = record.describe() | |
| print("main - record_submit: {}".format(record_submit)) | |
| return "Main Request Passed" | |
| except Exception as error: | |
| print('Main Error: ' + repr(error)) | |
| def sf_login(): | |
| # Create the Secret Manager client. | |
| client = secretmanager.SecretManagerServiceClient() | |
| # Organize the Secret Keys | |
| sf_user_prod = "SF_USER_PROD" | |
| sf_pass_prod = "SF_PASS_PROD" | |
| sf_token_prod = "SF_TOKEN_PROD" | |
| # Pass in the GCP Project ID | |
| # This will be found on the Secret Manager > Secret > Secret Details | |
| # projects/[gcp_project_id]/secrets/[secret] | |
| project_id = os.environ["gcp_project_id"] | |
| # Obtain the Secret Name Path | |
| sf_user_prod_name = f"projects/{project_id}/secrets/{sf_user_prod}/versions/latest" | |
| sf_pass_prod_name = f"projects/{project_id}/secrets/{sf_pass_prod}/versions/latest" | |
| sf_token_prod_name = f"projects/{project_id}/secrets/{sf_token_prod}/versions/latest" | |
| # Obtain the Latest Secret Version | |
| sf_user_prod_response = client.access_secret_version(sf_user_prod_name) | |
| sf_pass_prod_response = client.access_secret_version(sf_pass_prod_name) | |
| sf_token_prod_response = client.access_secret_version(sf_token_prod_name) | |
| # Parse the Secret Response & Decode Payload | |
| sf_user_prod_secret = sf_user_prod_response.payload.data.decode('UTF-8') | |
| sf_pass_prod_secret = sf_pass_prod_response.payload.data.decode('UTF-8') | |
| sf_token_prod_secret = sf_token_prod_response.payload.data.decode('UTF-8') | |
| # Assign Variables to Pass into Salesforce Login | |
| sf_username = sf_user_prod_secret | |
| sf_password = sf_pass_prod_secret | |
| sf_token = sf_token_prod_secret | |
| try: | |
| session_id, instance = SalesforceLogin( | |
| username = sf_username, | |
| password = sf_password, | |
| security_token = sf_token) | |
| return session_id, instance | |
| except Exception as error: | |
| print('Login Error: ' + repr(error)) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment