File: //opt/alt/python38/lib64/python3.8/site-packages/playhouse/__pycache__/djpeewee.cpython-38.pyc
U
S��W� � @ sx d Z ddlmZ ddlZddlT e�d�ZG dd� de�ZG 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� )�self�attrr r �E/opt/alt/python38/lib64/python3.8/site-packages/playhouse/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
ttdd�f|jtf|jtf|jjtfgS )Nr ��fieldsT��null)�django.db.modelsr Z AutoField�PrimaryKeyField�BigIntegerField� BooleanField� CharField�
DateTimeField� DateField� DecimalFieldZ
FilePathField�
FloatField� IntegerFieldZNullBooleanFieldr � TextField� TimeField�relatedZ
ForeignKey�ForeignKeyField)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}|�|j�r>d S d ||j< i }dd� } t|j| d�D �]}
| �|
�} | tk�rP|dk�r<|
jj } | jj}
|
|kr�d ||
< | j
| ||d ||d� ||
d kr�t � 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|k�r��q�| j
|||d ||d� �q�|j D ]2}t ||j j ��s4�q | j
|jj!||||d� �q d S )N���r 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_depth�backrefs�excludezCycle detected: %s: %s)� db_column�+z
__%s:%s:%s)� related_namer2 z%sc S s g | ]}||j f�qS r )�model��.0�fr r r �
<listcomp>{ s z5DjangoTranslator._translate_model.<locals>.<listcomp>c S s0 g | ](}|j s|jr|jr|js||jjf�qS r )Z one_to_manyZ
one_to_oneZ auto_createdZconcreter& r5 r6 r r r r9 ~ s �)"r r �_meta�getZ object_name�sortedr( r# �rel�to�_translate_model�logger�warn�namer �columnr4 Zrelated_query_name�endswith�stripr2 �type�Model�db_table�database�
interpolationZget_all_related_objects�AttributeErrorZ
get_fields� many_to_manyr% r" ZManyToManyFieldZthrough)r r5 �mappingr/ r0 r1 r$ �options�attrsZ field_keyZ model_fieldr' Z
related_modelZ
model_namer4 �klassZ all_relatedZrel_objrL r r r r? / s�
� ��� ����
�
�
�z!DjangoTranslator._translate_modelc O s� t � }|�dd�}|�dd�}|�dd�}|�dd�}|sH|rHtd��n|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/ Nr0 Fr1 z9Error, you cannot specify a max_depth when recurse=False.r r) r. )r r; �
ValueErrorr? ) r �modelsrN rM rQ r/ r0 r1 r5 r r r �translate_models� s( !
�z!DjangoTranslator.translate_models)NFN)r r r
r r r( r? rT r r r r r s �
ir )�__doc__� functoolsr �logging�peewee� getLoggerr@ �dictr �objectr ZdjangorT � translate� ImportErrorr r r r �<module> s
A