Esempio n. 1
0
def main():
    arguments_list = [['--profile', '-p'], ['--region',
                                            '-r'], ['--rds', '-rds'],
                      ['--create', '-c'], ['--delete', '-d'],
                      ['--delete-name', '-dn'], ['--protocol', '-pr'],
                      ['--ip', '-i'], ['--port', '-po'],
                      ['--rules-from', '-rf'], ['--fields', '-f']]

    parser = argparse.ArgumentParser()
    parser = fast_add_arguments(arguments_list, parser)
    args = parser.parse_args()

    client_config = Client_Config()

    if args.profile:
        client_config.set_profile(args.profile)
    if args.region:
        client_config.set_region(args.region)
    if args.create:
        create_sg(args)
    elif args.delete:
        ec2 = Client()
        sgclient = SG_Client()
        sgs_to_delete = args.delete.split(",")
        for sg_to_delete in sgs_to_delete:
            sgclient.set_client(ec2).set_group_id(sg_to_delete).delete_sg()
    elif args.delete_name:
        ec2 = Client()
        sgclient = SG_Client()
        sgsnames_to_delete = args.delete_name.split(",")
        for sgname_to_delete in sgsnames_to_delete:
            sgclient.set_client(ec2).set_group_name(
                sgname_to_delete).delete_sg()
    elif args.rules_from:
        get_rules_from(client_config, args)
    else:
        list_sg(args, client_config)
Esempio n. 2
0
import sys
import argparse
sys.path.insert(1, '..')
from awssg.Client_Config import Client_Config
from awssg.Fetcher import Fetcher
from awssg.Helpers import fast_add_arguments

client_config = Client_Config()

arguments_list = [
    ['--profile', '-p'],
    ['--region', '-r']
]
parser = argparse.ArgumentParser()
parser = fast_add_arguments(arguments_list, parser)
args = parser.parse_args()


if args.profile is not None:
    client_config.set_profile(args.profile)

if args.region is not None:
    client_config.set_region(args.region)

fetcher = Fetcher()
fetcher.set_client(client_config.get_client())
print(fetcher.get_all_regions_data())