U .b:@sddlZddlmZddlmZmZmZedZGdddZ GdddZ Gd d d e e Z Gd d d e Z Gd dde Z Gddde ZGddde ZGddde ZGddde ZGddde ZGddde ZGddde ZGddde ZGdd d e ZGd!d"d"e ZGd#d$d$e ZGd%d&d&e ZGd'd(d(e ZGd)d*d*e ZGd+d,d,e ZGd-d.d.e ZGd/d0d0e ZGd1d2d2e Zed3d4d5d6gZ Gd7d8d8Z!dS)9N) namedtuple)DynamoDBNeedsConditionErrorDynamoDBNeedsKeyConditionError"DynamoDBOperationNotSupportedErrorz[^.\[\]]+(?![^\[]*\])c@sPeZdZdZdZdZddZddZddZd d Z d d Z d dZ ddZ dS) ConditionBaseFcGs ||_dSN)_valuesselfvaluesr =/tmp/pip-unpacked-wheel-7h_bjh9e/boto3/dynamodb/conditions.py__init__szConditionBase.__init__cCst|tstd|t||SNAND) isinstancerrAndr otherr r r__and__"s  zConditionBase.__and__cCst|tstd|t||SNOR)rrrOrrr r r__or__'s  zConditionBase.__or__cCst|Sr)Notr r r r __invert__,szConditionBase.__invert__cCs|j|j|jdS)N)formatoperatorr )expression_formatexpression_operatorr rr r rget_expression/szConditionBase.get_expressioncCs"t|t|r|j|jkrdSdS)NTF)rtyper rr r r__eq__6s zConditionBase.__eq__cCs || Srr$rr r r__ne__<szConditionBase.__ne__N) __name__ __module__ __qualname__r r!has_grouped_valuesrrrrr"r$r&r r r rrsrc@steZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZdS) AttributeBasecCs ||_dSr)name)r r,r r rrAszAttributeBase.__init__cCstd|dSrrr valuer r rrDszAttributeBase.__and__cCstd|dSrr-r.r r rrGszAttributeBase.__or__cCstd|dS)NNOTr-rr r rrJszAttributeBase.__invert__cCs t||S)zCreates a condition where the attribute is equal to the value. :param value: The value that the attribute is equal to. )Equalsr.r r reqMszAttributeBase.eqcCs t||S)zCreates a condition where the attribute is less than the value. :param value: The value that the attribute is less than. )LessThanr.r r rltTszAttributeBase.ltcCs t||S)zCreates a condition where the attribute is less than or equal to the value. :param value: The value that the attribute is less than or equal to. )LessThanEqualsr.r r rlte[szAttributeBase.ltecCs t||S)zCreates a condition where the attribute is greater than the value. :param value: The value that the attribute is greater than. ) GreaterThanr.r r rgtcszAttributeBase.gtcCs t||S)zCreates a condition where the attribute is greater than or equal to the value. :param value: The value that the attribute is greater than or equal to. )GreaterThanEqualsr.r r rgtejszAttributeBase.gtecCs t||S)zCreates a condition where the attribute begins with the value. :param value: The value that the attribute begins with. ) BeginsWithr.r r r begins_withrszAttributeBase.begins_withcCs t|||S)a3Creates a condition where the attribute is greater than or equal to the low value and less than or equal to the high value. :param low_value: The value that the attribute is greater than or equal to. :param high_value: The value that the attribute is less than or equal to. )Between)r Z low_valueZ high_valuer r rbetweenyszAttributeBase.betweencCst|t|o|j|jkSr)rr#r,rr r rr$szAttributeBase.__eq__cCs || Srr%rr r rr&szAttributeBase.__ne__N)r'r(r)rrrrr2r4r6r8r:r<r>r$r&r r r rr+@s r+c@s(eZdZdZddZddZddZdS) ConditionAttributeBasezThis base class is for conditions that can have attribute methods. One example is the Size condition. To complete a condition, you need to apply another AttributeBase method like eq(). cGs&tj|f|t||djdS)Nr)rrr+r,r r r rrszConditionAttributeBase.__init__cCst||ot||Sr)rr$r+rr r rr$szConditionAttributeBase.__eq__cCs || Srr%rr r rr&szConditionAttributeBase.__ne__N)r'r(r)__doc__rr$r&r r r rr?sr?c@seZdZdZdS)ComparisonConditionz{0} {operator} {1}N)r'r(r)r r r r rrAsrAc@seZdZdZdS)r1=Nr'r(r)r!r r r rr1sr1c@seZdZdZdS) NotEqualsz<>NrCr r r rrDsrDc@seZdZdZdS)r3NrCr r r rr7sr7c@seZdZdZdS)r9z>=NrCr r r rr9sr9c@seZdZdZdZdS)InINTN)r'r(r)r!r*r r r rrGsrGc@seZdZdZdZdS)r=ZBETWEENz{0} {operator} {1} AND {2}Nr'r(r)r!r r r r rr=sr=c@seZdZdZdZdS)r;r<{operator}({0}, {1})NrIr r r rr;sr;c@seZdZdZdZdS)ContainscontainsrJNrIr r r rrKsrKc@seZdZdZdZdS)Sizesize{operator}({0})NrIr r r rrMsrMc@seZdZdZdZdS) AttributeTypeattribute_typerJNrIr r r rrPsrPc@seZdZdZdZdS)AttributeExistsZattribute_existsrONrIr r r rrRsrRc@seZdZdZdZdS)AttributeNotExistsZattribute_not_existsrONrIr r r rrSsrSc@seZdZdZdZdS)rr({0} {operator} {1})NrIr r r rrsrc@seZdZdZdZdS)rrrTNrIr r r rrsrc@seZdZdZdZdS)rr0z({operator} {0})NrIr r r rrsrc@s eZdZdS)KeyN)r'r(r)r r r rrUsrUc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)Attrz(Represents an DynamoDB item's attribute.cCs t||S)zCreates a condition where the attribute is not equal to the value :param value: The value that the attribute is not equal to. )rDr.r r rneszAttr.necCs t||S)zCreates a condition where the attribute is in the value, :type value: list :param value: The value that the attribute is in. )rGr.r r ris_insz Attr.is_incCst|S)z/Creates a condition where the attribute exists.)rRrr r rexistssz Attr.existscCst|S)z7Creates a condition where the attribute does not exist.)rSrr r r not_exists szAttr.not_existscCs t||S)z}Creates a condition where the attribute contains the value. :param value: The value the attribute contains. )rKr.r r rrLsz Attr.containscCst|S)zCreates a condition for the attribute size. Note another AttributeBase method must be called on the returned size condition to be a valid DynamoDB condition. )rMrr r rrNsz Attr.sizecCs t||S)zfCreates a condition for the attribute type. :param value: The type of the attribute. )rPr.r r rrQszAttr.attribute_typeN) r'r(r)r@rWrXrYrZrLrNrQr r r rrVsrVBuiltConditionExpressioncondition_expressionattribute_name_placeholdersattribute_value_placeholdersc@s\eZdZdZddZddZddZdd Zdd d Zd dZ ddZ ddZ dddZ dS)ConditionExpressionBuilderzCThis class is used to build condition expressions with placeholderscCsd|_d|_d|_d|_dS)Nrnv) _name_count _value_count_name_placeholder_value_placeholderrr r rr4sz#ConditionExpressionBuilder.__init__cCsd|jt|jS)N#)rdstrrbrr r r_get_name_placeholder:sz0ConditionExpressionBuilder._get_name_placeholdercCsd|jt|jS)N:)rergrcrr r r_get_value_placeholder=sz1ConditionExpressionBuilder._get_value_placeholdercCsd|_d|_dS)z&Resets the placeholder name and valuesrN)rbrcrr r rreset@sz ConditionExpressionBuilder.resetFcCs:t|tst|i}i}|j||||d}t|||dS)aBuilds the condition expression and the dictionary of placeholders. :type condition: ConditionBase :param condition: A condition to be built into a condition expression string with any necessary placeholders. :type is_key_condition: Boolean :param is_key_condition: True if the expression is for a KeyConditionExpression. False otherwise. :rtype: (string, dict, dict) :returns: Will return a string representing the condition with placeholders inserted where necessary, a dictionary of placeholders for attribute names, and a dictionary of placeholders for attribute values. Here is a sample return value: ('#n0 = :v0', {'#n0': 'myattribute'}, {':v1': 'myvalue'}) )is_key_condition)r\r]r^)rrr_build_expressionr[)r conditionrlr]r^r\r r rbuild_expressionEs z+ConditionExpressionBuilder.build_expressionc CsP|}g}|dD]"}|||||j|}||q|dj|d|diS)Nr rr)r"_build_expression_componentr*appendr) r rnr]r^rlZexpression_dictZreplaced_valuesr/Zreplaced_valuer r rrmhs   z,ConditionExpressionBuilder._build_expressioncCslt|tr|||||St|trZ|rNt|tsNtd|jdt|d|||S| |||SdS)NzAttribute object z is of type zD. KeyConditionExpression only supports Attribute objects of type Key) rrrmr+rUrr,r#_build_name_placeholder_build_value_placeholder)r r/r]r^r*rlr r rrps*  z6ConditionExpressionBuilder._build_expression_componentc Cs^|j}t|}td|}g}|D],}|}|jd7_|||||<q$|t|S)Nz%s)r,ATTR_NAME_REGEXfindallsubrhrbrqtuple) r r/r]Zattribute_nameZattribute_name_partsZplaceholder_formatZstr_format_argspartZname_placeholderr r rrrs    z2ConditionExpressionBuilder._build_name_placeholdercCsr|rLg}|D],}|}|jd7_|||||<q dd|dS|}|jd7_|||<|SdS)Nrt(z, ))rjrcrqjoin)r r/r^r*Zplaceholder_listraZvalue_placeholderr r rrss  z3ConditionExpressionBuilder._build_value_placeholderN)F)F) r'r(r)r@rrhrjrkrormrprrrsr r r rr_1s #%r_)"re collectionsrZboto3.exceptionsrrrcompilerurr+r?rAr1rDr3r5r7r9rGr=r;rKrMrPrRrSrrrrUrVr[r_r r r r sD  'I1