metadb.py 文件源码

def __init__(self):
        """
        Initialize the data base
        Returns:

        """
        self.__local_dynamo = os.environ.get('USING_DJANGO_TESTRUNNER') is not None
        if not self.__local_dynamo:
            session = get_session()
            dynamodb = session.resource('dynamodb')
            if 'test' in sys.argv:
                # TODO: This needs to be made more robust. Parameters should be mocked, not assumed.
                tablename = 'intTest.' + config["aws"]["meta-db"]
            else:
                tablename = config["aws"]["meta-db"]
        else:
            tablename = config["aws"]["meta-db"]
            session = boto3.Session(aws_access_key_id='foo', aws_secret_access_key='foo')
            dynamodb = session.resource('dynamodb', region_name='us-east-1', endpoint_url='http://localhost:8000')

        self.table = dynamodb.Table(tablename)