kinesis put get example

kinesis.py

import boto3
from pprint import pprint


client = boto3.client('kinesis', region_name='us-east-1')


def put():
    response = client.put_record(
        StreamName='teste',
        Data=b'bytes',
        PartitionKey='string'
    )
    pprint(response)


def get():
    shards = client.describe_stream(StreamName='teste')['StreamDescription']['Shards']
    shard_ids = [s['ShardId'] for s in shards]

    shard_iterator = client.get_shard_iterator(
        StreamName='teste',
        ShardId=shard_ids[0],
        ShardIteratorType='LATEST')['ShardIterator']

    response = client.get_records(ShardIterator=shard_iterator)
    pprint(response)
    pprint(len(response['Records']))


if __name__ == '__main__':
    put()
    get()