Skip to content

Commit

Permalink
fix: Add validations in place
Browse files Browse the repository at this point in the history
  • Loading branch information
ayush-shah committed Sep 17, 2024
1 parent a1a8d93 commit 3298cdc
Showing 1 changed file with 7 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
from typing import Optional

from simple_salesforce.api import Salesforce
from sqlalchemy.engine import Engine

from metadata.generated.schema.entity.automations.workflow import (
Workflow as AutomationWorkflow,
Expand All @@ -27,14 +26,18 @@
from metadata.ingestion.ometa.ometa_api import OpenMetadata


def get_connection(connection: SalesforceConnection) -> Engine:
def get_connection(connection: SalesforceConnection):
"""
Create connection
"""
return Salesforce(
username=connection.username,
password=connection.password.get_secret_value(),
security_token=connection.securityToken.get_secret_value(),
password=connection.password.get_secret_value()
if connection.password
else None,
security_token=connection.securityToken.get_secret_value()
if connection.securityToken
else None,
domain=connection.salesforceDomain,
version=connection.salesforceApiVersion,
**connection.connectionArguments.root if connection.connectionArguments else {},
Expand Down

0 comments on commit 3298cdc

Please sign in to comment.