Current File : //opt/alt/python38/lib64/python3.8/site-packages/playhouse/__pycache__/shortcuts.cpython-38.pyc
U
S��W# � @ s� d dl Z d dlT d dlmZ e jd dkr>d dlmZ dd� Zddd �Zd
d � Zd d
� Z ddd�Z
ddd�Z G dd� de �Z
dS )� N)�*)�Node� )�Callablec C s
t | t�S �N)�
isinstancer )�c� r �
/shortcuts.py�<lambda> � r c C sx t d�g}| dk }|r |�| � |D ]"\}}|�t d�|t d�|f� q$|dk rb|�t d�|f� |�t d�� t|� S )a&
CASE statement builder.
Example CASE statements:
SELECT foo,
CASE
WHEN foo = 1 THEN "one"
WHEN foo = 2 THEN "two"
ELSE "?"
END -- will be in column named "case" in postgres --
FROM bar;
-- equivalent to above --
SELECT foo,
CASE foo
WHEN 1 THEN "one"
WHEN 2 THEN "two"
ELSE "?"
END
Corresponding peewee:
# No predicate, use expressions.
Bar.select(Bar.foo, case(None, (
(Bar.foo == 1, "one"),
(Bar.foo == 2, "two")), "?"))
# Predicate, will test for equality.
Bar.select(Bar.foo, case(Bar.foo, (
(1, "one"),
(2, "two")), "?"))
ZCASENZWHENZTHEN�ELSEZEND)�SQL�append�extend�Clause)Z predicateZexpression_tuples�defaultZclausesZ simple_case�expr�valuer r r
�case s "
r c C s t �t| td| ���S )NzAS %s)�fnZCASTr r )ZnodeZas_typer r r
�cast< s r c C s | r t | �S t � S r )�set)�sr r r
�
_clone_set@ s r TFc C s |dkr dn|}|dkr d}t |�}t |�}|dk rr|jD ]6} t| t�rT|�| � q:t| t�r:| jr:|�| j� q:i }
t |�}t |�}||O }t| �} | jj D ]�} | |ks�|r�| |kr�q�| j
� | j �}
t| t
��r|�r|
�r|�| � t| | j �}t|||||||d d�}
ni }
|
|
| j < q�|�rZ|D ].}t| |�}t|��rN|� |
|< n||
|< �q*|�r|�r| jj�� D ]�\}}t| |�}||k�sr||k�r��qr|�r�||k�r�||k�r��qrg }|�|� t| |d t| |��}|D ]$}|�t||||||d d�� �q�||
|< �qr|
S ) a�
Convert a model instance (and any related objects) to a dictionary.
:param bool recurse: Whether foreign-keys should be recursed.
:param bool backrefs: Whether lists of related objects should be recursed.
:param only: A list (or set) of field instances indicating which fields
should be included.
:param exclude: A list (or set) of field instances that should be
excluded from the dictionary.
:param list extra_attrs: Names of model instance attributes or methods
that should be included.
:param SelectQuery fields_from_query: Query that was source of model. Take
fields explicitly selected by the query and serialize them.
:param int max_depth: Maximum depth to recurse, value <= 0 means no max.
N���r F� )�recurse�backrefs�only�exclude�seen� max_depthZ _prefetch)r r r r r" )r Z_selectr ZField�addr Z_alias�type�_metaZdeclared_fields�_data�get�name�ForeignKeyField�getattr�
model_to_dict�callable� reverse_rel�itemsr )Zmodelr r r r r! Z extra_attrsZfields_from_queryr" �item�data� model_class�fieldZ
field_dataZrel_objZ attr_name�attr� related_nameZ foreign_keyZ
descriptorZaccumZ
related_queryr r r
r+ E s�
�
�� r+ c s | � }| j }|�� D ]�\}}||jkr6|j| � d}n@|| j jkrR|j| � d}n$�rft|||� qntd|| f ��t� t�}|s�|r�t|t�r�t|� j t
� j |��� q|r�t|t t
f�r�� �fdd�|D �} | D ]}
t|
� j |� q�t|� j| � qt|� j |� q|S )NFTz/Unrecognized attribute "%s" for model class %s.c s g | ]}t � j|���qS r )�
dict_to_modelr1 )�.0Zrow_data�r2 �ignore_unknownr r
�
<listcomp>� s ��z!dict_to_model.<locals>.<listcomp>)r% r. Zfieldsr- �setattr�AttributeErrorr r) �dictr( r5 Z rel_model�list�tupler4 ) r1 r0 r8 �instance�meta�keyr Z
is_backrefZis_foreign_keyZ instancesZ rel_instancer r7 r
r5 � s>
�
� �r5 c s e Zd Zd� fdd� Z� ZS )�RetryOperationalErrorNTc s� zt t| ��|||�}W nd tk
r| | �� s8| �� | �� �2 | �� }|�||pVd� |rn| � � rn| �
� W 5 Q R X Y nX |S )Nr ) �superrB � execute_sqlZOperationalErrorZ is_closed�closeZexception_wrapperZ
get_cursorZexecuteZget_autocommitZcommit)�selfZsqlZparamsZrequire_commitZcursor�� __class__r r
rD � s
�
z!RetryOperationalError.execute_sql)NT)�__name__�
__module__� __qualname__rD �
__classcell__r r rG r
rB � s rB )N)TFNNNNNN)F)�sysZpeeweer � version_info� collectionsr r, r r r r+ r5 �objectrB r r r r
�<module> s"
1 �
a
(
Mini Shell