U cc(@sddlZddlZddlmZddlmZddlmZddlm Z ej rnddlm Z m Z ddl mZdd lmZGd d d eZdS) N)ResourceNotExistsError)AbstractPersistenceAdapter)PersistenceException)user_id_partition_keygen)CallableDict)RequestEnvelope)ServiceResourcec@sJeZdZdZdddeedfddZdd Zd d Z d d Z ddZ dS)DynamoDbAdapteraPersistence Adapter implementation using Amazon DynamoDb. Amazon DynamoDb based persistence adapter implementation. This internally uses the AWS Python SDK (`boto3`) to process the dynamodb operations. The adapter tries to create the table if ``create_table`` is set, during initialization. :param table_name: Name of the table to be created or used :type table_name: str :param partition_key_name: Partition key name to be used. Defaulted to 'id' :type partition_key_name: str :param attribute_name: Attribute name for storing and retrieving attributes from dynamodb. Defaulted to 'attributes' :type attribute_name: str :param create_table: Should the adapter try to create the table if it doesn't exist. Defaulted to False :type create_table: bool :param partition_keygen: Callable function that takes a request envelope and provides a unique partition key value. Defaulted to user id keygen function :type partition_keygen: Callable[[RequestEnvelope], str] :param dynamodb_resource: Resource to be used, to perform dynamo operations. Defaulted to resource generated from boto3 :type dynamodb_resource: boto3.resources.base.ServiceResource id attributesFdynamodbcCs0||_||_||_||_||_||_|dS)a^Persistence Adapter implementation using Amazon DynamoDb. Amazon DynamoDb based persistence adapter implementation. This internally uses the AWS Python SDK (`boto3`) to process the dynamodb operations. The adapter tries to create the table if `create_table` is set, during initialization. :param table_name: Name of the table to be created or used :type table_name: str :param partition_key_name: Partition key name to be used. Defaulted to 'id' :type partition_key_name: str :param attribute_name: Attribute name for storing and retrieving attributes from dynamodb. Defaulted to 'attributes' :type attribute_name: str :param create_table: Should the adapter try to create the table if it doesn't exist. Defaulted to False :type create_table: bool :param partition_keygen: Callable function that takes a request envelope and provides a unique partition key value. Defaulted to user id keygen function :type partition_keygen: Callable[[RequestEnvelope], str] :param dynamodb_resource: Resource to be used, to perform dynamo operations. Defaulted to resource generated from boto3 :type dynamodb_resource: boto3.resources.base.ServiceResource N) table_namepartition_key_nameattribute_name create_tablepartition_keygenr,_DynamoDbAdapter__create_table_if_not_exists)selfrrrrrZdynamodb_resourcers"zDynamoDbAdapter.__init__c CszN|j|j}||}|j|j|idd}d|krF|d|jWSiWSWn^tk rrtd |jYn<t k r}ztd t |j t |W5d}~XYnXdS)aGet attributes from table in Dynamodb resource. Retrieves the attributes from Dynamodb table. If the table doesn't exist, returns an empty dict if the ``create_table`` variable is set as True, else it raises PersistenceException. Raises PersistenceException if `get_item` fails on the table. :param request_envelope: Request Envelope passed during skill invocation :type request_envelope: ask_sdk_model.RequestEnvelope :return: Attributes stored under the partition keygen mapping in the table :rtype: Dict[str, object] :raises: :py:class:`ask_sdk_core.exceptions.PersistenceException` T)KeyZConsistentReadItemzqDynamoDb table {} doesn't exist or in the process of being created. Failed to get attributes from DynamoDb table.zTFailed to retrieve attributes from DynamoDb table. Exception of type {} occurred: {}N)rTablerrZget_itemrrrrformat Exceptiontype__name__str)rrequest_envelopetablepartition_key_valresponseerrrget_attributeshs.  zDynamoDbAdapter.get_attributesc Csz4|j|j}||}|j|j||j|idWn^tk rXtd |jYn<t k r}ztd t |j t |W5d}~XYnXdS)abSaves attributes to table in Dynamodb resource. Saves the attributes into Dynamodb table. Raises PersistenceException if table doesn't exist or ``put_item`` fails on the table. :param request_envelope: Request Envelope passed during skill invocation :type request_envelope: ask_sdk_model.RequestEnvelope :param attributes: Attributes stored under the partition keygen mapping in the table :type attributes: Dict[str, object] :rtype: None :raises: :py:class:`ask_sdk_core.exceptions.PersistenceException` )rzMDynamoDb table {} doesn't exist. Failed to save attributes to DynamoDb table.zNFailed to save attributes to DynamoDb table. Exception of type {} occurred: {}N)rrrrZput_itemrrrrrrrrr )rr!r r"r#r%rrrsave_attributess,  zDynamoDbAdapter.save_attributesc Csz.|j|j}||}|j|j|idWn^tk rRtd|jYn<t k r}ztdt |j t |W5d}~XYnXdS)aDeletes attributes from table in Dynamodb resource. Deletes the attributes from Dynamodb table. Raises PersistenceException if table doesn't exist or ``delete_item`` fails on the table. :param request_envelope: Request Envelope passed during skill invocation :type request_envelope: ask_sdk_model.RequestEnvelope :rtype: None :raises: :py:class:`ask_sdk_core.exceptions.PersistenceException` )rzQDynamoDb table {} doesn't exist. Failed to delete attributes from DynamoDb table.zPFailed to delete attributes in DynamoDb table. Exception of type {} occurred: {}N) rrrrZ delete_itemrrrrrrrr )rr!r"r#r%rrrdelete_attributess&  z!DynamoDbAdapter.delete_attributesc Cs|jrz4|jj|j|jddg|jddgddddWnHtk r}z*|jjdkrrtd t |jt |W5d }~XYnXd S) zCreates table in Dynamodb resource if it doesn't exist and create_table is set as True. :rtype: None :raises: PersistenceException: When `create_table` fails on dynamodb resource. HASH) AttributeNameZKeyTypeS)r*Z AttributeType)ZReadCapacityUnitsZWriteCapacityUnits)Z TableNameZ KeySchemaZAttributeDefinitionsZProvisionedThroughputZResourceInUseExceptionzLCreate table if not exists request failed: Exception of type {} occurred: {}N) rrrrr __class__rrrrr )rr%rrrZ__create_table_if_not_existss0   z,DynamoDbAdapter.__create_table_if_not_existsN) r __module__ __qualname____doc__rboto3resourcerr&r'r(rrrrrr s *'"r )r1typingZ boto3.sessionrZask_sdk_core.attributes_managerrZask_sdk_core.exceptionsrrr TYPE_CHECKINGrrZ ask_sdk_modelr Zboto3.resources.baser r rrrrs