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_invariantsZ __invariant__)_pclass_frozencss|]}|VqdSNr.0keyrrr sz%PClassMeta.__new__.. __slots__) __weakref__)rrtuplersuperr__new__)ZmcsrrZdct __class__rrr" s zPClassMeta.__new__)__name__ __module__ __qualname__r" __classcell__rrr#rr srcCs>t||||||\}}|s||dSt|||dSr)rZ invariantappendsetattr)clsfieldrvalueresultinvariant_errorsZis_okZ error_coderrr_check_and_set_attrs r0cseZdZdZfddZddZed"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)r3z{0}.{1}zField invariant failedz0'{0}' are not among the specified fields for {1}, T)r!r1r"popritemsrfactoryr0initialr callableZ mandatoryr)formatr%rr AttributeErrorjoinr rr) r+kwargsr.factory_fieldsr3Zmissing_fieldsr/rr,r-r8r#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) rrr2Nr)setrgetattr_MISSING_VALUEr$)selfargsr=r>rr-rrrr?Rs  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)rkr=rr sz!PClass.create..)r2r3Nr) isinstancer)r+r=r2r3rrErcreaters 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. )rr@rAr Z serializer)rBr:r.rr-rrrr 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 )rBZtransformationsrrrr s zPClass.transformcCs>t||jr|jD]}t||tt||tkrdSq dStS)NFT)rGr$rr@rANotImplemented)rBotherrrrr__eq__s  z PClass.__eq__cCs ||k Srr)rBrJrrr__ne__ z PClass.__ne__csttfddjDS)Nc3s |] }|t|tfVqdSr)r@rArrBrrrsz"PClass.__hash__..)hashr rrNrrNr__hash__szPClass.__hash__cs2t|ddrtd||tt|||dS)NrFz'Can't set attribute, key={0}, value={1})r@r;r:r!r1 __setattr__rBrr-r#rrrQs zPClass.__setattr__cCstd|)Nz-Can't delete attribute, key={0}, use remove())r;r:)rBrrrr __delattr__zPClass.__delattr__cCs0i}|jD]}t||t}|tur|||<q|Sr)rr@rA)rBr.rr-rrr_to_dicts  zPClass._to_dictcCs(d|jjddd|DS)Nz{0}({1})r4css$|] \}}d|t|VqdS)z{0}={1}N)r:repr)rrDvrrrrs"z"PClass.__repr__..)r:r$r%r<rUr6rNrrr__repr__s zPClass.__repr__cs&tfddjD}tj|ffS)Nc3s(|]}t|r|t|fVqdSr)hasattrr@rrNrrrs&z$PClass.__reduce__..)dictrrr$)rBdatarrNr __reduce__szPClass.__reduce__cCst||S)z5 Returns an evolver for this object. )_PClassEvolverrUrNrrrevolverszPClass.evolvercCs|}||=|S)z Remove attribute given by name from the current instance. Raises AttributeError if the attribute doesn't exist. )r^ persistent)rBrr^rrrremovesz PClass.removeNFr)r%r&r'__doc__r"r? classmethodrHr r rKrLrPrQrSrUrXr\r^r`r(rrr#rr1%s$ $       r1) metaclasscs`eZdZdZddZddZddZdd Zd d Zd d Z ddZ fddZ ddZ Z S)r])_pclass_evolver_original_pclass_evolver_data_pclass_evolver_data_is_dirtyr2cCs||_||_d|_t|_dSra)rerfrgr?r2)rBZoriginalZ initial_dictrrr__init__s z_PClassEvolver.__init__cCs |j|Sr)rfrBitemrrr __getitem__rMz_PClassEvolver.__getitem__cCs2|j|t|ur||j|<|j|d|_|SNT)rfgetrAr2addrgrRrrrr?s   z_PClassEvolver.setcCs|||dSr)r?rRrrr __setitem__sz_PClassEvolver.__setitem__cCs0||jvr|j|=|j|d|_|St|rl)rfr2discardrgr;rirrrr`s  z_PClassEvolver.removecCs||dSr)r`rirrr __delitem__rTz_PClassEvolver.__delitem__cCs&|jr|jjdd|ji|jS|jS)Nr2r)rgrer$r2rfrNrrrr_s z_PClassEvolver.persistentcs0||jvr |||dStt|||dSr)rr?r!r]rQrRr#rrrQs z_PClassEvolver.__setattr__cCs||Srrrirrr __getattr__sz_PClassEvolver.__getattr__)r%r&r'rrhrkr?ror`rqr_rQrrr(rrr#rr]s  r]N)Zpyrsistent._checked_typesrrrrZpyrsistent._field_commonrrrr r r Zpyrsistent._transformationsr rtyperobjectrAr0r1r]rrrrs    1