diff --git a/airflow/providers/salesforce/hooks/salesforce.py b/airflow/providers/salesforce/hooks/salesforce.py index 36426e8522fbd..47549e5c7bceb 100644 --- a/airflow/providers/salesforce/hooks/salesforce.py +++ b/airflow/providers/salesforce/hooks/salesforce.py @@ -134,19 +134,19 @@ def get_conn(self) -> api.Salesforce: self.conn = Salesforce( username=connection.login, password=connection.password, - security_token=extras["extra__salesforce__security_token"] or None, - domain=extras["extra__salesforce__domain"] or None, + security_token=extras.get("extra__salesforce__security_token"), + domain=extras.get("extra__salesforce__domain"), session_id=self.session_id, - instance=extras["extra__salesforce__instance"] or None, - instance_url=extras["extra__salesforce__instance_url"] or None, - organizationId=extras["extra__salesforce__organization_id"] or None, - version=extras["extra__salesforce__version"] or api.DEFAULT_API_VERSION, - proxies=extras["extra__salesforce__proxies"] or None, + instance=extras.get("extra__salesforce__instance"), + instance_url=extras.get("extra__salesforce__instance_url"), + organizationId=extras.get("extra__salesforce__organization_id"), + version=extras.get("extra__salesforce__version", api.DEFAULT_API_VERSION), + proxies=extras.get("extra__salesforce__proxies"), session=self.session, - client_id=extras["extra__salesforce__client_id"] or None, - consumer_key=extras["extra__salesforce__consumer_key"] or None, - privatekey_file=extras["extra__salesforce__private_key_file_path"] or None, - privatekey=extras["extra__salesforce__private_key"] or None, + client_id=extras.get("extra__salesforce__client_id"), + consumer_key=extras.get("extra__salesforce__consumer_key"), + privatekey_file=extras.get("extra__salesforce__private_key_file_path"), + privatekey=extras.get("extra__salesforce__private_key"), ) return self.conn