def assisted_log_enabler()

in assisted_log_enabler.py [0:0]

• 78 lines of code
• 23 McCabe index (conditional complexity)

def assisted_log_enabler():
    """Function to run Assisted Log Enabler"""
    output_handle = logging.FileHandler('ALE_' + timestamp_date_string + '.log')
    output_handle.setLevel(logging.INFO)
    logger.addHandler(output_handle)
    formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
    output_handle.setFormatter(formatter)

    parser = argparse.ArgumentParser(description='Assisted Log Enabler - Find resources that are not logging, and turn them on.')
    parser.add_argument('--mode',help=' Choose the mode that you want to run Assisted Log Enabler in. Available modes: single_account, multi_account, cleanup, dryrun. WARNING: For multi_account, You must have the associated CloudFormation template deployed as a StackSet. See the README file for more details.')
    
    function_parser_group = parser.add_argument_group('Single & Multi Account Options', 'Use these flags to choose which services you want to turn logging on for.')
    function_parser_group.add_argument('--all', action='store_true', help=' Turns on all of the log types within the Assisted Log Enabler for AWS.')
    function_parser_group.add_argument('--eks', action='store_true', help=' Turns on Amazon EKS audit & authenticator logs.')
    function_parser_group.add_argument('--vpcflow', action='store_true', help=' Turns on Amazon VPC Flow Logs.')
    function_parser_group.add_argument('--r53querylogs', action='store_true', help=' Turns on Amazon Route 53 Resolver Query Logs.')
    function_parser_group.add_argument('--s3logs', action='store_true', help=' Turns on Amazon Bucket Logs.')
    function_parser_group.add_argument('--cloudtrail', action='store_true', help=' Turns on AWS CloudTrail. Only available in Single Account version.')

    cleanup_parser_group = parser.add_argument_group('Cleanup Options', 'Use these flags to choose which resources you want to turn logging off for.')
    cleanup_parser_group.add_argument('--single_r53querylogs', action='store_true', help=' Removes Amazon Route 53 Resolver Query Log resources created by Assisted Log Enabler for AWS.')
    cleanup_parser_group.add_argument('--single_cloudtrail', action='store_true', help=' Removes AWS CloudTrail trails created by Assisted Log Enabler for AWS.')
    cleanup_parser_group.add_argument('--single_vpcflow', action='store_true', help=' Removes Amazon VPC Flow Log resources created by Assisted Log Enabler for AWS.')
    cleanup_parser_group.add_argument('--single_all', action='store_true', help=' Turns off all of the log types within the Assisted Log Enabler for AWS.')
    cleanup_parser_group.add_argument('--single_s3logs', action='store_true', help=' Removes Amazon Bucket Log resources created by Assisted Log Enabler for AWS.')

    dryrun_parser_group = parser.add_argument_group('Dry Run Options', 'Use these flags to run Assisted Log Enabler for AWS in Dry Run mode.')
    dryrun_parser_group.add_argument('--single_account', action='store_true', help=' Runs Assisted Log Enabler for AWS in Dry Run mode for a single AWS account.')
    dryrun_parser_group.add_argument('--multi_account', action='store_true', help=' Runs Assisted Log Enabler for AWS in Dry Run mode for a multi-account AWS environment, using AWS Organizations.')

    args = parser.parse_args()
    banner()

    event = 'event'
    context = 'context'
    if args.mode == 'single_account':
        if args.eks:
            ALE_single_account.run_eks()
        elif args.vpcflow:
            ALE_single_account.run_vpc_flow_logs()
        elif args.r53querylogs:
            ALE_single_account.run_r53_query_logs()
        elif args.s3logs:
            ALE_single_account.run_s3_logs()
        elif args.cloudtrail:
            ALE_single_account.run_cloudtrail()
        elif args.all:
            ALE_single_account.lambda_handler(event, context)
        else:
            logging.info("No valid option selected. Please run with -h to display valid options.")
    elif args.mode == 'multi_account':
        if args.eks:
            ALE_multi_account.run_eks()
        elif args.vpcflow:
            ALE_multi_account.run_vpc_flow_logs()
        elif args.r53querylogs:
            ALE_multi_account.run_r53_query_logs()
        elif args.s3logs:
            ALE_multi_account.run_s3_logs()
        elif args.all:
            ALE_multi_account.lambda_handler(event, context)
        else:
            logging.info("No valid option selected. Please run with -h to display valid options.")
    elif args.mode == 'cleanup':
        if args.single_r53querylogs:
            ALE_cleanup_single.run_r53_cleanup()
        elif args.single_s3logs:
            ALE_cleanup_single.run_s3_cleanup()
        elif args.single_cloudtrail:
            ALE_cleanup_single.run_cloudtrail_cleanup()
        elif args.single_vpcflow:
            ALE_cleanup_single.run_vpcflow_cleanup()
        elif args.single_all:
            ALE_cleanup_single.lambda_handler(event, context)
        else:
            logging.info("No valid option selected. Please run with -h to display valid options.")
    elif args.mode == 'dryrun':
        if args.single_account:
            ALE_dryrun_single.lambda_handler(event, context)
        elif args.multi_account:
            ALE_dryrun_multi.lambda_handler(event, context)
        else:
            logging.info("No valid option selected. Please run with -h to display valid options.")
    else:
        print("No valid option selected. Please run with -h to display valid options.")