U
SW . @ sB z d dl ZW n ek
r( d dlZY nX d dlZd dlZejd dkZzd dlmZ W n< ek
r ejd dkrd dlmZ n d dl mZ Y nX z d dl
Z
W n ek
r dZ
Y nX z d dl Z W n ek
r dZ Y nX z d dl m
Z
d dlmZ W n ek
r d Z
ZY nX zd dlmZmZ W n ek
rP d ZZY nX d dlT d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ ererG d
d deZG dd deZ G dd de Z G dd deZ G dd deZ G dd deZ!G dd deZ"G dd d eZ#e
r>er>G d d d eZ$dS ) N )StringIO)AES)Random)hashpwgensalt)*)binary_construct)Field)FieldDescriptor) SelectQuery) unicode_typec @ s e Zd Zdd ZdS ) PasswordHashc C s | d}t|| | kS Nutf-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 Zd Zd fdd Zdd Zdd Z ZS )
PasswordField c s6 d t tfkrtd|| _d | _tt| 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 sF t |trt|S t |tr&|d}t| j}|dkr<|S t||S )z5Convert the python value for storage in the database.r N)
isinstancer bytesr
r r r r )r valueZsaltr r r db_value> s
zPasswordField.db_valuec C s t |tr|d}t|S )z/Convert the database value to a pythonic value.r )r# r
r r r r% r r r python_valueH s
zPasswordField.python_value)r )r r r r r& r(
__classcell__r r r! r r 6 s
r c @ s e Zd Zdd Zdd ZdS )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 d S r+ )r- _through_model add_to_classr, r. )r
through_modelr r r set_modelU s z DeferredThroughModel.set_modelN)r r r r/ r3 r r r r r* O s r* c @ s6 e Zd Zd
ddZdd Zdd Zd d
Zd d ZdS )ManyToManyFieldNFc C sP |d k r(t |ttfs(t|ts(td|| _|| _|| _|| _ d| _
d | _ d S )Nz[Unexpected value for `through_model`. Expected `Model`, `Proxy` or `DeferredThroughModel`.F) r# Proxyr*
issubclassModel TypeError rel_model
_related_namer0 _is_backref primary_key verbose_name)r r9 related_namer2 r; r r r r Z s zManyToManyField.__init__c C s t | S r+ )ManyToManyFieldDescriptorr r r r _get_descriptorh s z ManyToManyField._get_descriptorc s t jtr, fdd}j| d S t jtrLj d S _ _jsrt
dd _t
jstjjdd}jp jjd }|j| d S )Nc s | _ d S r+ )r0 r1 )r2 r, r. r r r callbackm s z.ManyToManyField.add_to_class.<locals>.callbackz_+ T)r2 r; s)r# r0 r5 Zattach_callbackr* r/ r. r, r= resubtitlesetattrrA r; r4 r: _metar1 r9 )r r, r. rC Zbackrefr> r rB r r1 k s( z ManyToManyField.add_to_classc C s( dd t | j| jf| j | jffD S )Nc S s g | ] \}}|qS r r ).0_modelr r r
<listcomp> s z.ManyToManyField.get_models.<locals>.<listcomp>)sortedr; r, r9 r@ r r r
get_models s
zManyToManyField.get_modelsc s j s \ dd fD G fddd} jjt djjtdi}||d< td jjf tf|_ j S )Nc S s g | ] }|j jqS r )rJ db_table)rK rM r r r rN s z5ManyToManyField.get_through_model.<locals>.<listcomp>c s< e Zd ZjjjZde Z jjjjfdffZ dZ
dS )z/ManyToManyField.get_through_model.<locals>.Metaz
%s_%s_throughTFN) r r r r, rJ databasetuplerQ r. ZindexesZvalidate_backrefsr ZlhsZrhsr Ztablesr r Meta s
rU )r9 z %s%sThrough)r0 rP rJ r. ZForeignKeyFieldtyper r7 )r rU Zattrsr rT r get_through_model s z!ManyToManyField.get_through_model)NNF)r r r r rA r1 rP rW r r r r r4 Y s
r4 c s. e Zd Z fddZdddZdd Z ZS ) r? c sR t t| | |j| _|j| _| | _| jj| j| _ | jj| j| _
d S r+ ) r r? r r, r9 rW r2 rJ Z
rel_for_modelsrc_fkdest_fk)r r- r! r r r s
z"ManyToManyFieldDescriptor.__init__Nc C s< |d k r6t || | j | j| j| j|kS | jS r+ ) ManyToManyQueryr9 selectjoinr2 r, whererX r- )r instanceZ
instance_typer r r __get__ s z!ManyToManyFieldDescriptor.__get__c C s | |}|j|dd d S )NT)clear_existing)r_ add)r r^ r% queryr r r __set__ s
z!ManyToManyFieldDescriptor.__set__)N)r r r r r_ rc r) r r r! r r? s
r? c sF e Zd Z fddZdd Zdd Zddd Zd
d Zd d
Z Z S )rZ c s" || _ || _tt| j|| d S r+ ) _instance_field_descriptorr rZ r )r r^ Zfield_descriptorr r r! r r r s zManyToManyQuery.__init__c C s( t | | j| j| j}| j|_| |S r+ )rV rd re r, rR Z_clone_attributes)r rb r r r clone s zManyToManyQuery.clonec C s t |d tr dd |D S |S )Nr c S s g | ] }| qS r )get_id)rK objr r r rN s z,ManyToManyQuery._id_list.<locals>.<listcomp>)r# r7 )r Zmodel_or_id_listr r r _id_list s zManyToManyQuery._id_listFc s |r j t|trZ|ttj j j
j } j j
j jg|d nFt|ttfsn|g}|svd S fdd|D } j | d S )N)Zfieldsrb c s& g | ] } j jj jj|iqS r )rX r. rd rg rY )rK Zrel_idfdr r r rN s z'ManyToManyQuery.add.<locals>.<listcomp>)clearre r# r r[ ZSQLstrrd rg r9 rJ r< r2 Z insert_fromrX rY executelistrS ri Z insert_many)r r% r` rb Zinsertsr rj r ra s(
zManyToManyQuery.addc C s | j }t|trJ||jjj}|j |j
|> |j | j
k@ S t|ttfs^|g}|sfd S |j |j
| |> |j | j
k@ S d S r+ )re r# r r[ r, rJ r< r2 deleter] rY rX rd rg rn ro rS ri )r r% rk Zsubqueryr r r remove s"
zManyToManyQuery.removec C s | j j | j j| jk S r+ )re r2 rp r] rX rd rn r@ r r r rl s zManyToManyQuery.clear)F)
r r r r rf ri ra rq rl r) r r r! r rZ s
rZ c s\ e Zd ZdZdZeeeeiZdef fdd Ze rDdd Z
dd Z nd
d Z
d d Z Z S ) CompressedFieldzlibbz2 c sf || _ || jkr td| | j| }|d kr:td| || _|j| _|j| _tt| j|| d S )NzUnrecognized algorithm %sz Missing library required for %s.) compression_levelalgorithm_to_importr algorithmcompress
decompressr rr r )r rv rx r r Zcompress_moduler! r r r s
zCompressedField.__init__c C s |d k rt | || jS d S r+ )r ry rv r' r r r r& s zCompressedField.db_valuec C s |d k r| |S d S r+ )rz r' r r r r( s z CompressedField.python_valuec C s |d k r| t|| jS d S r+ )ry r rv r' r r r r& ! s
c C s |d k r| |dS d S r )rz decoder' r r r r( & s )
r r r ZZLIBZBZ2rs rt rw r PY2r& r( r) r r r! r rr s
rr c @ s e Zd Zdd Zdd ZdS ) PickledFieldc C s |d k rt |S d S r+ )pickledumpsr' r r r r& , s zPickledField.db_valuec C s* |d k r&t |tr |d}t|S d S )NZraw_unicode_escape)r# r
r r~ loadsr' r r r r( 0 s
zPickledField.python_valueN)r r r r& r( r r r r r} + s r} c sZ e Zd Z fddZdd Zdd Zdd ZerBd d
Zd d Z nd
d
Zdd Z Z
S )AESEncryptedFieldc s || _ tt| j|| d S r+ )keyr r r )r r r r r! r r r 9 s zAESEncryptedField.__init__c C s8 t |dkrtd|ddt | }t|tj|S )N z"Key length cannot exceed 32 bytes.rD )lenr r newZMODE_CFB)r r ivr r r
get_cipher= s z AESEncryptedField.get_cipherc C s( t tj}| | j|}||| S r+ )r Zget_random_bytesr
block_sizer r encryptr r% r Zcipherr r r r C s zAESEncryptedField.encryptc C s0 |d t j }| | j|}||t jd S r+ )r r r r decryptr r r r r H s zAESEncryptedField.decryptc C s |d k rt | |S d S r+ )r r r' r r r r& N s zAESEncryptedField.db_valuec C s |d k r| |S d S r+ r r' r r r r( R s z AESEncryptedField.python_valuec C s |d k r| |S d S r+ )r r' r r r r& V s c C s |d k r| |S d S r+ r r' r r r r( Z s ) r r r r r r r r| r& r( r) r r r! r r 8 s
r )%ZcPickler~ ImportErrorrF sys version_infor| Z cStringIOr iort rs Z
Crypto.Cipherr ZCryptor Zbcryptr r Zpeeweer r
r r r
r$ r Z BlobFieldr objectr* r4 r? rZ rr r} r r r r r <module> s\
JG*