Simple funciton to connect to Salesforce (and sandbox when needed)

connect_salesforce.py

def connect_salesforce(username: str, password: str, security_token: str, is_test: bool) -> Salesforce:
    '''Returns a Salesforce connection object. If test, it'll connect to Sf sandbox'''
    domain = None
    if is_test:
        domain = 'test'
        username += '.full'
    return Salesforce(username, password, security_token, domain=domain)


sf = connect_salesforce(username, password, security_token, False)