o ӷ6i@sddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl mZddlmZiZeZ Gd d d ejZGd d d Zd dZdddZdS)N) b64decode b64encode)checks)models) force_bytes) gettext_lazy)SignatureExpired) FernetBytescseZdZdZedZddgZdZfddZdd Z d d Z fd d Z ddZ fddZ fddZfddZdfdd ZddZddZddZZS) PickledFieldz- A field for storing pickled objects z Pickled dataN)exactinisnullcsd|d<tj|i|dS)NFeditable)super__init__selfargskwargs __class___/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/django_cryptography/fields.pyrszPickledField.__init__cC t|SN)pickledumpsrvaluerrr_dump zPickledField._dumpcCrr)rloadsrrrr_load"r!zPickledField._loadcs$t\}}}}|d=||||fS)Nr)r deconstructrnamepathrrrrrr$%s zPickledField.deconstructcCdSN BinaryFieldrrrrrget_internal_type*zPickledField.get_internal_typecst}|dkr dS|S)Nr )r get_default)rdefaultrrrr/-s zPickledField.get_defaultc||jvrdSt|Srsupported_lookupsr get_lookupr lookup_namerrrr43  zPickledField.get_lookupcr1rr3r get_transformr5rrrr98r7zPickledField.get_transformFcs.t|||}|dur|j||S|Sr)rget_db_prep_valueDatabaseBinaryr rr connectionpreparedrrrr:=szPickledField.get_db_prep_valuecO|dur |t|S|Srr#rrrrrrrr from_db_valueCzPickledField.from_db_valuecCs||}t||dS)z$Pickled data is serialized as base64ascii)value_from_objectrr decode)robjrrrrvalue_to_stringHs zPickledField.value_to_stringcCs t|tr|tt|S|Sr) isinstancestrr#rrrrrr to_pythonMs zPickledField.to_pythonF)__name__ __module__ __qualname____doc___ description empty_valuesr3rr r#r$r,r/r4r9r:rCrIrL __classcell__rrrrr s"     r cseZdZdZdZfddZefddZddZd d Z fd d Z fd dZ fddZ fddZ fddZddZdddZejjZddZZS)EncryptedMixina A field mixin storing encrypted data :param bytes key: This is an optional argument. Allows for specifying an instance specific encryption key. :param int ttl: This is an optional argument. The amount of time in seconds that a value can be stored for. If the time to live of the data has passed, it will become unreadable. The expired value will return an :class:`Expired` object. )rcs>|dd|_|dd|_t|j|_tj|i|dS)Nkeyttl)poprWrXr _fernetrrrrrrrds zEncryptedMixin.__init__cstdtjS)Nz Encrypted %s)rRrrSr+rrrrSkszEncryptedMixin.descriptioncCs|jt|Sr)rZencryptrrrrrrr oszEncryptedMixin._dumpcCs0z t|j||jWStytYSwr)rr"rZdecryptrXrExpiredrrrrr#rs  zEncryptedMixin._loadc s:tjdi|}t|ddr|tjdd|dd|S)N remote_fieldz3Base field for encrypted cannot be a related field.zencrypted.E002)hintrHidr)rcheckgetattrappendrError)rrerrorsrrrraxs zEncryptedMixin.checkcsPt\}}}}|jjtdks t|j|i||j|j S|j|i|S)N) rr$r__mro__indexrVr[ base_classrWrXr%rrrcloneszEncryptedMixin.clonecslt\}}}}|jjtdks0tjdtj}|j |i|g}i}|j dur0|j |d<||||fS)Nrf.rX) rr$rrgrhrVr[rOrNrirXr%rrrr$s   zEncryptedMixin.deconstructcr1rr2r5rrrr4r7zEncryptedMixin.get_lookupcr1rr8r5rrrr9r7zEncryptedMixin.get_transformcCr(r)rr+rrrr,r-z EncryptedMixin.get_internal_typeFcCs0tj||||}|dur|j||S|Sr)rFieldr:r;r<r r=rrrr:sz EncryptedMixin.get_db_prep_valuecOr@rrArBrrrrCrDzEncryptedMixin.from_db_valuerM)rNrOrPrQr3rpropertyrSr r#rarjr$r4r9r,r:rrlget_db_prep_saverCrUrrrrrVTs"     rVcCsBt|tjrJd|j}|tvrt|t|fd|it|<t|S)a A get or create method for encrypted fields, we cache the field in the module to avoid recreation. This also allows us to always return the same class reference for a field. :type base_class: models.Field[T] :rtype: models.Field[EncryptedMixin, T] Encryptedri)rJrrlrN FIELD_CACHEtyperV)ri field_namerrrget_encrypted_fields rscCsbt|tjs|dus J|dusJt|S|\}}}}|||dtt||i|S)a A decorator for creating encrypted model fields. :type base_field: models.Field[T] :param bytes key: This is an optional argument. Allows for specifying an instance specific encryption key. :param int ttl: This is an optional argument. The amount of time in seconds that a value can be stored for. If the time to live of the data has passed, it will become unreadable. The expired value will return an :class:`Expired` object. :rtype: models.Field[EncryptedMixin, T] N)rWrX)rJrrlrsr$updaterq) base_fieldrWrXr&r'rrrrrr[s   r[)NN)rbase64rr django.corer django.dbrdjango.utils.encodingrdjango.utils.translationrrR django_cryptography.core.signingr django_cryptography.utils.cryptor rpobjectr]rlr rVrsr[rrrrs      B^