R6�W��@s�dZd
dl
mZ
d
dlZd
dlTejd�
ZGdd�de�ZGdd	�d	e	�Z
yd
dlZe
�jZ
Wnek
r�


Yn
XdS)
zE
Simple translation of Django model classes to peewee model classes.
�)
�partialN)
�
*zpeewee.playhouse.djpeeweec@seZ
dZd
d�ZdS)�AttrDictcCs||
S)
N�)�self�attrrr�/djpeewee.py�__getattr__s
zAttrDict.__getattr__N)�__name__�
__module__�__qualname__r	rrrrrs
rc@sUeZ
dZd
d�Zdd�Zdd�Zdddd	d
�Zdd�ZdS)
�DjangoTranslatorc

Cs|j�|_
dS)
N)�get_django_field_map�
_field_map)
rrrr�__init__s
zDjangoTranslator.__init__c
Cs�d
dlm
}

|
jtf|
jtf|
jtf|
jtf|
jtf|
jtf|
j	t	f|
j
tf|
jtf|
jtf|
j
ttdd�

f|
jtf|
jtf|
jjtfgS)Nr
)
�fields�nullT)�django.db.modelsrZ	AutoFieldZPrimaryKeyFieldZBigIntegerFieldZBooleanFieldZ	CharFieldZ
DateTimeFieldZ	DateFieldZDecimalFieldZ
FilePathFieldZ
FloatField�IntegerFieldZNullBooleanFieldrZ	TextFieldZ	TimeField�relatedZ
ForeignKey�ForeignKeyField)r�djfrrrrs












z%DjangoTranslator.get_django_field_mapcCs:d}x-|jD]"\}}t
|
|�r|}PqW|S)
N)r�
isinstance)r�field�	convertedZdjango_fieldZpeewee_fieldrrr�
convert_field's





zDjangoTranslator.convert_fieldNFcCs;|r|
|krdS|dkr(d}ddlm
}
|
j}|j|j�
rWdSd||j<i}dd�}	x�
t|j
d|	�

D]m
}
|j|
�
}|tkr�
|dkr�
|
jj	}|jj}
|
|kr

d||
<|j
||d|d
d|d	|�
||
dkrO
tjd
|
j
|
�
td|
j�
||
j
<q�
|
jjpd
|
j�}|jd�
r�
d
||
j
|jd�
f}t||
d|d|
j�
||
j
<q�
td|
j�
||
j
<q�|r�|d|
j�
||
j
<q�Wt|jtf
|�}|j|j_d|jj_|||j<|r�ydd�|j�D�
}Wn+tk
r�


dd�|j�D�
}Yn
XxO|D]G\}}
|
jj|kr�q�|j
|
|d|d
d|d	|�
q�WxQ|jD]F}t||jj �sq�|j
|jj!|d|d|d	|�
q�WdS)N�
r
)
rc

Ss|jrd
pd|fS)Nrr
)
r)
rrrr�<lambda>Csz3DjangoTranslator._translate_model.<locals>.<lambda>�key�	max_depth�backrefs�excludezCycle detected: %s: %s�	db_column�
+z
__%s:%s:%s�related_namez%sc
Ssg|]}
|
|
jf�qSr)
�model)�.0�
frrr�
<listcomp>{s	z5DjangoTranslator._translate_model.<locals>.<listcomp>c
SsGg|]=}
|
js|
j
r|
jr|
jr|
|
jjf�qSr)Zone_to_manyZ
one_to_oneZauto_createdZconcreterr%)r&r'rrrr(~s	

���)"rrZ_meta�getZobject_name�sortedrrZrelZto�_translate_model�logger�warn�namer�columnr$Zrelated_query_name�endswith�stripr"�typeZModelZdb_tableZdatabaseZ
interpolationZget_all_related_objects�AttributeErrorZ
get_fields�many_to_manyrrZManyToManyFieldZthrough)rr%�mappingrr r!r�optionsZattrsZ	field_keyZmodel_fieldrZ
related_modelZ
model_namer$�klassZall_relatedZrel_objr5rrrr,/s�


	

















	





















	
















	



z!DjangoTranslator._translate_modelc
	
Os�t�}|j
d
d�}|j
dd�}|j
dd�}|j
dd�}|rm|rmtd�
�
n'|s|d	}n|r�|dkr�d}x0|
D](}|j||d|d|d|�
q�W|S)av
        Generate a group of peewee models analagous to the provided Django
        models for the purposes of creating queries.

        :param model: A Django model class.
        :param options: A dictionary of options, see note below.
        :returns: A dictionary mapping model names to peewee model classes.
        :rtype: dict

        Recognized options:
            `recurse`: Follow foreign keys (default: True)
            `max_depth`: Max depth to recurse (default: None, unlimited)
            `backrefs`: Follow backrefs (default: False)
            `exclude`: A list of models to exclude

        Example::

            # Map Django models to peewee models. Foreign keys and M2M will be
            # traversed as well.
            peewee = translate(Account)

            # Generate query using peewee.
            PUser = peewee['User']
            PAccount = peewee['Account']
            query = (PUser
                     .select()
                     .join(PAccount)
                     .where(PAccount.acct_type == 'foo'))

            # Django raw query.
            users = User.objects.raw(*query.sql())
        �recurseTrNr Fr!z9Error, you cannot specify a max_depth when recurse=False.r
rr))rr*�
ValueErrorr,)	rZmodelsr7r6r9rr r!r%rrr�translate_models�s&!	







	









z!DjangoTranslator.translate_models)r
rrrrrr,r;rrrrr
s


dr
)�__doc__�	functoolsr�loggingZpeewee�	getLoggerr-�dictr�objectr
Zdjangor;�	translate�ImportErrorrrrr�<module>s


�