U ],@slddlZddlZddlmZmZmZmZmZmZm Z m Z m Z ddlm Z ddlmZddlZejddkZddZdd Zd d Zd d ZddZddZGdddeZdZddZddZeZddZeeedeefddZddZ Gddde!Z"ed ed!iZ#iZ$d"d#Z%d$d%Z&d&d'Z'd(d)Z(d5d*d+Z)d6d,d-Z*d.dZ+iZ,d/d0Z-d1d2Z.defd3d4Z/dS)7N) CheckedPMap CheckedPSetCheckedPVector CheckedTypeInvariantException_restore_pickleget_typemaybe_parse_user_typemaybe_parse_many_user_types)optional)wrap_invariantcsVttfdd|Dg|<t|D]$\}}t|tr,|||<||=q,dS)Ncs"g|]}t|jiqS)list__dict__getitems).0bnamer:/usr/lib/python3/dist-packages/pyrsistent/_field_common.py szset_fields..)dictsumrr isinstance_PField)Zdctbasesrkvrrr set_fieldss    r cs4tddfdd|DD}|r0t|dddS)Ncss|]\}}|s|VqdSNr)rZis_okZ error_coderrr sz*check_global_invariants..c3s|]}|VqdSr!r)r invariantsubjectrrr"!srzGlobal invariant failed)tupler)r%Z invariantsZ error_codesrr$rcheck_global_invariantss r'cCs&t|tr|tkr||S|||Sr!)rrPFIELD_NO_SERIALIZER serialize) serializerformatvaluerrrr)&s r)csN|jrJtfdd|jDsJt}d|j||j}t|||j||dS)Nc3s|]}tt|VqdSr!)rrrtr,rrr".szcheck_type..z'Invalid type for field {0}.{1}, was {2})typeanyr+__name__ PTypeError)Zdestination_clsfieldrr, actual_typemessagerr/r check_type-sr7cCs:t|tkrdSt|}t|dkr(dStt|d|S)NTrF)r0setr&len issubclassr)type_clsZ field_typetypesrrr is_type_cls4s   r=cCsD|sdSt||jsdStr.dt|jjkSdt|jjkSdS)NF ignore_extra) r=r0PY2inspectZ getargspecfactoryargsZ signatureZ parameters)r;r4r>rrris_field_ignore_extra_complaint=s rCc@s$eZdZdZddZeddZdS)rr0r#initial mandatory_factoryr*cCs(||_||_||_||_||_||_dSr!rD)selfr0r#rErFrAr*rrr__init__Os z_PField.__init__cCs@|jtkr:t|jdkr:tt|jd}t|tr:|jS|jS)Nr) rGPFIELD_NO_FACTORYr9r0rr&r:rcreate)rHtyprrrrAWs  z_PField.factoryN)r2 __module__ __qualname__ __slots__rIpropertyrArrrrrLsrrcCsdS)N)TNr)_rrrbrScCs|Sr!r)xrrrrScrTcCs|Sr!r)rRr,rrrrSerTFc Csft|tttfrtt|}n tt|}|tkrBt|rBt|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#rErFrAr*) rrr8r&r r PFIELD_NO_INVARIANTcallabler r_check_field_parameters) r0r#rErFrAr*r<Zinvariant_functionr4rrrr4hs r4csjD],}t|tst|tjstdt|qjtk rztjszjrzt fddjDsztdtjtj stdtj stdtj stddS)Nz Type parameter expected, not {0}c3s|]}tj|VqdSr!)rrEr-r4rrr"sz*_check_field_parameters..zInitial has invalid type {0}zInvariant must be callablezFactory must be callablezSerializer must be callable) r0rsixZ string_types TypeErrorr+rEPFIELD_NO_INITIALrWr1r#rAr*)r4r.rrYrrXs"     rXcs 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 cs.tt|j||||_||_||_||_dSr!)superr3rI source_classr4expected_typesr5)rHr^r4r_r5rBkwargs __class__rrrIs zPTypeError.__init__)r2rNrO__doc__rI __classcell__rrrarr3s r3ZPVectorZPSetcCst||f}t||S)z=Unpickling function for auto-generated PVec/PSet field types.)_seq_field_typesr) checked_class item_typedatatype_rrr_restore_seq_field_pickles rjcCsddd|DS)z4Convert a tuple of types to a human-readable string.css|]}t|jVqdSr!)rr2 capitalize)rrMrrrr"sz"_types_to_names..)join)r<rrr_types_to_namessrncsXtf}|dk r|SGfddd}t}t|j||_|tf<|S)zFCreate a subclass of the given checked class with the given item type.NcseZdZZfddZdS)z%_make_seq_field_type..TheTypecstt|ffSr!)rjrrHrfrgrr __reduce__s z0_make_seq_field_type..TheType.__reduce__N)r2rNrOZ__type__rqrrprrTheTypesrr)rerSEQ_FIELD_TYPE_SUFFIXESrnZ_checked_typesr2)rfrgrirrsuffixrrpr_make_seq_field_types rucsBt|||rfdd}nj}t|r0tn|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. cs|dkr dS|SdSr!rLZargumentrrrrrAsz _sequence_field..factoryT)r0rArFrE)rurLr4 optional_type)rfrgr rErArrxr_sequence_fields rzcCstt|||S)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. )rzrrgr rErrr pset_fields r|cCstt|||S)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. )rzrr{rrr pvector_fields r}cCsdS)N)Trkr)itemrrrrSrTcCst||f}t||S)z8Unpickling function for auto-generated PMap field types.)_pmap_field_typesr)key_type value_typerhrirrr_restore_pmap_field_pickles rcsZtf}|dk r|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__rrorrrrq&sz0_make_pmap_field_type..TheMap.__reduce__N)r2rNrOrrrqrrrrrTheMap"srz {0}To{1}PMap)rrrr+rnZ_checked_key_typesZ_checked_value_typesr2)rrrirrrr_make_pmap_field_types rcsBt|||rfdd}nj}td|r6tn||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|dkr dS|SdSr!rvrwrrrrA@szpmap_field..factoryT)rFrEr0rAr#)rrLr4ry)rrr r#rArrr pmap_field1s r)Fr)Fr)0rZsysZpyrsistent._checked_typesrrrrrrrr r r ryr r@ version_infor?r r'r)r7r=rCobjectrZPFIELD_NO_TYPErVrKr\r(r4rXr[r3rsrerjrnrurzr|r}Z_validrrrrrrrrsT,     $