o a.@sdddlmZmZmZmZmZmZmZmZm Z ddlm Z ddlm Z ddl Z ddZddZd d Zd d Zd dZddZGdddeZdZddZddZeZddZeeedeefddZddZGdddeZeded iZiZ d!d"Z!d#d$Z"d%d&Z#eefd'd(Z$ddeefd)d*Z%ddeefd+d,Z&d-dZ'iZ(d.d/Z)d0d1Z*defd2d3Z+dS)4) CheckedPMap CheckedPSetCheckedPVector CheckedTypeInvariantException_restore_pickleget_typemaybe_parse_user_typemaybe_parse_many_user_types)optional)wrap_invariantNcsVttfdd|Dg|<t|D]\}}t|tr(|||<||=qdS)Ncs"g|] }t|jiqS)list__dict__getitems).0bnamer :/usr/lib/python3/dist-packages/pyrsistent/_field_common.py s"zset_fields..)dictsumrr isinstance_PField)Zdctbasesrkvr rr set_fieldss   rcs4tddfdd|DD}|rt|dddS)Ncss|] \}}|s|VqdSNr )rZis_okZ error_coder r r s z*check_global_invariants..c3s|]}|VqdSr r )r invariantsubjectr rr!sr zGlobal invariant failed)tupler)r$Z invariantsZ error_codesr r#rcheck_global_invariantss  r&cCs&t|tr|tur||S|||Sr )rrPFIELD_NO_SERIALIZER serialize) serializerformatvaluer r rr(!s  r(csR|jr%tfdd|jDs't}d|j||j}t|||j||dSdS)Nc3s|] }tt|VqdSr )rrrtr+r rr!)zcheck_type..z'Invalid type for field {0}.{1}, was {2})typeanyr*__name__ PTypeError)Zdestination_clsfieldrr+ actual_typemessager r.r check_type(s r7cCs:t|turdSt|}t|dkrdStt|d|S)NTrF)r0setr%len issubclassr)type_clsZ field_typetypesr r r is_type_cls/s  r=cCs*|sdSt||js dSdt|jjvS)NF ignore_extra)r=r0inspectZ signaturefactoryZ parameters)r;r4r>r r ris_field_ignore_extra_complaint8s  rAc@s$eZdZdZddZeddZdS)rr0r"initial mandatory_factoryr)cCs(||_||_||_||_||_||_dSr rB)selfr0r"rCrDr@r)r r r__init__Gs  z_PField.__init__cCs@|jturt|jdkrtt|jd}t|tr|jS|jS)Nr) rEPFIELD_NO_FACTORYr9r0rr%r:rcreate)rFtypr r rr@Os  z_PField.factoryN)r2 __module__ __qualname__ __slots__rGpropertyr@r r r rrDs rr cCdS)N)TNr )_r r rZrRcCs|Sr r )xr r rrR[rScCs|Sr r )rQr+r r rrR]rSFc Csft|tttfrtt|}ntt|}|tkr!t|r!t|n|}t ||||||d}t ||S)a Field specification factory for :py:class:`PRecord`. :param type: a type or iterable with types that are allowed for this field :param invariant: a function specifying an invariant that must hold for the field :param initial: value of field if not specified when instantiating the record :param mandatory: boolean specifying if the field is mandatory or not :param factory: function called when field is set. :param serializer: function that returns a serialized version of the field )r0r"rCrDr@r)) rrr8r%r r PFIELD_NO_INVARIANTcallabler r_check_field_parameters) r0r"rCrDr@r)r<Zinvariant_functionr4r r rr4`s r4csjD]}t|tst|tstdt|qjtur.zInitial has invalid type {0}zInvariant must be callablezFactory must be callablezSerializer must be callable) r0rstr TypeErrorr*rCPFIELD_NO_INITIALrVr1r"r@r))r4r-r rXrrWs&     rWcs eZdZdZfddZZS)r3a Raised when trying to assign a value with a type that doesn't match the declared type. Attributes: source_class -- The class of the record field -- Field name expected_types -- Types allowed for the field actual_type -- The non matching type cs2tt|j|i|||_||_||_||_dSr )superr3rG source_classr4expected_typesr5)rFr]r4r^r5argskwargs __class__r rrGs  zPTypeError.__init__)r2rLrM__doc__rG __classcell__r r rarr3s r3ZPVectorZPSetcCt||f}t||S)z=Unpickling function for auto-generated PVec/PSet field types.)_seq_field_typesr) checked_class item_typedatatype_r r r_restore_seq_field_pickle  rkcCsddd|DS)z4Convert a tuple of types to a human-readable string.css|] }t|jVqdSr )rr2 capitalize)rrKr r rr!r/z"_types_to_names..)join)r<r r r_types_to_namessrpcsZtf}|dur |SGfddd}t}t|j||_|tf<|S)zFCreate a subclass of the given checked class with the given item type.Ncs"eZdZZZfddZdS)z%_make_seq_field_type..TheTypecstt|ffSr )rkrrF)rgrhr r __reduce__s z0_make_seq_field_type..TheType.__reduce__N)r2rLrMZ__type__Z __invariant__rrr rgitem_invariantrhr rTheTypesru)rfrSEQ_FIELD_TYPE_SUFFIXESrpZ_checked_typesr2)rgrhrtrjrusuffixr rsr_make_seq_field_types rxcsHt||||rdfdd }nj}t|rtn|d|||dS)a Create checked field for either ``PSet`` or ``PVector``. :param checked_class: ``CheckedPSet`` or ``CheckedPVector``. :param item_type: The required type for the items in the set. :param optional: If true, ``None`` can be used as a value for this field. :param initial: Initial value to pass to factory. :return: A ``field`` containing a checked class. NFcs|durdSj|||dS)N)_factory_fieldsr>rJ)argumentryr>rur rr@sz _sequence_field..factoryT)r0r@rDr"rC)NF)rxrJr4 optional_type)rgrhr rCr"rtr@r r|r_sequence_fields r~cCtt|||||dS)al Create checked ``PSet`` field. :param item_type: The required type for the items in the set. :param optional: If true, ``None`` can be used as a value for this field. :param initial: Initial value to pass to factory if no value is given for the field. :return: A ``field`` containing a ``CheckedPSet`` of the given type. r"rt)r~rrhr rCr"rtr r r pset_field rcCr)au Create checked ``PVector`` field. :param item_type: The required type for the items in the vector. :param optional: If true, ``None`` can be used as a value for this field. :param initial: Initial value to pass to factory if no value is given for the field. :return: A ``field`` containing a ``CheckedPVector`` of the given type. r)r~rrr r r pvector_fieldrrcCrP)N)Trmr )itemr r rrRrScCre)z8Unpickling function for auto-generated PMap field types.)_pmap_field_typesr)key_type value_typerirjr r r_restore_pmap_field_picklerlrcsZtf}|dur |SGfdddt}dt|jt|j|_|tf<|S)zDCreate a subclass of CheckedPMap with the given key and value types.NcseZdZZZddZdS)z%_make_pmap_field_type..TheMapcSst|j|jt|ffSr )r __key_type____value_type__rrqr r rrr(sz0_make_pmap_field_type..TheMap.__reduce__N)r2rLrMrrrrr rrr rTheMap$s rz {0}To{1}PMap)rrrr*rpZ_checked_key_typesZ_checked_value_typesr2)rrrjrr rr_make_pmap_field_types rcsBt|||rfdd}nj}td|rtn||dS)ad Create a checked ``PMap`` field. :param key: The required type for the keys of the map. :param value: The required type for the values of the map. :param optional: If true, ``None`` can be used as a value for this field. :param invariant: Pass-through to ``field``. :return: A ``field`` containing a ``CheckedPMap``. cs|durdS|Sr rz)r{rr rr@Bs zpmap_field..factoryT)rDrCr0r@r")rrJr4r})rrr r"r@r rr pmap_field3s r),Zpyrsistent._checked_typesrrrrrrrr r r r}r r?rr&r(r7r=rAobjectrZPFIELD_NO_TYPErUrIr[r'r4rWrZr3rvrfrkrprxr~rrZ_validrrrrr r r rsX,     $