R6�W . � 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
r�d 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
r�e r�Gd d � d e � Z$ d S) � N� )�StringIO)�AES)�Random)�hashpw�gensalt)�*)�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.py�check_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 �
ValueError�bcrypt_iterationsZ raw_password�superr �__init__)r Z
iterations�args�kwargs)� __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_class�field�name)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% �
issubclass�Model� 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 T�s)r r* r/ Zattach_callbackr% r) r( r& r7 �re�sub�title�setattrr: 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 �database�tuplerI r( ZindexesZvalidate_backrefsr )�lhs�rhsr �tablesr r �Meta� s
rO r3 z %s%sThrough)r* rH rB r( ZForeignKeyField�typer 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_model�src_fk�dest_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 �select�joinr, r& �whererR r' )r �instanceZ
instance_typer r r �__get__� s z!ManyToManyFieldDescriptor.__get__c C s& | j | � } | j | d d �d S)N�clear_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 ZSQL�strr^ ra r3 rB r6 r, Z insert_fromrR rS �execute�listrK 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) �CompressedField�zlib�bz2� 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_level�algorithm_to_importr � algorithm�compress�
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)�pickle�dumps)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*