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)dctbasesrkvr rr set_fieldss   r cs4tddfdd|DD}|rt|dddS)Ncss|] \}}|s|VqdSNr )ris_ok error_coder r r s z*check_global_invariants..c3s|]}|VqdSr!r )r invariantsubjectr rr$sr zGlobal invariant failed)tupler)r' invariants 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!)rrrtr0r rr$)zcheck_type..z'Invalid type for field {0}.{1}, was {2})typeanyr/__name__ PTypeError)destination_clsfieldrr0 actual_typemessager r3r check_type(s r=cCs:t|turdSt|}t|dkrdStt|d|S)NTrF)r5setr(len issubclassr)type_cls field_typetypesr r r is_type_cls/s  rDcCs*|sdSt||js dSdt|jjvS)NF ignore_extra)rDr5inspect signaturefactory parameters)rAr:rEr r ris_field_ignore_extra_complaint8s  rJc@s$eZdZdZddZeddZdS)rr5r%initial mandatory_factoryr.cCs(||_||_||_||_||_||_dSr!rK)selfr5r%rLrMrHr.r r r__init__Gs  z_PField.__init__cCs@|jturt|jdkrtt|jd}t|tr|jS|jS)Nr) rNPFIELD_NO_FACTORYr?r5rr(r@rcreate)rOtypr r rrHOs  z_PField.factoryN)r7 __module__ __qualname__ __slots__rPpropertyrHr r r rrDs rr cCdS)N)TNr )_r r rZr[cCs|Sr!r )xr r rr[[r\cCs|Sr!r )rZr0r r rr[]r\Fc 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 )r5r%rLrMrHr.) rrr>r(r r PFIELD_NO_INVARIANTcallabler r_check_field_parameters) r5r%rLrMrHr.rCinvariant_functionr:r r rr:`s r:csjD]}t|tst|tstdt|qjtur.zInitial has invalid type {0}zInvariant must be callablezFactory must be callablezSerializer must be callable) r5rstr TypeErrorr/rLPFIELD_NO_INITIALr_r6r%rHr.)r:r2r rbrr`s&     r`cs eZdZdZfddZZS)r8a 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!)superr8rP source_classr:expected_typesr;)rOrgr:rhr;argskwargs __class__r rrPs  zPTypeError.__init__)r7rUrV__doc__rP __classcell__r r rkrr8s r8PVectorPSetcCt||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  rwcCsddd|DS)z4Convert a tuple of types to a human-readable string.css|] }t|jVqdSr!)rr7 capitalize)rrTr r rr$r4z"_types_to_names..)join)rCr r r_types_to_namessr|csZtf}|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!)rwrrO)rsrtr r __reduce__s z0_make_seq_field_type..TheType.__reduce__N)r7rUrV__type__ __invariant__r~r rsitem_invariantrtr rTheTypesr)rrrSEQ_FIELD_TYPE_SUFFIXESr|_checked_typesr7)rsrtrrvrsuffixr rr_make_seq_field_types rcsHt||||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_fieldsrErS)argumentrrErr rrHsz _sequence_field..factoryT)r5rHrMr%rL)NF)rrSr: optional_type)rsrtr rLr%rrHr rr_sequence_fields rcCtt|||||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%r)rrrtr rLr%rr 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)rrrr r r pvector_fieldrrcCrY)N)Tryr )itemr r rr[r\cCrq)z8Unpickling function for auto-generated PMap field types.)_pmap_field_typesr)key_type value_typerurvr r r_restore_pmap_field_picklerxrcsZtf}|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__rr}r r rr~(sz0_make_pmap_field_type..TheMap.__reduce__N)r7rUrVrrr~r rrr rTheMap$s rz {0}To{1}PMap)rrrr/r|_checked_key_types_checked_value_typesr7)rrrvrr 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!r)rrr rrHBs zpmap_field..factoryT)rMrLr5rHr%)rrSr:r)rrr r%rHr rr pmap_field3s r),pyrsistent._checked_typesrrrrrrrr r r rr rFr r+r-r=rDrJobjectrPFIELD_NO_TYPEr^rRrer,r:r`rdr8rrrrwr|rrrr_validrrrrr r r rsX,     $