full_scan.py

#! /usr/bin/env python3
from sys import exit

try:
    import config
except ModuleNotFoundError:
    print("Please create a 'config.py' before running this tool")
    exit(1)

import crayons
import helper
import timeit
from datetime import datetime

from modules import (
    ami,
    ec2_snapshots,
    ec2,
    eip,
    rds_snapshots,
    securitygroups,
    volumes,
    lambdafunctions,
    cloudwatch,
    cloudformation,
    s3,
)

all_regions = helper.get_all_regions()

if len(config.REGIONS) == 0:
    print(crayons.red("REGIONS must contain at least one region".format(r)))
    exit(1)

for r in config.REGIONS:
    if r not in all_regions:
        print(crayons.red("{} is not a valid region".format(r)))
        exit(1)


print("Starting full account scan for these Regions: {}".format(",".join(config.REGIONS)))
account_id = helper.get_account_id()
print("Account: {}".format(account_id))
start_date = datetime.now().strftime("%Y-%m-%d_%H:%M")
print("Start date: {}".format(start_date))

start = timeit.default_timer()

print(crayons.yellow("Scanning AMIs"))
ami.scan()

print(crayons.yellow("Scanning EC2 instances"))
ec2.scan()

print(crayons.yellow("Scanning EC2 volumes"))
volumes.scan()

print(crayons.yellow("Scanning EC2 snapshots"))
ec2_snapshots.scan()

print(crayons.yellow("Scanning RDS snapshots"))
rds_snapshots.scan()

print(crayons.yellow("Scanning security groups"))
securitygroups.scan()

print(crayons.yellow("Scanning CloudWatch loggroups"))
cloudwatch.scan()

print(crayons.yellow("Scanning Lambda functions"))
lambdafunctions.scan()

print(crayons.yellow("Scanning CloudFormation"))
cloudformation.scan()

print(crayons.yellow("Scanning S3"))
s3.scan()

print(crayons.yellow("Scanning elastic IPs"))
eip.scan()

stop = timeit.default_timer()
runtime = int(stop - start)
print("Scan finished after {} seconds".format(crayons.yellow(runtime)))