]j- d dl Z d dlT d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d Z dd Zdd Z dd
Z
ddZ G d de Z
ddZ G d de ZdS ) N)*)Alias)CompoundSelectQuery)Metadata) callable_)__deprecated__c @ | rt | n
t S N)set)ss J/opt/imunify360/venv/lib64/python3.11/site-packages/playhouse/shortcuts.py<lambda>r s -s1vvv TFc
|dn|}|dk rd}t t | }fd}
|y d |j D ]\} t | t r | -t | t
r| | j ]i } t t | }|z t | }
| r| j j
D ]\ }} |
| r ||j j j |j
f |j j j D ]} | g }t! | | D ]-}| t% ||||dz
.|| |< | j j D ]} |
| r| j |j }t |t. rJ|rH|D| | t! | |j }t% |||||dz
}nd}|| |j < |r7|D ]4}t! | | }t1 | r | | |< /|| |< 5|r|r| j j
D ]\ }}|j
d
k rt! |
|j
}|v s|v r/r |vr|vr:g } | t! | |j
}|D ]-}| t% ||||dz
.|| |j
< | S ) a8
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.
:param bool manytomany: Process many-to-many fields.
Nr Fc | v po| vS r
)nexcludeonlys r
r z model_to_dict.<locals>.<lambda>' s Q'\ Ht/G$ r __sentinel__ )recursebackrefsr r max_depth)r r r r seenr +)
_clone_setadd
_returning
isinstanceFieldr _aliastype_meta
manytomanyitemsupdate rel_modelbackref
through_modelrefsgetattrappend
model_to_dict
sorted_fields__data__getnameForeignKeyFieldr r )modelr r r r r extra_attrsfields_from_queryr r& should_skipitemdata model_classr3 m2mfkfaccumrel_objfield
field_data attr_nameattr foreign_keyr)
descriptor
related_querys `` r
r/ r/ s' & 'YIA~~ d D [))KHHHHHK$ %0 - -D$&&
-D%((
- ,,,
D!!G d D tOGu++K /5577 ID#{4
NNC !4!? !LM N N N(.3
!
! C E"5$//
.
. ] # % # '!m
- - - . . . . DJJ* & & ;u ^''
33
e_ - -
"'
"%!%44 * # % # '!m - - -
"
%UZ '$ ' 'I5),,D
'"&$&&Y"&Y .G .&+k&:&@&@&B&B . . "K"c))8 k.ABBJW$$ w(>(>
4//k6M6ME KK $ $ $ #E;+> ? ?M(
.
. ] # % # '!m
- - - . . . . ).D$ % % Kr c
N | j }t d |j D }| D ]g\ }}||j v r|j | d}nF||v r || d}n7rt | || Bt
d|dt | d t t }|su|rst |t r^ | j
j } n$# t $ r
} Y nw xY wt | j t | | |r^t |t t f rB fd|D }
|
D ]} t | j | t | j |
Qt | j | i| S )Nc g | ] }|j |f S r )r* ).0fks r
<listcomp>z*update_model_from_dict.<locals>.<listcomp> s >>>"bj"%>>>r FTzUnrecognized attribute "z" for model class .c < g | ]}t j | S r )
dict_to_modelr5 )rI row_datar@ ignore_unknowns r
rK z*update_model_from_dict.<locals>.<listcomp> s7 ' ' ' ek8^DD' ' 'r )r% dictr r' combinedsetattrAttributeErrorr$ r! r4 __rel__r3 KeyErrorr) update_model_from_dictlisttupler* ) instancer: rP metar keyvalue
is_backrefis_foreign_key rel_instance instancesr@ s ` @r
rW rW s >D>>
>>>??Hjjll !1 !1
U $- M#&E JJ
H__ SME JJ
F Hc5 ) ) ) .03T(^^^^"E F F
F $E?;; 1n 1E41H1H 1
1 ' /
;
1
1
1 $ 0 0
1
&|UNKK
M
M
M
M 1JutUm < < 1' ' ' ' ' %' ' 'I !*
<
< ej(;;;; HemY 7 7 7 7 Hej% 0 0 0 0 Os C## DDc 4 t | || S r
)rW )r; r: rP s r
rN rN s !++--~ F FFr c x | j j D ]3\ }}|j |v s||v rt | r
| n|}|||< 4t | \ }}t
| | }| || S )a
Helper for generating conditional INSERT queries.
For example, prevent INSERTing a new tweet if the user has tweeted within
the last hour::
INSERT INTO "tweet" ("user_id", "content", "timestamp")
SELECT 234, 'some content', now()
WHERE NOT EXISTS (
SELECT 1 FROM "tweet"
WHERE user_id = 234 AND timestamp > now() - interval '1 hour')
Using this helper:
cond = ~fn.EXISTS(Tweet.select().where(
Tweet.user == user_obj,
Tweet.timestamp > one_hour_ago))
iq = insert_where(Tweet, {
Tweet.user: user_obj,
Tweet.content: 'some content'}, where=cond)
res = iq.execute()
)columns)
r% defaultsr' r3 r zipSelectwhere insert_from as_rowcount) clsr: rh r@ defaultr] fieldsvaluessqs r
insert_whererp s 2 ),2244 w : $ &w / /< WU $**,,'NFF % %e , ,B ??2v & & 2 2 4 44r c ` e Zd ZdZedfedfedfedfedffZ fdZd fd Z fd
Zd Z xZ
S )
ReconnectMixina
Mixin class that attempts to automatically reconnect to the database under
certain error conditions.
For example, MySQL servers will typically close connections that are idle
for 28800 seconds ("wait_timeout" setting). If your application makes use
of long-lived connections, you may find your connections are closed after
a period of no activity. This mixin will attempt to reconnect automatically
when these errors occur.
This mixin class probably should not be used with Postgres (unless you
REALLY know what you are doing) and definitely has no business being used
with Sqlite. If you wish to use with Postgres, you will need to adapt the
`reconnect_errors` attribute to something appropriate for Postgres.
2006201320144031z MySQL Connection not available.c
t t | j |i | i | _ | j D ]R\ }}| j |g | j | | Sd S r
)superrr __init___reconnect_errorsreconnect_errors
setdefaultr. lower)selfargskwargs exc_class err_fragment __class__s r
ry zReconnectMixin.__init__ s ,nd##,d=f=== "$ '+'< K K #I| " - -i < < < "9 - 4 4\5G5G5I5I J J J J K Kr Nc |t d | t t | j || S )Nz,"commit" has been deprecated and is a no-op.)r
_reconnectrx rr execute_sql)r~ sqlparamscommitr s r
r zReconnectMixin.execute_sql s; I J J Ju^T : : FVTTTr c ^ | t t | j S r
)r rx rr begin)r~ r s r
r zReconnectMixin.begin s" u^T : : @AAAr c ||i |S # t $ r}| r|t | }|| j vr|t | }| j | D ]}||v r n || s(| | ||i |cY d }~S d }~ww xY wr
) Exceptionin_transactionr$ rz strr} is_closedcloseconnect)r~ funcr r excr exc_reprr s r
r zReconnectMixin._reconnect s )4((( ( ) ) ) ""$$
S I 666 3xx~~''H $ 6y A
8++E , >>##
4((( ( ( ( ( ( (+ )s
C B,C C C )NN) __name__
__module__ __qualname____doc__OperationalErrorr{ ry r r r
__classcell__r s @r
rr rr s "
6" 6" 6" 6"
<= K K K K KU U U U U U
B B B B B) ) ) ) ) ) )r rr _model_identifierc v i g}|r| }t |t r"| |j |j f M|j }|j j }||< |j
t | | fd} | S )Nc 3 K D ])} | }| } |di | V *d S )Nr )dictsiteratorpop)row
identifierr5 r\ mappingquerys r
wrapped_iteratorz2resolve_multimodel_query.<locals>.wrapped_iterator. sh ;;==))++ C JJ'E%,,#,, r )
r r! r extendlhsrhsr5 r%
table_namer r. Valuealias)r r\ r> currr; r3 r r s `` @r
resolve_multimodel_queryr s G
GE
7yy{{ d / 0 0 LL$(DH- . . . j +#
uT{{ 0 0 5 5666 7 r c F e Zd ZdZ fdZd Zd Z eee Z xZ S )ThreadSafeDatabaseMetadataz
Metadata class to allow swapping database at run-time in a multi-threaded
application. To use:
class Base(Model):
class Meta:
model_metadata_class = ThreadSafeDatabaseMetadata
c d | _ t j | _ t t
| j |i | d S r
) _database threadinglocal_localrx r ry )r~ r r r s r
ry z#ThreadSafeDatabaseMetadata.__init__@ sB o'' 8($//8$I&IIIIIr c 8 t | j d| j S )Ndatabase)r- r r )r~ s r
_get_dbz"ThreadSafeDatabaseMetadata._get_dbF s t{J???r c : | j || _ || j _ d S r
)r r r )r~ dbs r
_set_dbz"ThreadSafeDatabaseMetadata._set_dbH s! > ! DN ! r )
r r r r ry r r propertyr r r s @r
r r 7 so J J J J J @ @ @" " " x))HHHHHr r ) TFNNNNNNF)Fr
)r )r peeweer r r r r r r/ rW rN rp objectrr r r r r r
<module>r sc & & & & & & ! ! ! ! ! ! .
-
=A7;EJs s s sl' ' ' 'TG G G G 5 5 5 5DI) I) I) I) I)V I) I) I)X .* * * * * * * * * *r