def lambda_handler()

in src/create_master_roles.py [0:0]

• 34 lines of code
• 5 McCabe index (conditional complexity)

def lambda_handler(event, context):
    logger.debug(f'Lambda event:{event}')
    account = None
    status = Constant.StateMachineStates.LINKED_ACCOUNT_FLOW
    account_id = None

    try:
        company_name = event['CompanyName']

        accounts = get_master_account(company_name=company_name)
        if not accounts:
            status = Constant.StateMachineStates.STANDALONE_ACCOUNT_FLOW
        else:
            account = accounts[0]
            account_id = account['AccountId']
            role_arn = f"arn:aws:iam::{account_id}:role/{account['AdminRole']}"
            account_session = get_session(role_arn)
            create_roles(account_session)

    except ClientError as ce:
        error_msg = log_error(logger=logger, account_id=account['AccountId'], company_name=account['CompanyName'],
                              error_type=Constant.ErrorType.CRME, error=ce,
                              notify=True, slack_handle=account['SlackHandle'])

        account["Error"] = error_msg
        raise ce
    except Exception as ex:
        log_error(logger=logger, account_id=event['AccountId'], company_name=account.get('CompanyName'),
                  error_type=Constant.ErrorType.CRME, notify=True, error=ex)
        raise ex
    finally:
        if account:
            update_item(Constant.DB_TABLE, account)

    return {'Data': {
        'Status': status,
        'CompanyName': company_name,
        'AccountId': account_id,
        'ProcessName': f"{company_name}-{account_id}-{time.monotonic_ns()}"}}