R6�W� � @ s� d Z d d l m Z d d l Z d d l Te j d � Z Gd d � d e � Z Gd d � d e � Z
y d d l Z e
� j Z
Wn e k
r� Yn Xd S)
zE
Simple translation of Django model classes to peewee model classes.
� )�partialN)�*zpeewee.playhouse.djpeeweec @ s e Z d Z d d � Z d S)�AttrDictc C s | | S)N� )�self�attrr r � /djpeewee.py� __getattr__ s zAttrDict.__getattr__N)�__name__�
__module__� __qualname__r r r r r r s r c @ sU e Z d Z d d � Z d d � Z d d � Z d d d d d
� Z d d � Z d S)
�DjangoTranslatorc C s | j � | _ d S)N)�get_django_field_map�
_field_map)r r r r �__init__ s zDjangoTranslator.__init__c C s� d d l m } | j t f | j t f | j t f | j t f | j t f | j t f | j t f | j
t f | j t f | j t f | j
t t d d �f | j t f | j t f | j j t f g S)Nr )�fields�nullT)�django.db.modelsr Z AutoFieldZPrimaryKeyFieldZBigIntegerFieldZ BooleanFieldZ CharFieldZ
DateTimeFieldZ DateFieldZ DecimalFieldZ
FilePathFieldZ
FloatField� IntegerFieldZNullBooleanFieldr Z TextFieldZ TimeField�relatedZ
ForeignKey�ForeignKeyField)r �djfr r r r s z%DjangoTranslator.get_django_field_mapc C s: d } x- | j D]" \ } } t | | � r | } Pq W| S)N)r �
isinstance)r �field� convertedZ django_fieldZ peewee_fieldr r r �
convert_field' s z DjangoTranslator.convert_fieldNFc C s; | r | | k r d S| d k r( d } d d l m } | j } | j | j � rW d Sd | | j <i } d d � } x�t | j d | �D]m}
| j |
� } | t k r�| d k r�|
j j } | j j }
|
| k r
d | |
<| j
| | d | d d | d | �| |
d k rOt j d
|
j
|
� t d |
j � | |
j
<q�|
j j pd|
j � } | j d � r�d
| |
j
| j d � f } t | |
d | d |
j �| |
j
<q�t d |
j � | |
j
<q� | r� | d |
j � | |
j
<q� Wt | j t f | � } | j | j _ d | j j _ | | | j <| r�y d d � | j � D� } Wn+ t k
r�d d � | j � D� } Yn XxO | D]G \ } } | j j | k r�q�| j
| | d | d d | d | �q�WxQ | j D]F } t | | j j � s q�| j
| j j! | d | d | d | �q�Wd S)N� r )r c S s | j r d p d | f S)Nr r )r )r r r r �<lambda>C s z3DjangoTranslator._translate_model.<locals>.<lambda>�key� max_depth�backrefs�excludezCycle detected: %s: %s� db_column�+z
__%s:%s:%s� related_namez%sc S s g | ] } | | j f � q Sr )�model)�.0�fr r r �
<listcomp>{ s z5DjangoTranslator._translate_model.<locals>.<listcomp>c S sG g | ]= } | j s | j r | j r | j r | | j j f � q Sr )Z one_to_manyZ
one_to_oneZ auto_createdZconcreter r% )r& r' r r r r( ~ s ���)"r r Z_meta�getZ object_name�sortedr r ZrelZto�_translate_model�logger�warn�namer �columnr$ Zrelated_query_name�endswith�stripr" �typeZModelZdb_tableZdatabaseZ
interpolationZget_all_related_objects�AttributeErrorZ
get_fields� many_to_manyr r ZManyToManyFieldZthrough)r r% �mappingr r r! r �optionsZattrsZ field_keyZ model_fieldr Z
related_modelZ
model_namer$ �klassZ all_relatedZrel_objr5 r r r r, / s�
z!DjangoTranslator._translate_modelc
O s� t � } | j d d � } | j d d � } | j d d � } | j d d � } | rm | rm t d � � n' | s| d } n | r� | d k r� 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())
�recurseTr Nr Fr! z9Error, you cannot specify a max_depth when recurse=False.r r r) )r r* �
ValueErrorr, ) r Zmodelsr7 r6 r9 r r r! r% r r r �translate_models� s& !
z!DjangoTranslator.translate_models)r
r r r r r r, r; r r r r r
s dr
)�__doc__� functoolsr �loggingZpeewee� getLoggerr- �dictr �objectr
Zdjangor; � translate� ImportErrorr r r r �<module> s
�