dynamodb.py

from pynamodb.models import Model
from pynamodb.attributes import (
    UnicodeAttribute,
    JSONAttribute
)
from settings import DYNAMO_SETTINGS

__all__ = [
    "TwitterStreamData",
    "TwitterExcelData",
]


class TwitterStreamData(Model):
    class Meta:
        write_capacity_units = 2
        read_capacity_units = 2
        table_name = "TwitterStreamData"
        host = DYNAMO_SETTINGS['HOST']
        region = DYNAMO_SETTINGS['REGION']
    _id = UnicodeAttribute(hash_key=True)
    data = JSONAttribute()


class TwitterExcelData(Model):
    class Meta:
        write_capacity_units = 5
        read_capacity_units = 1
        table_name = "TwitterExcelData"
        host = DYNAMO_SETTINGS['HOST']
        region = DYNAMO_SETTINGS['REGION']
    _id = UnicodeAttribute(hash_key=True)
    data = JSONAttribute()


if not TwitterStreamData.exists():
    TwitterStreamData.create_table(wait=True)

if not TwitterExcelData.exists():
    TwitterExcelData.create_table(wait=True)

if __name__ == "__main__" and __package__ is None:
    __package__ = "twitter_code_base.services"