R6�W# �
@ s� d d l Z d d l Td d l m Z e j d d k rU d d l m Z d d � Z d d d � Z d
d � Z d d
� Z d d d d d d d d d d � Z
d d d � Z Gd d � d e � Z
d S)� N)�*)�Node� )�Callablec C s
t | t � S)N)�
isinstancer )�c� r �
/shortcuts.py�<lambda> s r
c C s� t d � g } | d k } | r. | j | � x9 | D]1 \ } } | j t d � | t d � | f � q5 W| d k r� | j t d � | f � | j 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)� predicateZexpression_tuples�defaultZclausesZ simple_case�expr�valuer r r �case s "
) r c C s t j t | t d | � � � S)NzAS %s)�fnZCASTr r )ZnodeZas_typer r r �cast< s r c C s | r t | � St � S)N)�set)�sr r r �
_clone_set@ s
r TFc C s� | d k r d n | } | d k r* d } t | � } t | � } | d k r� xX | j D]M } t | t � r} | j | � qX t | t � rX | j rX | j | j � qX Wi }
t | � } t | � } | | O} t | � } x� | j j D]� } | | k s� | r| | k rq� | j
j | j � }
t | t
� r�| r�|
r�| j | � t | | j � } t | d | d | d | d | d | d
| d �}
n i }
|
|
| j <q� W| r�xC | D]; } t | | � } t | � r�| � |
| <q�| |
| <q�W| r�| r�x� | j j j � D]� \ } } t | | � } | | k s| | k rOq| rp| | k rp| | k rpqg } | j | � t | | d t | | � � } xC | D]; } | j t | d | d | d | d | d
| d �� q�W| |
| <qW|
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 �_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 } xm| j � D]_\ } } | | j k rP | j | � d } nZ | | j j k rx | j | � d } n2 � r� t | | | � q n t d | | f � � t � t � } | r� | r� t | t � r� t | � j t
� j | � � � q | rkt | t t
f � rk� � f d d � | D� } x! | D] }
t |
� j | � q8Wt | � j | � q t | � j | � q W| S)NFTz/Unrecognized attribute "%s" for model class %s.c s% g | ] } t � j | � � � q Sr )�
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 r7 �instance�meta�keyr Z
is_backrefZis_foreign_keyZ instancesZ rel_instancer )r2 r7 r r5 � s8
r5 c s( e Z d Z d d � f d d � Z � S)�RetryOperationalErrorNTc s� y" t t | � j | | | � } Wnz t k
r� | j � sH | j � | j � �C | j � } | j | | ps f � | r� | j � r� | j
� Wd QRXYn X| S)N) �superrA � execute_sqlZOperationalErrorZ is_closed�closeZexception_wrapperZ
get_cursorZexecuteZget_autocommitZcommit)�selfZsqlZparamsZrequire_commitZcursor)� __class__r r rC � s
z!RetryOperationalError.execute_sql)�__name__�
__module__� __qualname__rC r r )rF r rA � s rA )�sysZpeeweer � version_info� collectionsr r, r r r r+ r5 �objectrA r r r r �<module> s
1 _(