R6W @ 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 )selfattrr 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 )fieldsnullT)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 } 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 } xt | 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 rd
| |
j
| j d f } t | |
d | d |
j | |
j
<qt d |
j | |
j
<q | r | d |
j | |
j
<q Wt | j t f | } | j | j _ d | j j _ | | | j <| ry d d | j D } Wn+ t k
rd d | j D } Yn XxO | D]G \ } } | j j | k rq| j
| | d | d d | d | qWxQ | j D]F } t | | j j s q| j
| j j! | d | d | d | qWd 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_depthbackrefsexcludezCycle detected: %s: %s db_column+z
__%s:%s:%s related_namez%sc S s g | ] } | | j f q Sr )model).0fr 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_metagetZ object_namesortedr r ZrelZto_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_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