U .b)1@sddlZddlmZddlmZddlmZmZddlm Z m Z ddZ dd Z Gd d d Z Gd d d ZGdddZGdddZdS)N)collections_abc)DocumentModifiedShape) ConditionBaseConditionExpressionBuilder)TypeDeserializerTypeSerializercKs|dtdS)Nr)insertDynamoDBHighLevelResource)Z base_classeskwargsr `.z`'string'|123|Binary(b'bytes')|True|None|set(['string'])|set([123])|set([Binary(b'bytes')])|[]|{})new_typeZnew_descriptionZnew_example_value KeyExpressionz?condition from :py:class:`boto3.dynamodb.conditions.Key` methodzThe condition(s) a key(s) must meet. Valid conditions are listed in the :ref:`DynamoDB Reference Guide`.zKey('mykey').eq('myvalue')ConditionExpressionz@condition from :py:class:`boto3.dynamodb.conditions.Attr` methodzThe condition(s) an attribute(s) must meet. Valid conditions are listed in the :ref:`DynamoDB Reference Guide`.z!Attr('myattribute').eq('myvalue')z"docs.*.dynamodb.*.complete-sectionzdynamodb-attr-value-docszdynamodb-key-expression-docszdynamodb-cond-expression-docs)super__init__metaclienteventsregisterrTransformationInjectorZ _injectorinject_condition_expressionsinject_attribute_value_inputinject_attribute_value_outputrZ(replace_documentation_for_matching_shape)selfargsr Zattr_value_shape_docsZkey_expression_shape_docsZcon_expression_shape_docs __class__r r rsn       z"DynamoDBHighLevelResource.__init__)__name__ __module__ __qualname__r __classcell__r r r#r r sr c@s2eZdZdZd ddZddZddZd d ZdS) rz>Injects the transformations into the user provided parameters.NcCs\||_|dkrt|_||_|dkr,t|_||_|dkrBt|_||_|dkrXt|_dSr) _transformerParameterTransformer_condition_builderr _serializerr _deserializerr)r!Z transformercondition_builder serializerZ deserializerr r r rszTransformationInjector.__init__c Ks|ji}i}t|j||dd}|j||j|dt|j||dd}|j||j|dd}d}||kr~|||n |r|||<||kr|||n |r|||<dS) a7Injects the condition expression transformation into the parameters This injection includes transformations for ConditionExpression shapes and KeyExpression shapes. It also handles any placeholder names and values that are generated when transforming the condition expressions. F)placeholder_namesplaceholder_valuesis_key_conditionrTrZExpressionAttributeNamesZExpressionAttributeValuesN)r+reset!ConditionExpressionTransformationr) transform input_shapeupdate) r!rmodelr Zgenerated_namesZgenerated_valuestransformationZexpr_attr_names_inputZexpr_attr_values_inputr r r rsJ z3TransformationInjector.inject_condition_expressionscKs|j||j|jjddS)z3Injects DynamoDB serialization into parameter inputrN)r)r5r6r, serialize)r!rr8r r r r rs z3TransformationInjector.inject_attribute_value_inputcKs&|jdk r"|j||j|jjddS)z/Injects DynamoDB deserialization into responsesNr)Z output_shaper)r5r-Z deserialize)r!parsedr8r r r r r s z4TransformationInjector.inject_attribute_value_output)NNNN)r%r&r'__doc__rrrr r r r r rs 2 rc@s"eZdZdZdddZddZdS) r4zProvides a transformation for condition expressions The ``ParameterTransformer`` class can call this class directly to transform the condition expressions in the parameters provided. FcCs||_||_||_||_dSr)r+_placeholder_names_placeholder_values_is_key_condition)r!r.r0r1r2r r r rsz*ConditionExpressionTransformation.__init__cCsBt|tr>|jj||jd}|j|j|j|j |j S|S)N)r2) isinstancerr+Zbuild_expressionr?r=r7Zattribute_name_placeholdersr>Zattribute_value_placeholdersZcondition_expression)r!valueZbuilt_expressionr r r __call__s z*ConditionExpressionTransformation.__call__N)F)r%r&r'r<rrBr r r r r4s r4c@s8eZdZdZddZddZddZdd Zd d Zd S) r*z?Transforms the input to and output from botocore based on shapecCs|||||dS)aTransforms the dynamodb input to or output from botocore It applies a specified transformation whenever a specific shape name is encountered while traversing the parameters in the dictionary. :param params: The parameters structure to transform. :param model: The operation model. :param transformation: The function to apply the parameter :param target_shape: The name of the shape to apply the transformation to N)_transform_parameters)r!rr8r9 target_shaper r r r5 s zParameterTransformer.transformcCs,|j}|dkr(t|d|||||dS)N)Z structuremaplistZ _transform_) type_namegetattr)r!r8rr9rDrGr r r rCsz*ParameterTransformer._transform_parameterscCsft|tjsdS|D]L}||jkr|j|}|j}||krL|||||<q||||||qdSr)r@rMappingmembersnamerC)r!r8rr9rDparam member_model member_shaper r r _transform_structure"s   z)ParameterTransformer._transform_structurec Cs\t|tjsdS|j}|j}|D]2\}}||krB||||<q$||||||q$dSr)r@rrIrArKitemsrC) r!r8rr9rDZ value_modelZ value_shapekeyrAr r r _transform_map5s z#ParameterTransformer._transform_mapc Cs\t|tjsdS|j}|j}t|D]2\}}||krB||||<q$||||||q$dSr)r@rMutableSequencememberrK enumeraterC) r!r8rr9rDrMrNiitemr r r _transform_listBs z$ParameterTransformer._transform_listN) r%r&r'r<r5rCrOrRrXr r r r r*s   r*)rZ boto3.compatrZboto3.docs.utilsrZboto3.dynamodb.conditionsrrZboto3.dynamodb.typesrrr rr rr4r*r r r r  s  d`'