import sys


def get_state_machines():
    # Add state machine as key with array of lambdas that run within it
    state_machines = {
        "workflow-transfer-assets": [
            'prep_db_for_transfer',
            'dd_asset_location_update']
    }
    return state_machines


def main(lambda_name):
    state_machines = get_state_machines()
    for state_machine in state_machines.keys():
        lambdas = state_machines[state_machine]
        if lambda_name in lambdas:
            print(state_machine)


main(sys.argv[1])