def lambda_handler()

in lambda/traffic_simulator/handler.py [0:0]


def lambda_handler(event, context):

    s3.download_file(BUCKET, KEY, "/tmp/test.csv")

    with open("/tmp/test.csv") as f:
        reader = csv.reader(f)
        chosen_row = random.choice(list(reader))
        chosen_row = ",".join([str(elem) for elem in chosen_row])

    _ = sqs.send_message(QueueUrl=QUEUE_URL, MessageBody=(chosen_row))