sg-lambda-function.py

import json
import boto3

def get_network_settings():
    ec2=boto3.client('ec2')
    vpcs = ec2.describe_vpcs()
    
    vpcid = vpcs['Vpcs'][0]['VpcId']
    
    print(vpcid)
    
    subnets = ec2.describe_subnets()
    lst_subnets=[]
    for subnet in subnets['Subnets']:
        #print(subnet)
        #print(subnet['SubnetId'])
        lst_subnets.append(subnet['SubnetId'])
    print(lst_subnets)
    
    sggroups = ec2.describe_security_groups()
    lst_sgs=[]
    for sg in sggroups['SecurityGroups']:
        if 'default' in sg['GroupName']:
            lst_sgs.append(sg['GroupId'])
    print(lst_sgs)
    
    return(vpcid, lst_subnets, lst_sgs)

def get_iam_execution_role():
    client = boto3.client('iam')
    
    response = client.get_role(
    RoleName='SandboxServiceRole'
                                )
    role = response['Role']['Arn']
    print(role)
    return role
    #return ('arn:aws:iam::897994370356:role/SandboxServiceRole')

def create_sm_domain():
    vpc,subnets,securitygroups = get_network_settings()
    client = boto3.client('sagemaker')
    
    response = client.create_domain(
    DomainName='innovative-sdbx-domain',
    AuthMode='IAM',
    DefaultUserSettings={
        'ExecutionRole': get_iam_execution_role(),
        'SecurityGroups':securitygroups,
        
        
        
        'RStudioServerProAppSettings': {
            'AccessStatus': 'DISABLED',
            #'UserGroup': 'R_STUDIO_ADMIN'|'R_STUDIO_USER'
        },
       
        
    },
    SubnetIds=subnets,
    VpcId=vpc,
    
    AppNetworkAccessType='VpcOnly',
    
    DomainSettings={
        'SecurityGroupIds':securitygroups,
        
        #'ExecutionRoleIdentityConfig': 'USER_PROFILE_NAME'|'DISABLED'
    },
   )
    return response
    
def lambda_handler(event, context):
    # TODO implement
    #get_iam_execution_role()
    
    smclient = boto3.client('sagemaker')
    vpcid, lst_subnets,lst_sgs = get_network_settings()
    domainstatus = create_sm_domain()
    print(domainstatus)
    
    return {
        'statusCode': 200,
        'body': json.dumps('Hello from Lambda!')
    }