R6W . 4 @ s y d d l Z Wn e k
r0 d d l Z Yn Xd d l Z d d l Z e j d d k Z y d d l m Z WnH e k
r e j d d k r d d l m Z n d d l m Z Yn Xy d d l
Z
Wn e k
r d Z
Yn Xy d d l Z Wn e k
rd Z Yn Xy$ d d l m
Z
d d l m Z Wn e k
rSd Z
Z Yn Xy d d l m Z m Z Wn e k
rd Z Z Yn Xd d l Td d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z e r e r Gd
d d e Z Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z! Gd d d e Z" Gd d d e Z# e
re rGd d d e Z$ d S) N )StringIO)AES)Random)hashpwgensalt)*)binary_construct)Field)FieldDescriptor) SelectQuery) unicode_typec @ s e Z d Z d d Z d S) PasswordHashc C s" | j d } t | | | k S)Nzutf-8)encoder )selfZpassword r
/fields.pycheck_password1 s zPasswordHash.check_passwordN)__name__
__module__ __qualname__r r r r r r 0 s r c s= e Z d Z d f d d Z d d Z d d Z S)
PasswordField c sM d t t f k r t d | | _ d | _ t t | j | | d S)Nz2Missing library required for PasswordField: bcrypt)r r
ValueErrorbcrypt_iterationsZ raw_passwordsuperr __init__)r Z
iterationsargskwargs) __class__r r r 7 s
zPasswordField.__init__c C sc t | t r t | St | t r7 | j d } t | j } | d k rV | St | | S)z5Convert the python value for storage in the database.zutf-8N)
isinstancer bytesr
r r r r )r valueZsaltr r r db_value> s
zPasswordField.db_valuec C s( t | t r | j d } t | S)z/Convert the database value to a pythonic value.zutf-8)r r
r r )r r" r r r python_valueH s zPasswordField.python_value)r r r r r# r$ r r )r r r 6 s
r c @ s( e Z d Z d d Z d d Z d S)DeferredThroughModelc C s | | _ | | _ | | _ d S)N) model_classfieldname)r r& r' r( r r r set_fieldP s z DeferredThroughModel.set_fieldc C s) | | j _ | j j | j | j d S)N)r' _through_model add_to_classr& r( )r
through_modelr r r set_modelU s z DeferredThroughModel.set_modelN)r r r r) r- r r r r r% O s r% c @ sU e Z d Z d d d d d Z d d Z d d Z d d
Z d d Z d S)
ManyToManyFieldNFc C sw | d k r= t | t t f p- t | t r= t d | | _ | | _ | | _ | | _ d | _
d | _ d S)Nz[Unexpected value for `through_model`. Expected `Model`, `Proxy` or `DeferredThroughModel`.F) r Proxyr%
issubclassModel TypeError rel_model
_related_namer* _is_backref primary_key verbose_name)r r3 related_namer, r5 r r r r Z s zManyToManyField.__init__c C s
t | S)N)ManyToManyFieldDescriptor)r r r r _get_descriptorh s z ManyToManyField._get_descriptorc s t j t r> f d d } j j | d St j t rj j j d S _ _ j s t j
d d j _ t j
j s t j d j d d } j p j j d } | j j | d S) Nc s | _ j d S)N)r* r+ )r, )r& r( r r r callbackm s z.ManyToManyField.add_to_class.<locals>.callbackz_+ r, r5 Ts)r r* r/ Zattach_callbackr% r) r( r& r7 resubtitlesetattrr: r5 r. r4 _metar+ r3 )r r& r( r; Zbackrefr8 r )r& r( r r r+ k s& z ManyToManyField.add_to_classc C s6 d d t | j | j f | j | j f f D S)Nc S s g | ] \ } } | q Sr r ).0_modelr r r
<listcomp> s z.ManyToManyField.get_models.<locals>.<listcomp>)sortedr5 r& r3 )r r r r
get_models s zManyToManyField.get_modelsc s j s j \ d d f D G f d d d } j j t d j j t d i } | | d <t d j j f t f | _ j S)Nc S s g | ] } | j j q Sr )rB db_table)rC rE r r r rF s z5ManyToManyField.get_through_model.<locals>.<listcomp>c sV e Z d Z j j j Z d e Z j j j j f d f f Z d Z
d S)z/ManyToManyField.get_through_model.<locals>.Metaz
%s_%s_throughTFN) r r r r& rB databasetuplerI r( ZindexesZvalidate_backrefsr )lhsrhsr tablesr r Meta s
rO r3 z %s%sThrough)r* rH rB r( ZForeignKeyFieldtyper r1 )r rO Zattrsr )rL rM r rN r get_through_model s "
z!ManyToManyField.get_through_model)r r r r r: r+ rH rQ r r r r r. Y s
r. c s= e Z d Z f d d Z d d d Z d d Z S)r9 c sw t t | j | | j | _ | j | _ | j | _ | j j j | j | _ | j j j | j | _
d S)N) r r9 r r& r3 rQ r, rB Z
rel_for_modelsrc_fkdest_fk)r r' )r r r r s z"ManyToManyFieldDescriptor.__init__Nc C sV | d k rO t | | | j j j | j j | j j | j | k S| j S)N) ManyToManyQueryr3 selectjoinr, r& whererR r' )r instanceZ
instance_typer r r __get__ s z!ManyToManyFieldDescriptor.__get__c C s& | j | } | j | d d d S)Nclear_existingT)rY add)r rX r" queryr r r __set__ s z!ManyToManyFieldDescriptor.__set__)r r r r rY r] r r )r r r9 s r9 c sa e Z d Z f d d Z d d Z d d Z d d d Z d
d Z d d
Z S)rT c s/ | | _ | | _ t t | j | | d S)N) _instance_field_descriptorr rT r )r rX Zfield_descriptorr r )r r r r s zManyToManyQuery.__init__c C s: t | | j | j | j } | j | _ | j | S)N)rP r^ r_ r& rJ Z_clone_attributes)r r\ r r r clone s zManyToManyQuery.clonec C s( t | d t r$ d d | D S| S)Nr c S s g | ] } | j q Sr )get_id)rC objr r r rF s z,ManyToManyQuery._id_list.<locals>.<listcomp>)r r1 )r Zmodel_or_id_listr r r _id_list s zManyToManyQuery._id_listFc s | r j j t | t r | j t t j j j j
j } j j
d j j g d | j nc t | t t f s | g } | s d S f d d j | D } j j | j d S)NZfieldsr\ c s7 g | ]- } j j j j j j | i q Sr )rR r( r^ ra rS )rC Zrel_id)fdr r r rF s z'ManyToManyQuery.add.<locals>.<listcomp>)clearr_ r r rU ZSQLstrr^ ra r3 rB r6 r, Z insert_fromrR rS executelistrK rc Z insert_many)r r" rZ r\ Zinsertsr )rd r r r[ s"
zManyToManyQuery.addc C s | j } t | t ri | j | j j j } | j j j | j
| >| j | j j
k @ j St | t t f s | g } | s d S| j j j | j
| j | >| j | j j
k @ j Sd S)N)r_ r r rU r& rB r6 r, deleterW rS rR r^ ra rg rh rK rc )r r" rd Zsubqueryr r r remove s
zManyToManyQuery.removec C s. | j j j j | j j | j k j S)N)r_ r, ri rW rR r^ rg )r r r r re s zManyToManyQuery.clear) r r r r r` rc r[ rj re r r )r r rT s rT c s e Z d Z d Z d Z e e e e i Z d e f d d Z e rc d d Z
d d Z n d
d Z
d d Z S) CompressedFieldzlibbz2 c s | | _ | | j k r( t d | | j | } | d k rQ t d | | | _ | j | _ | j | _ t t | j | | d S)NzUnrecognized algorithm %sz Missing library required for %s.) compression_levelalgorithm_to_importr algorithmcompress
decompressr rk r )r ro rq r r Zcompress_module)r r r r s
zCompressedField.__init__c C s) | d k r% t | j | | j Sd S)N)r rr ro )r r" r r r r# s zCompressedField.db_valuec C s | d k r | j | Sd S)N)rs )r r" r r r r$ s z CompressedField.python_valuec C s) | d k r% | j t | | j Sd S)N)rr r ro )r r" r r r r# ! s c C s& | d k r" | j | j d Sd S)Nzutf-8)rs decode)r r" r r r r$ & s ) r r r ZZLIBZBZ2rl rm rp r PY2r# r$ r r )r r rk s rk c @ s( e Z d Z d d Z d d Z d S) PickledFieldc C s | d k r t j | Sd S)N)pickledumps)r r" r r r r# , s zPickledField.db_valuec C s; | d k r7 t | t r* | j d } t j | Sd S)NZraw_unicode_escape)r r
r rw loads)r r" r r r r$ 0 s zPickledField.python_valueN)r r r r# r$ r r r r rv + s rv c s e Z d Z f d d Z d d Z d d Z d d Z e rc d d
Z d d Z n d
d
Z d d Z S)AESEncryptedFieldc s& | | _ t t | j | | d S)N)keyr rz r )r r{ r r )r r r r 9 s zAESEncryptedField.__init__c C sL t | d k r t d | d d t | } t j | t j | S)N z"Key length cannot exceed 32 bytes.r< )lenr r newZMODE_CFB)r r{ ivr r r
get_cipher= s z AESEncryptedField.get_cipherc C s8 t j t j } | j | j | } | | j | S)N)r Zget_random_bytesr
block_sizer r{ encrypt)r r" r cipherr r r r C s zAESEncryptedField.encryptc C sB | d t j } | j | j | } | j | t j d S)N)r r r r{ decrypt)r r" r r r r r r H s zAESEncryptedField.decryptc C s# | d k r t | j | Sd S)N)r r )r r" r r r r# N s zAESEncryptedField.db_valuec C s | d k r | j | Sd S)N)r )r r" r r r r$ R s z AESEncryptedField.python_valuec C s | d k r | j | Sd S)N)r )r r" r r r r# V s c C s | d k r | j | Sd S)N)r )r r" r r r r$ Z s )
r r r r r r r ru r# r$ r r )r r rz 8 s rz )%cPicklerw ImportErrorr> sys version_inforu cStringIOr iorm rl Z
Crypto.Cipherr ZCryptor Zbcryptr r Zpeeweer r
r r r
r! r Z BlobFieldr objectr% r. r9 rT rk rv rz r r r r <module> s\
JG*