U 3 d]@slddlZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl mZGdd d eZdS) N)crc32)AWSAuthConnection)DynamoDBResponseError)Provider) exceptions)jsonc seZdZdZdZdZdZdZdZdZ dZ e Z d Z d5fd d ZddZddZd6ddZddZddZd7ddZddZddZd d!Zd"d#Zd8d%d&Zd9d'd(Zd:d)d*Zd;d+d,Zdd1d2Zd?d3d4Z Z!S)@Layer1ae This is the lowest-level interface to DynamoDB. Methods at this layer map directly to API requests and parameters to the methods are either simple, scalar values or they are the Python equivalent of the JSON input as defined in the DynamoDB Developer's Guide. All responses are direct decoding of the JSON response bodies to Python data structures via the json or simplejson modules. :ivar throughput_exceeded_events: An integer variable that keeps a running total of the number of ThroughputExceeded responses this connection has received from Amazon DynamoDB. z us-east-1DynamoDBZ20111205Z&ProvisionedThroughputExceededExceptionz.com.amazon.coral.service#ExpiredTokenExceptionZConditionalCheckFailedExceptionZValidationException NTrc  s| s8tjdd|j} tjD]}|j| kr |} q8q | |_tt |j |jj ||||||||| | d d|_ tj dd| |_dS)Nr region)debugsecurity_tokenvalidate_certs profile_namervalidate_checksums)botoconfiggetDefaultRegionNameZdynamodbZregionsnamer superr__init__endpointthroughput_exceeded_eventsZgetbool_validate_checksums)selfZaws_access_key_idZaws_secret_access_keyZ is_secureportproxyZ proxy_portr r r rrrZ region_namereg __class__8/tmp/pip-unpacked-wheel-d7dsrkjd/boto/dynamodb/layer1.pyrJs6  zLayer1.__init__cCst|j|_|j|jdS)N)rZ_provider_typeZproviderZ _auth_handlerZupdate_providerrr!r!r"_get_session_tokenbs zLayer1._get_session_tokencCsdgS)Nzhmac-v4r!r#r!r!r"_required_auth_capabilityfsz Layer1._required_auth_capabilityc Csd|j|j|f|jjdtt|d}|dddi||d}t}|j|d|j |j d}t|d}| d } t j d | t jd |d | t||d } t j | tj| |dS)zW :raises: ``DynamoDBExpiredTokenError`` if the security token expires. z%s_%s.%szapplication/x-amz-json-1.0) X-Amz-TargetHostz Content-TypezContent-LengthPOST/N)ZsenderZoverride_num_retriesZ retry_handlerizx-amzn-RequestIdz RequestId: %sz%s: id=%s time=%smsr'utf-8 object_hook) ServiceNameVersionr rstrlenZbuild_base_http_requesttimeZ_mexe NumberRetries_retry_handler getheaderrlogr Zperflogintreaddecoderloads) ractionbodyr-headers http_requeststartresponseelapsedZ request_id response_bodyr!r!r" make_requestis<    zLayer1.make_requestc Csd}|jdkr(|d}tj|t|}|j| dkr|j d7_ d|j|f}| |}|d7}|||f}||j krt |j|j|n|j| dkrd}||||jddf}n\|j| dkrt |j|j|n8|j| dkrt |j|j|n||j|j||d}|jr|dk rtjd |dt|d @} t|}| |krd | |f}||d| |f}|S) Nir+Z__typez%s, retry attempt %szRenewing Session Tokenrz x-amz-crc32z&Validating crc32 checksum for body: %slzAThe calculated checksum %s did not match the expected checksum %s)statusr8r9rr6r rr: ThruputErrorrr_exponential_timer3dynamodb_exceptionsZDynamoDBThroughputExceededErrorreasonSessionExpiredErrorr$Z num_retriesConditionalCheckFailedErrorZ#DynamoDBConditionalCheckFailedErrorValidationErrorZDynamoDBValidationError ResponseErrorr5rrr7) rr@i next_sleeprErBdatamsgZexpected_crc32Z actual_crc32r!r!r"r4sb          zLayer1._retry_handlercCs0|dkrd}ntdd|tjddd}|S)Nrg?ZBotoZmax_retry_delay<)minrrr)rrNrOr!r!r"rGs  zLayer1._exponential_timecCs2i}|r||d<|r||d<t|}|d|S)at Returns a dictionary of results. The dictionary contains a **TableNames** key whose value is a list of the table names. The dictionary could also contain a **LastEvaluatedTableName** key whose value would be the last table name returned if the complete list of table names was not returned. This value would then be passed as the ``start_table`` parameter on a subsequent call to this method. :type limit: int :param limit: The maximum number of tables to return. :type start_table: str :param start_table: The name of the table that starts the list. If you ran a previous list_tables and not all results were returned, the response dict would include a LastEvaluatedTableName attribute. Use that value here to continue the listing. LimitZExclusiveStartTableNameZ ListTablesrdumpsrC)rlimitZ start_tablerP json_inputr!r!r" list_tabless zLayer1.list_tablescCsd|i}t|}|d|S)z Returns information about the table including current state of the table, primary key schema and when the table was created. :type table_name: str :param table_name: The name of the table to describe. TableNameZ DescribeTablerVr table_namerPrYr!r!r"describe_tables  zLayer1.describe_tablecCs&|||d}t|}|d|}|S)a Add a new table to your account. The table name must be unique among those associated with the account issuing the request. This request triggers an asynchronous workflow to begin creating the table. When the workflow is complete, the state of the table will be ACTIVE. :type table_name: str :param table_name: The name of the table to create. :type schema: dict :param schema: A Python version of the KeySchema data structure as defined by DynamoDB :type provisioned_throughput: dict :param provisioned_throughput: A Python version of the ProvisionedThroughput data structure defined by DynamoDB. )r[Z KeySchemaProvisionedThroughputZ CreateTablerV)rr]Zschemaprovisioned_throughputrPrYZ response_dictr!r!r" create_tables  zLayer1.create_tablecCs ||d}t|}|d|S)a_ Updates the provisioned throughput for a given table. :type table_name: str :param table_name: The name of the table to update. :type provisioned_throughput: dict :param provisioned_throughput: A Python version of the ProvisionedThroughput data structure defined by DynamoDB. )r[r_Z UpdateTablerV)rr]r`rPrYr!r!r" update_tables  zLayer1.update_tablecCsd|i}t|}|d|S)a Deletes the table and all of it's data. After this request the table will be in the DELETING state until DynamoDB completes the delete operation. :type table_name: str :param table_name: The name of the table to delete. r[Z DeleteTablerVr\r!r!r" delete_table s  zLayer1.delete_tableFc CsR||d}|r||d<|r"d|d<t|}|jd||d}d|krNtd|S) a Return a set of attributes for an item that matches the supplied key. :type table_name: str :param table_name: The name of the table containing the item. :type key: dict :param key: A Python version of the Key data structure defined by DynamoDB. :type attributes_to_get: list :param attributes_to_get: A list of attribute names. If supplied, only the specified attribute names will be returned. Otherwise, all attributes will be returned. :type consistent_read: bool :param consistent_read: If True, a consistent read request is issued. Otherwise, an eventually consistent request is issued. r[KeyAttributesToGetTConsistentReadZGetItemr,ItemzKey does not exist.)rrWrCrHZDynamoDBKeyNotFoundError) rr]keyattributes_to_getconsistent_readr-rPrYr@r!r!r"get_items  zLayer1.get_itemcCs*|siSd|i}t|}|jd||dS)a  Return a set of attributes for a multiple items in multiple tables using their primary keys. :type request_items: dict :param request_items: A Python version of the RequestItems data structure defined by DynamoDB. RequestItemsZ BatchGetItemr,rVrZ request_itemsr-rPrYr!r!r"batch_get_item<s  zLayer1.batch_get_itemcCs"d|i}t|}|jd||dS)a This operation enables you to put or delete several items across multiple tables in a single API call. :type request_items: dict :param request_items: A Python version of the RequestItems data structure defined by DynamoDB. rmZBatchWriteItemr,rVrnr!r!r"batch_write_itemMs  zLayer1.batch_write_itemcCs<||d}|r||d<|r"||d<t|}|jd||dS)a Create a new item or replace an old item with a new item (including all attributes). If an item already exists in the specified table with the same primary key, the new item will completely replace the old item. You can perform a conditional put by specifying an expected rule. :type table_name: str :param table_name: The name of the table in which to put the item. :type item: dict :param item: A Python version of the Item data structure defined by DynamoDB. :type expected: dict :param expected: A Python version of the Expected data structure defined by DynamoDB. :type return_values: str :param return_values: Controls the return of attribute name-value pairs before then were changed. Possible values are: None or 'ALL_OLD'. If 'ALL_OLD' is specified and the item is overwritten, the content of the old item is returned. )r[rhExpected ReturnValuesZPutItemr,rV)rr]itemexpected return_valuesr-rPrYr!r!r"put_item[s zLayer1.put_itemc Cs>|||d}|r||d<|r$||d<t|}|jd||dS)af Edits an existing item's attributes. You can perform a conditional update (insert a new attribute name-value pair if it doesn't exist, or replace an existing name-value pair if it has certain expected attribute values). :type table_name: str :param table_name: The name of the table. :type key: dict :param key: A Python version of the Key data structure defined by DynamoDB which identifies the item to be updated. :type attribute_updates: dict :param attribute_updates: A Python version of the AttributeUpdates data structure defined by DynamoDB. :type expected: dict :param expected: A Python version of the Expected data structure defined by DynamoDB. :type return_values: str :param return_values: Controls the return of attribute name-value pairs before then were changed. Possible values are: None or 'ALL_OLD'. If 'ALL_OLD' is specified and the item is overwritten, the content of the old item is returned. )r[reZAttributeUpdatesrqrrZ UpdateItemr,rV) rr]riZattribute_updatesrtrur-rPrYr!r!r" update_items zLayer1.update_itemcCs<||d}|r||d<|r"||d<t|}|jd||dS)aK Delete an item and all of it's attributes by primary key. You can perform a conditional delete by specifying an expected rule. :type table_name: str :param table_name: The name of the table containing the item. :type key: dict :param key: A Python version of the Key data structure defined by DynamoDB. :type expected: dict :param expected: A Python version of the Expected data structure defined by DynamoDB. :type return_values: str :param return_values: Controls the return of attribute name-value pairs before then were changed. Possible values are: None or 'ALL_OLD'. If 'ALL_OLD' is specified and the item is overwritten, the content of the old item is returned. rdrqrrZ DeleteItemr,rV)rr]rirtrur-rPrYr!r!r" delete_items zLayer1.delete_itemc Cs||d} |r|| d<|r"|| d<|r.|| d<| r:d| d<|rFd| d<|rTd| d<nd | d<|rh|| d <t| } |jd | | d S) a Perform a query of DynamoDB. This version is currently punting and expecting you to provide a full and correct JSON body which is passed as is to DynamoDB. :type table_name: str :param table_name: The name of the table to query. :type hash_key_value: dict :param key: A DynamoDB-style HashKeyValue. :type range_key_conditions: dict :param range_key_conditions: A Python version of the RangeKeyConditions data structure. :type attributes_to_get: list :param attributes_to_get: A list of attribute names. If supplied, only the specified attribute names will be returned. Otherwise, all attributes will be returned. :type limit: int :param limit: The maximum number of items to return. :type count: bool :param count: If True, Amazon DynamoDB returns a total number of items for the Query operation, even if the operation has no matching items for the assigned filter. :type consistent_read: bool :param consistent_read: If True, a consistent read request is issued. Otherwise, an eventually consistent request is issued. :type scan_index_forward: bool :param scan_index_forward: Specified forward or backward traversal of the index. Default is forward (True). :type exclusive_start_key: list or tuple :param exclusive_start_key: Primary key of the item from which to continue an earlier query. This would be provided as the LastEvaluatedKey in that query. )r[Z HashKeyValueZRangeKeyConditionrfrUTCountrgZScanIndexForwardFExclusiveStartKeyZQueryr,rV) rr]Zhash_key_valueZrange_key_conditionsrjrXrkZscan_index_forwardexclusive_start_keyr-countrPrYr!r!r"querys,.  z Layer1.queryc Cs^d|i}|r||d<|r ||d<|r,||d<|r8d|d<|rD||d<t|} |jd| |d S) a Perform a scan of DynamoDB. This version is currently punting and expecting you to provide a full and correct JSON body which is passed as is to DynamoDB. :type table_name: str :param table_name: The name of the table to scan. :type scan_filter: dict :param scan_filter: A Python version of the ScanFilter data structure. :type attributes_to_get: list :param attributes_to_get: A list of attribute names. If supplied, only the specified attribute names will be returned. Otherwise, all attributes will be returned. :type limit: int :param limit: The maximum number of items to evaluate. :type count: bool :param count: If True, Amazon DynamoDB returns a total number of items for the Scan operation, even if the operation has no matching items for the assigned filter. :type exclusive_start_key: list or tuple :param exclusive_start_key: Primary key of the item from which to continue an earlier query. This would be provided as the LastEvaluatedKey in that query. r[Z ScanFilterrfrUTryrzZScanr,rV) rr]Z scan_filterrjrXr{r-r|rPrYr!r!r"scans! z Layer1.scan) NNTNNNrNNTTN)r&N)NN)NFN)N)N)NNN)NNN)NNN)NNNFTNNF)NNNNNF)"__name__ __module__ __qualname____doc__rr.r/rFrJrKrLrrMr3rr$r%rCr4rGrZr^rarbrcrlrorprvrwrxr}r~ __classcell__r!r!rr"r"s  +    &   ( + $ Dr)r2binasciirrZboto.connectionrZboto.exceptionrZ boto.providerrZ boto.dynamodbrrHZ boto.compatrrr!r!r!r"s