import json
import boto3
import uuid

def lambda_handler(event, context):

    c = boto3.client('iam')

    if 'group_name' in event.keys():
        response = c.add_user_to_group(
            GroupName = event['group_name'],
            UserName = event['user_name'],
        )
    else:
        password = str(uuid.uuid4())
        response = c.create_user(
            UserName = event['user_name'],
        )
        c.create_login_profile(
            UserName=event['user_name'],
            Password=password,
            PasswordResetRequired=True
        )

    return {
        'statusCode': 200,
        'body': json.dumps('Done')
    }