# Pastebin 8vPXewsW
import boto3


class DynamoDB(object):
    def __init__(self):
        client = boto3.resource('dynamodb')
        self.table = client.Table('table_name')
        self.primary_key = 'id'

    def get_item(self, item):
        try:
            response = self.table.get_item(Key={self.primary_key: item})
            return response['Item']
        except:
            raise Exception('Unable to retrieve item from DynamoDB table')

    def put_item(self, item):
        try:
            self.table.put_item(Item=item)
        except Exception as e:
            raise Exception('Unable to update DynamoDB with item data')

    def get_all_items(self):
        try:
            response = self.table.scan()
            return response['Items']

        except Exception as e:
            raise Exception('Unable to retrieve DynamoDB data')