U
SW @ sx d Z ddlmZ ddlZddlT edZG dd deZG dd d e Z
zddl Z e
j Z
W n ek
rr Y nX dS )
zE
Simple translation of Django model classes to peewee model classes.
)partialN)*zpeewee.playhouse.djpeeweec @ s e Zd Zdd ZdS )AttrDictc C s | | S N )selfattrr r /djpeewee.py __getattr__ s zAttrDict.__getattr__N)__name__
__module__ __qualname__r
r r r r r s r c @ s6 e Zd Zdd Zdd Zdd Zd
d d
Zd d ZdS )DjangoTranslatorc C s | | _d S r )get_django_field_map
_field_map)r r r r __init__ s zDjangoTranslator.__init__c C s ddl m} |jtf|jtf|jtf|jtf|jtf|jtf|j t f|j
tf|j t f|j t f|j
ttddf|jtf|jtf|jjtfgS )Nr fieldsTZnull)django.db.modelsr Z AutoFieldZPrimaryKeyFieldZBigIntegerFieldZ BooleanFieldZ CharFieldZ
DateTimeFieldZ DateFieldZ DecimalFieldZ
FilePathFieldZ
FloatField IntegerFieldZNullBooleanFieldr Z TextFieldZ TimeFieldrelatedZ
ForeignKeyForeignKeyField)r djfr r r r s
z%DjangoTranslator.get_django_field_mapc C s* d }| j D ]\}}t||r
|} q&q
|S r )r
isinstance)r field convertedZ django_fieldZ peewee_fieldr r r
convert_field' s
z DjangoTranslator.convert_fieldNFc C sT |r||krd S |d kr d}ddl m} |j}||jr>d S d ||j< i }dd } t|j| dD ]}
| |
} | tkrP|dkr<|
jj } | jj}
|
|krd ||
< | j
| ||d ||d ||
d krt d |
j
|
t|
jd
||
j
< nL|
jjp|
}|d r d ||
j
|d f }t||
||
jd
||
j
< nt|
jd
||
j
< qb| rb| |
jd
||
j
< qbt|jtf|}|j|j_d|jj_|||j< |rzdd | D }W n( tk
r dd | D }Y nX |D ]4\}}|jj|krq| j
|||d ||d q|j D ]2}t ||j j s4q | j
|jj!||||d q d S )Nr r c S s | j r
dp d| fS )N r r )r r r r <lambda>C z3DjangoTranslator._translate_model.<locals>.<lambda>)keyr max_depthbackrefsexcludezCycle detected: %s: %s) db_column+z
__%s:%s:%s) related_namer' z%sc S s g | ]}||j fqS r )model.0fr r r
<listcomp>{ s z5DjangoTranslator._translate_model.<locals>.<listcomp>c S s0 g | ](}|j s|jr|jr|js||jjfqS r )Z one_to_manyZ
one_to_oneZ auto_createdZconcreter r* r+ r r r r. ~ s )"r r Z_metagetZ object_namesortedr r Zrelto_translate_modelloggerwarnnamer columnr) Zrelated_query_nameendswithstripr' typeZModelZdb_tableZdatabaseZ
interpolationZget_all_related_objectsAttributeErrorZ
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_objr; r r r r2 / s
z!DjangoTranslator._translate_modelc O s t }|dd}|dd}|dd}|dd}|sH|rHtdn|sRd }n|rb|dkrbd
}|D ]}| j|||||d qf|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/
ValueErrorr2 ) r Zmodelsr= r< r? r$ r% r& r* r r r translate_models s( !
z!DjangoTranslator.translate_models)NFN)r r r
r r r r2 rA r r r r r s
ir )__doc__ functoolsr ZloggingZpeeweeZ getLoggerr3 dictr objectr ZdjangorA translate ImportErrorr r r r <module> s
A