o a%@sddlmZmZmZmZddlmZmZmZm Z m Z m Z ddl m Z ddZGdddeZeZdd ZGd d d eed ZGd ddeZdS))InvariantException CheckedType_restore_picklestore_invariants) set_fields check_typeis_field_ignore_extra_complaintPFIELD_NO_INITIAL serializecheck_global_invariants transformcCst|dko |dtkS)Nr)lenr)basesr4/usr/lib/python3/dist-packages/pyrsistent/_pclass.py _is_pclasssrcseZdZfddZZS) PClassMetacsht||ddt||dddtdd|dD|d<t|r)|dd 7<tt|||||S) N_pclass_fields)name_pclass_invariants __invariant__)_pclass_frozencss|]}|VqdSNr.0keyrrr sz%PClassMeta.__new__.. __slots__) __weakref__)rrtuplersuperr__new__)mcsrrdct __class__rrr# s zPClassMeta.__new__)__name__ __module__ __qualname__r# __classcell__rrr&rr srcCs>t||||||\}}|s||dSt|||dSr)r invariantappendsetattr)clsfieldrvalueresultinvariant_errorsis_ok error_coderrr_check_and_set_attrs r6cseZdZdZfddZddZed"dd Zd#d d Zd d Z ddZ ddZ ddZ fddZ ddZddZddZddZddZd d!ZZS)$PClassa A PClass is a python class with a fixed set of specified fields. PClasses are declared as python classes inheriting from PClass. It is defined the same way that PRecords are and behaves like a PRecord in all aspects except that it is not a PMap and hence not a collection but rather a plain Python object. More documentation and examples of PClass usage is available at https://github.com/tobgu/pyrsistent c  sLtt||}|dd}|dd}g}g}|jD]a\}}||vrV|dus-||vrEtt||r=|j|||d} n |||} n||} t|||| ||||=q|j t urqt |j rd| n|j } t|||| ||q|j r~| d|j|q|s|rtt|t|d|rtdd||jt||jd|_|S) N_factory_fields ignore_extra)r9z{0}.{1}zField invariant failedz0'{0}' are not among the specified fields for {1}, T)r"r7r#popritemsrfactoryr6initialr callable mandatoryr-formatr(rr!AttributeErrorjoinr rr) r/kwargsr2factory_fieldsr9missing_fieldsr3rr0r1r>r&rrr#.s>       zPClass.__new__cOs`|r |d||d<t|}|jD]}||vr%t||t}|tur%|||<q|jdd|i|S)a Set a field in the instance. Returns a new instance with the updated value. The original instance remains unmodified. Accepts key-value pairs or single string representing the field name and a value. >>> from pyrsistent import PClass, field >>> class AClass(PClass): ... x = field() ... >>> a = AClass(x=1) >>> a2 = a.set(x=2) >>> a3 = a.set('x', 3) >>> a AClass(x=1) >>> a2 AClass(x=2) >>> a3 AClass(x=3) rrr8Nr)setrgetattr_MISSING_VALUEr')selfargsrDrErr1rrrrGRs  z PClass.setNFcs:t|rS|rfdd|jD|d||dS)a: Factory method. Will create a new PClass of the current type and assign the values specified in kwargs. :param ignore_extra: A boolean which when set to True will ignore any keys which appear in kwargs that are not in the set of fields on the PClass. csi|] }|vr||qSrr)rkrDrr sz!PClass.create..)r8r9Nr) isinstancer)r/rDr8r9rrMrcreaters z PClass.createcCs@i}|jD]}t||t}|turt|j|j||||<q|S)z Serialize the current PClass using custom serializer functions for fields where such have been supplied. )rrHrIr serializer)rJrAr2rr1rrrr s  zPClass.serializecGs t||S)aT Apply transformations to the currency PClass. For more details on transformations see the documentation for PMap. Transformations on PClasses do not support key matching since the PClass is not a collection. Apart from that the transformations available for other persistent types work as expected. r )rJtransformationsrrrr s zPClass.transformcCs>t||jr|jD]}t||tt||tkrdSq dStS)NFT)rOr'rrHrINotImplemented)rJotherrrrr__eq__s  z PClass.__eq__cCs ||k Srr)rJrTrrr__ne__ z PClass.__ne__csttfddjDS)Nc3s |] }|t|tfVqdSr)rHrIrrJrrrsz"PClass.__hash__..)hashr!rrXrrXr__hash__szPClass.__hash__cs2t|ddrtd||tt|||dS)NrFz'Can't set attribute, key={0}, value={1})rHrBrAr"r7 __setattr__rJrr1r&rrr[s zPClass.__setattr__cCstd|)Nz-Can't delete attribute, key={0}, use remove())rBrA)rJrrrr __delattr__zPClass.__delattr__cCs0i}|jD]}t||t}|tur|||<q|Sr)rrHrI)rJr2rr1rrr_to_dicts  zPClass._to_dictcCs(d|jjddd|DS)Nz{0}({1})r:css$|] \}}d|t|VqdS)z{0}={1}N)rArepr)rrLvrrrrs"z"PClass.__repr__..)rAr'r(rCr_r<rXrrr__repr__s zPClass.__repr__cs&tfddjD}tj|ffS)Nc3s(|]}t|r|t|fVqdSr)hasattrrHrrXrrrs&z$PClass.__reduce__..)dictrrr')rJdatarrXr __reduce__szPClass.__reduce__cCst||S)z5 Returns an evolver for this object. )_PClassEvolverr_rXrrrevolverszPClass.evolvercCs|}||=|S)z Remove attribute given by name from the current instance. Raises AttributeError if the attribute doesn't exist. )rh persistent)rJrrhrrrremovesz PClass.removeNFr)r(r)r*__doc__r#rG classmethodrPr r rUrVrZr[r]r_rbrfrhrjr+rrr&rr7%s$ $       r7) metaclasscs`eZdZdZddZddZddZdd Zd d Zd d Z ddZ fddZ ddZ Z S)rg)_pclass_evolver_original_pclass_evolver_data_pclass_evolver_data_is_dirtyr8cCs||_||_d|_t|_dSrk)rorprqrGr8)rJoriginal initial_dictrrr__init__s z_PClassEvolver.__init__cCs |j|Sr)rprJitemrrr __getitem__rWz_PClassEvolver.__getitem__cCs2|j|t|ur||j|<|j|d|_|SNT)rpgetrIr8addrqr\rrrrGs   z_PClassEvolver.setcCs|||dSr)rGr\rrr __setitem__sz_PClassEvolver.__setitem__cCs0||jvr|j|=|j|d|_|St|rx)rpr8discardrqrBrurrrrjs  z_PClassEvolver.removecCs||dSr)rjrurrr __delitem__r^z_PClassEvolver.__delitem__cCs&|jr|jjdd|ji|jS|jS)Nr8r)rqror'r8rprXrrrris z_PClassEvolver.persistentcs0||jvr |||dStt|||dSr)rrGr"rgr[r\r&rrr[s z_PClassEvolver.__setattr__cCs||Srrrurrr __getattr__sz_PClassEvolver.__getattr__)r(r)r*rrtrwrGr{rjr}rir[r~r+rrr&rrgs  rgN)pyrsistent._checked_typesrrrrpyrsistent._field_commonrrrr r r pyrsistent._transformationsr rtyperobjectrIr6r7rgrrrrs    1