^p d Z ddlZddlZddlZddlmZm Z ddl
Z ddl Z ddl m
Z ddl
mZ ddl
mZ ddl
mZmZ dd lmZ dd
lmZmZmZ m Z m Z dd l m!Z!m"Z"m#Z#m$Z$m%Z% dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z. dd
l/m0Z0 G d de Z1 G d de Z2 G d de1 Z3 G d de4 Z5 G d de Z6 G d de1 Z7 G d de Z8 G d d e Z9 G d d e Z: ed d! Z; e G d& d'e= Z? G d( d)e> Z@ G d* d+e ZA G d, d-e ZB G d. d/eC ZD G d0 d1e Z G d2 d3 ZEdId4ZF G d5 d6 ZGdId7ZHd8 ZId9 ZJ G d: d; ZK dJd=ZL G d> d? ZMdKd@ZNdA ZOdLdCZPdMdDZQdE ZR ejS dF ejT dG z dHz ZUdS )NzMiscellaneous goodies for psycopg2
This module is a generic place used to hold little helper functions
and classes until a better place in the distribution is found.
N)
namedtuple OrderedDict)
extensions )cursor)
connection)adapt quote_ident) lru_cache)REPLICATION_PHYSICALREPLICATION_LOGICALReplicationConnectionReplicationCursorReplicationMessage)jsonJson
register_jsonregister_default_jsonregister_default_jsonb)Range NumericRange DateRange
DateTimeRangeDateTimeTZRangeregister_range RangeAdapter RangeCaster)register_ipaddressc L e Zd ZdZ fdZ fdZd fd Z fdZ fdZ xZ S ) DictCursorBasez%Base class for all dict-like cursors.c d|v r |d }|d= nt d t j |i | d| _ d| _ || _ d S )N row_factoryz;DictCursorBase can't be instantiated without a row factory.F)NotImplementedErrorsuper__init___query_executed _prefetchr" )selfargskwargsr" __class__s c/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/psycopg2/extras.pyr% zDictCursorBase.__init__D st F " " /K}%%%MO O
O$)&))) $ & c | j r t }| j r| | j s t }|S N)r' r$ fetchoner& _build_indexr( resr+ s r, r0 zDictCursorBase.fetchoneP c > %''""$$C ~ %''""$$C
r- Nc | j r!t | }| j r| | j s!t | }|S r/ )r' r$ fetchmanyr& r1 )r( sizer3 r+ s r, r6 zDictCursorBase.fetchmanyY sg > *''##D))C ~ *''##D))C
r- c | j r t }| j r| | j s t }|S r/ )r' r$ fetchallr& r1 r2 s r, r9 zDictCursorBase.fetchallb r4 r- c # h K | j r/t }t | }| j r| | j s/t }t | }|V t | V # t $ r Y d S w xY wr/ )r' r$ __iter__nextr& r1
StopIteration)r( r3 firstr+ s r, r; zDictCursorBase.__iter__k s ~
"gg&&(( S #
$!!###>
"gg&&(( S KKK
3ii
FF s B B# #
B10B1r/ )
__name__
__module__ __qualname____doc__r% r0 r6 r9 r;
__classcell__r+ s @r, r r A s //
'
'
'
'
' r- r c " e Zd ZdZ fdZ xZS )DictConnectionz2A connection that uses `DictCursor` automatically.c z | d| j pt t j |i |S Ncursor_factory)
setdefaultrI
DictCursorr$ r r( r) r* r+ s r, r zDictConnection.cursor s= *D,?,M:NNN uww~t.v...r- r? r@ rA rB r rC rD s @r, rF rF } s> <</ / / / / / / / /r- rF c @ e Zd ZdZ fdZd fd Zd fd Zd Z xZS )rK zA cursor that keeps a list of column name -> index mappings__.
.. __: https://docs.python.org/glossary.html#term-mapping
c \ t |d< t j |i | d| _ d S )Nr" T)DictRowr$ r% r' rL s r, r% zDictCursor.__init__ s4 '} $)&))) r- Nc | t | _ d| _ t || S NT)r indexr& r$ executer( queryvarsr+ s r, rT zDictCursor.execute s. ]]
# wwud+++r- c | t | _ d| _ t || S rR )r rS r& r$ callprocr( procnamerW r+ s r, rY zDictCursor.callproc s0 ]]
# ww $///r- c | j rO| j rJt t | j D ] }|| j | j | d < d| _ d S d S d S )Nr F)r& descriptionrangelenrS )r( is r, r1 zDictCursor._build_index sw )D$4 )3t /0011
7
756
4+A.q122#(D ) ) ) )r- r/ r? r@ rA rB r% rT rY r1 rC rD s @r, rK rK s
, , , , , ,
0 0 0 0 0 0
) ) ) ) ) ) )r- rK c z e Zd ZdZdZd Z fdZ fdZ fdZd Z fdZ
dd
Z d Z d Z
fd
Zd Zd Z xZS )rP z6A row object that allow by-column-name access to data._indexc X |j | _ d gt |j z | d d < d S r/ )rS rd r_ r] )r( r s r, r% zDictRow.__init__ s. l &3v 1222QQQr- c t |t t f s
| j | }t | S r/ )
isinstanceintslicerd r$ __getitem__)r( xr+ s r, rj zDictRow.__getitem__ s< !c5\** AAww""1%%%r- c t |t t f s
| j | }t || d S r/ )rg rh ri rd r$ __setitem__)r( rk vr+ s r, rm zDictRow.__setitem__ sD !c5\** AA
Aq!!!!!r- c V t j fd j D S )Nc 3 H K | ] }| j | fV d S r/ rc .0ngr( s r, z DictRow.items.. s8 <<1AAdk!n%%&<<<<<. s3 77a$+a.!!777777r- rv rw s `@r, valueszDictRow.values s0 GG 777774;7777r- Nc 8 | | S # t $ r |cY S w xY wr/ ) Exception)r( rk defaults r, getz DictRow.get s4 7N NNN s
c D t | S r/ )r rx r{ s r, copyz DictRow.copy s 4::<<(((r- c || j v S r/ rc )r( rk s r, __contains__zDictRow.__contains__ s DK r- c D t S r/ )r$
__reduce__)r( r+ s r, r zDictRow.__reduce__ s ww!!###r- c H | d d | j fS r/ )rd r r{ s r, __getstate__zDictRow.__getstate__ s# AAAw ((****r- c : |d | d d < |d | _ d S Nr r rc )r( datas r, __setstate__zDictRow.__setstate__ s" q'QQQ1g r- r/ )r? r@ rA rB __slots__r% rj rm rx r| r r r r r r r rC rD s @r, rP rP s @@I3 3 3& & & & &
" " " " "
= = = = =! ! !8 8 8 8 8 ) ) ) $ $ $ $ $+ + + r- rP c " e Zd ZdZ fdZ xZS )RealDictConnectionz6A connection that uses `RealDictCursor` automatically.c z | d| j pt t j |i |S rH )rJ rI RealDictCursorr$ r rL s r, r zRealDictConnection.cursor s= *D,?,Q>RRR uww~t.v...r- rM rD s @r, r r s> @@/ / / / / / / / /r- r c @ e Zd ZdZ fdZd fd Zd fd Zd Z xZS )r aQ A cursor that uses a real dict as the base type for rows.
Note that this cursor is extremely specialized and does not allow
the normal access (using integer indices) to fetched data. If you need
to access database rows both as a dictionary and a list, then use
the generic `DictCursor` instead of `!RealDictCursor`.
c N t |d< t j |i | d S )Nr" ) RealDictRowr$ r% rL s r, r% zRealDictCursor.__init__ s/ +} $)&)))))r- Nc d g | _ d| _ t || S rR )column_mappingr& r$ rT rU s r, rT zRealDictCursor.execute s+ # wwud+++r- c d g | _ d| _ t || S rR )r r& r$ rY rZ s r, rY zRealDictCursor.callproc s- # ww $///r- c d | j r&| j r!d | j D | _ d| _ d S d S d S )Nc g | ]
}|d S r rr ds r,
z/RealDictCursor._build_index.. s "B"B"BA1Q4"B"B"Br- F)r& r] r r{ s r, r1 zRealDictCursor._build_index sR )D$4 )"B"B1A"B"B"BD #(D ) ) ) )r- r/ ra rD s @r, r r s * * * * *, , , , , ,
0 0 0 0 0 0
) ) ) ) ) ) )r- r c , e Zd ZdZ fdZ fdZ xZS )r z.A `!dict` subclass representing a data record.c |r.t |d t r|d }|dd }nd } t j |i | |3|j r|j s| |j | t < d S d S r )rg _cursorr$ r% r] r r1 r )r( r) r* r r+ s r, r% zRealDictRow.__init__ s JtAw00 !WF8DDF$)&))) !
&&*?
&##%%%
!' 5D r- c t | v rU| t }t || | |t | dz
k r| t = d S t || d S )Nr )r r$ rm r_ )r( keyvaluemappingr+ s r, rm zRealDictRow.__setitem__ sy $ ;'G GG e 4 4 4c'llQ&&&% F
C'''''r- )r? r@ rA rB r% rm rC rD s @r, r r sW 886 6 6 6 6&
(
(
(
(
(
(
(
(
(r- r c " e Zd ZdZ fdZ xZS )NamedTupleConnectionz8A connection that uses `NamedTupleCursor` automatically.c z | d| j pt t j |i |S rH )rJ rI NamedTupleCursorr$ r rL s r, r zNamedTupleConnection.cursor s> *D,?,SCSTTT uww~t.v...r- rM rD s @r, r r s> BB/ / / / / / / / /r- r c e Zd ZdZdZdZd
fd Z fdZd
fd Z fdZ d
fd Z
fd Z fd
Z d Z
ed Z xZS )r a A cursor that generates results as `~collections.namedtuple`.
`!fetch*()` methods will return named tuples instead of regular tuples, so
their elements can be accessed both as regular numeric items as well as
attributes.
>>> nt_cur = conn.cursor(cursor_factory=psycopg2.extras.NamedTupleCursor)
>>> rec = nt_cur.fetchone()
>>> rec
Record(id=1, num=100, data="abc'def")
>>> rec[1]
100
>>> rec.data
"abc'def"
Ni c V d | _ t || S r/ )Recordr$ rT rU s r, rT zNamedTupleCursor.execute6 s" wwud+++r- c V d | _ t || S r/ )r r$ executemanyrU s r, r z NamedTupleCursor.executemany: s$ ww""5$///r- c V d | _ t || S r/ )r r$ rY rZ s r, rY zNamedTupleCursor.callproc> s$ ww $///r- c t }|9| j }|| x}| _ | | S d S r/ )r$ r0 r _make_nt_make)r( tntr+ s r, r0 zNamedTupleCursor.fetchoneB sT GG =Bz#'==??2T[88A;; =r- c t | }| j }|| x}| _ t t |j | S r/ )r$ r6 r r listmapr )r( r7 tsr r+ s r, r6 zNamedTupleCursor.fetchmanyJ sT
WW
t
$
$
[
: #}} .BC"%%&&&r- c t }| j }|| x}| _ t t |j | S r/ )r$ r9 r r r r r )r( r r r+ s r, r9 zNamedTupleCursor.fetchallQ sR
WW
[
: #}} .BC"%%&&&r- c # L K t }t | }| j }|| x}| _ | | V | t | V %# t $ r Y d S w xY wr/ )r$ r; r< r r r r= )r( itr r r+ s r, r; zNamedTupleCursor.__iter__X s
!!##BRABz#'==??2T[((1++
)hhtBxx(((((
) FF s BB
B#"B#c z | j r t d | j D nd}| | S )Nc 3 & K | ] }|d V
dS )r Nr r s r, ru z,NamedTupleCursor._make_nt..i s& 33QAaD333333r- r )r] tuple_cached_make_nt)r( r s r, r zNamedTupleCursor._make_nth sB 7;7GOe33$"2333333R##C(((r- c g }|D ]Y}t d| }|d dk sd|d cxk rdk rn nd|z }| | Zt d| }|S )N_r 09fr ) _re_cleansubappendr )clsr fieldssr s r, _do_make_ntz NamedTupleCursor._do_make_ntl s A
c1%%A ts{{cQqT 0 0 0 0S 0 0 0 0 0!G MM!
&
)
) r- r/ )r? r@ rA rB r MAX_CACHErT r rY r0 r6 r9 r; r classmethodr rC rD s @r, r r # s FI, , , , , ,0 0 0 0 00 0 0 0 0 0 ' ' ' ' ' '' ' ' ' ' ) ) ) [ r- r i c , | | S r/ )r )r r s r, r r { s ??3 r- c @ e Zd ZdZd Zd Zd Zd Zd Z fdZ xZ
S )LoggingConnectionz~A connection that logs all queries to a file or logger__ object.
.. __: https://docs.python.org/library/logging.html
c || _ t r4t |t j t j f r| j | _ dS | j | _ dS )zInitialize the connection to log to `!logobj`.
The `!logobj` parameter can be an open file object or a Logger/LoggerAdapter
instance from the standard logging module.
N)_logobj_loggingrg Logger
LoggerAdapter _logtologgerlog
_logtofile)r( logobjs r,
initializez LoggingConnection.initialize sP '
(*@AC C '(DHHHDHHHr- c |S )zFilter the query before logging it.
This is the method to overwrite to filter unwanted queries out of the
log or to add some extra data to the output. The default implementation
just does nothing.
r r( msgcurss r, filterzLoggingConnection.filter s
r- c | || }|rit |t r+| t j | j d }| j |t j
z d S d S )Nreplace) r rg bytesdecode_ext encodingsencodingr write_oslinesepr s r, r z LoggingConnection._logtofile sy kk#t$$ 2#u%%
Kjj
!> JJ L sS[ 0 1 1 1 1 1 2 2r- c n | || }|r | j | d S d S r/ )r r debugr s r, r z LoggingConnection._logtologger sC kk#t$$ $ L s # # # # # $ $r- c P t | d s| d d S )Nr z3LoggingConnection object has not been initialize()d)hasattrProgrammingErrorr{ s r, _checkzLoggingConnection._check s? tY'' G''EG G
G G Gr- c | | d| j pt t j |i |S rH )r rJ rI
LoggingCursorr$ r rL s r, r zLoggingConnection.cursor sJ
*D,?,P=QQQ uww~t.v...r- ) r? r@ rA rB r r r r r r rC rD s @r, r r s
' ' ' 2 2 2$ $ $
G G G
/ / / / / / / / /r- r c 0 e Zd ZdZd fd Zd fd Z xZS )r zCA cursor that logs queries using its connection logging facilities.Nc t || | j | j | S # | j | j | w xY wr/ )r$ rT r r rV rU s r, rT zLoggingCursor.execute sZ 277??5$// O
D 1 1 1 1DO
D 1 1 1 1 !A "A'c t || | j | j | S # | j | j | w xY wr/ )r$ rY r r rV rZ s r, rY zLoggingCursor.callproc s\ 277##Hd33 O
D 1 1 1 1DO
D 1 1 1 1r r/ )r? r@ rA rB rT rY rC rD s @r, r r sa MM2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2r- r c & e Zd ZdZddZd Zd ZdS )MinTimeLoggingConnectiona A connection that logs queries based on execution time.
This is just an example of how to sub-class `LoggingConnection` to
provide some extra filtering for the logged queries. Both the
`initialize()` and `filter()` methods are overwritten to make sure
that only queries executing for more than ``mintime`` ms are logged.
Note that this connection uses the specialized cursor
`MinTimeLoggingCursor`.
r c J t | | || _ d S r/ )r r _mintime)r( r mintimes r, r z#MinTimeLoggingConnection.initialize s# $$T6222
r- c t j |j z
dz }|| j k rTt |t
r+| t j | j d }| t j d| dS d S )Ni r z (execution time: z ms)) _timetime timestampr rg r r r r r r r )r( r r r s r, r z MinTimeLoggingConnection.filter s{
Z\\DN
*d 2 t} #u%%
Kjj
!> JJB3;BB1BBB B r- c p | d| j pt t j | g|R i |S rH )rJ rI MinTimeLoggingCursorr r )r( r) r* s r, r z MinTimeLoggingConnection.cursor sJ * 7#7 9 9 9 '>t>>>v>>>r- Nr )r? r@ rA rB r r r r r- r, r r sS C C C? ? ? ? ?r- r c " e Zd ZdZddZddZdS )r z=The cursor sub-class companion to `MinTimeLoggingConnection`.Nc j t j | _ t | || S r/ )r r r r rT )r( rV rW s r, rT z MinTimeLoggingCursor.execute s' $$T5$777r- c j t j | _ t | || S r/ )r r r r rY )r( r[ rW s r, rY z MinTimeLoggingCursor.callproc s' %%dHd;;;r- r/ )r? r@ rA rB rT rY r r- r, r r sB GG8 8 8 8< < < < < >>SVR >>>CC "_z1RRRi*6LRRRC3 q== /// /EG G G
.H.. .G
000 /SU U U
tOG
B
B1 '',, $tOGk!T 2 2AARAZZAAA sNG %% FO & E E E E Er- c 4 | j S r/ )r filenor{ s r, r9 zReplicationCursor.filenol s %%'''r- NN)NNr r NFr" )r? r@ rA rB r r! r7 r9 r r- r, r r
sm EE < =>DF8E 8E 8E 8Ev( ( ( ( (r- r c * e Zd ZdZd Zd Zd Zd ZdS ) UUID_adapterzAdapt Python's uuid.UUID__ type to PostgreSQL's uuid__.
.. __: https://docs.python.org/library/uuid.html
.. __: https://www.postgresql.org/docs/current/static/datatype-uuid.html
c || _ d S r/ _uuid)r( uuids r, r% zUUID_adapter.__init__y
r- c & |t j u r| S d S r/ r ISQLQuoter( protos r, __conform__zUUID_adapter.__conform__| DN " "K # "r- c > d| j d d S )N''::uuidutf8)r? encoder{ s r, getquotedzUUID_adapter.getquoted s# 'DJ'''//777r- c d| j dS )NrJ rK r> r{ s r, __str__zUUID_adapter.__str__ s &4:&&&&r- N)r? r@ rA rB r% rG rN rP r r- r, r< r< r sZ 8 8 8' ' ' ' 'r- r< c ddl | sd}d}n&t | t t f r| \ }}n| }d}t j |fdfd t _ t j |fdt j t _ t j t j | t j t j | t j
j t t j S )aR Create the UUID type and an uuid.UUID adapter.
:param oids: oid for the PostgreSQL :sql:`uuid` type, or 2-items sequence
with oids of the type and the array. If not specified, use PostgreSQL
standard oids.
:param conn_or_curs: where to register the typecaster. If not specified,
register it globally.
r Ni i UUIDc 6 | r | pd S r/ )rR )r r r@ s r, z register_uuid.. s !9$))D//!AT r- zUUID[]) r@ rg r r r new_typerR new_array_type UUIDARRAY
register_typeregister_adapterr< )oids conn_or_cursoid1oid2r@ s @r,
register_uuidr^ s KKK D4- ( (
dd
th A A A AC CDI($(DIFFDNty,///t~|444$)\222 9r- c 6 e Zd ZdZd Zd Zd Zd Zd Zd Z dS ) InetaC Wrap a string to allow for correct SQL-quoting of inet values.
Note that this adapter does NOT check the passed value to make
sure it really is an inet-compatible address but DOES call adapt()
on it to make sure it is impossible to execute an SQL-injection
by passing an evil value to the initializer.
c || _ d S r/ )addr)r( rb s r, r% z
Inet.__init__ s
r- c 0 | j j d| j dS )Nr) r* )r+ r? rb r{ s r, __repr__z
Inet.__repr__ s .)::DI::::r- c || _ d S r/ )_connr( conns r, preparez Inet.prepare rA r- c t | j }t |d r| | j | dz S )Nri s ::inet)r0 rb r ri rf rN )r( objs r, rN zInet.getquoted sH mm 3 " " $ KK
# # #}}**r- c & |t j u r| S d S r/ rC rE s r, rG zInet.__conform__ rH r- c * t | j S r/ )r- rb r{ s r, rP z Inet.__str__ s 49~~ r- N)
r? r@ rA rB r% rd ri rN rG rP r r- r, r` r` sx ; ; ; + + + r- r` c ddl }| dt | sd}d}n&t | t t
f r| \ }}n| }d}t
j |fdd t _ t
j |fdt j t _
t
j t j | t
j t j
| t j S ) aL Create the INET type and an Inet adapter.
:param oid: oid for the PostgreSQL :sql:`inet` type, or 2-items sequence
with oids of the type and the array. If not specified, use PostgreSQL
standard oids.
:param conn_or_curs: where to register the typecaster. If not specified,
register it globally.
r Nz4the inet adapter is deprecated, it's not very usefulie i INETc ( | rt | pd S r/ )r` )r r s r, rT z register_inet.. s !4$t**!< r- INETARRAY) warningswarnDeprecationWarningrg r r r rU ro rV rq rX )oidr[ rr r\ r] s r,
register_inetrv s OOO MM> C$ ' '
dd
th < <> >DI($; JJDNty,///t~|444 9r- c ddl }ddlm}m}m} | }||k rdS ||k r+| | gg g nI||k r+| g | gg n| d| n$# t $ r | Y w xY w)aR Wait until a connection or cursor has data available.
The function is an example of a wait callback to be registered with
`~psycopg2.extensions.set_wait_callback()`. This function uses
:py:func:`~select.select()` to wait for data to become available, and
therefore is able to handle/receive SIGINT/KeyboardInterrupt.
r N)POLL_OK POLL_READ
POLL_WRITETzbad state from poll: )
selectpsycopg2.extensionsrx ry rz pollr9 OperationalErrorKeyboardInterruptcancel)rh r{ rx ry rz states r, wait_selectr s MMMBBBBBBBBBB
IIKKE )##
t{{}}or26666*$$
b4;;==/26666++,KE,K,KLLL KKMMM H s B( A:B( ( C C c | t j d t | d r#| j }| t
}n | }| t
}||fS )zEReturn the connection and a DBAPI cursor from a connection or cursor.Nz no connection or cursor providedrT )rI )r r r r r r )r[ rh r s r, _solve_conn_cursr sk '(JKKK|Y'' 3&{{'{22{{'{22 :r- c e Zd ZdZd Zd Zd Zd ZeZ e j
de j Z e
e j
d fd Ze
d Ze
d
Zd S )
HstoreAdapterz)Adapt a Python dict to the hstore syntax.c || _ d S r/ )wrapped)r( r s r, r% zHstoreAdapter.__init__ s
r- c P || _ |j j dk r| j | _ d S d S )Ni_ )rh infoserver_version _getquoted_8rN rg s r, ri zHstoreAdapter.prepare s1 9 #e + + ! .DNNN , +r- c | j sdS t j }g }| j D ]\ }} || }| | j | }|: || }| | j | }nd}| d|z dz |z dz dd | z dz S )z*Use the operators available in PG pre-9.0.
''::hstoreNs NULL (s => )s ||) r r r rx ri rh rN r join)r( r rvr6 rn s r, r zHstoreAdapter._getquoted_8 s | ! =
L&&((
5
5DAqaA
IIdi
A}E!HH $)$$$KKMM
IIdQh(1,t3 4 4 4 4ejjnn$t++r- c | j sdS t j t | j }| | j t j t | j }| | j d| z dz | z dz S )z(Use the hstore(text[], text[]) function.r s hstore(s , r ) r r r r r| ri rh r rN )r( r6 rn s r, _getquoted_9zHstoreAdapter._getquoted_98 s | ! = JtDL - - / /00 1 1 $) JtDL / / 1 122 3 3 $) AKKMM)E1AKKMMADHHr- am
# hstore key:
# a string of normal or escaped chars
"((?: [^"\\] | \\. )*)"
\s*=>\s* # hstore value
(?:
NULL # the value can be null - not catched
# or a quoted string like the key
| "((?: [^"\\] | \\. )*)"
)
(?:\s*,\s*|$) # pairs separated by comma or end of string.
z\\(.)c |dS i }d}| j | D ]}|| |k rt j d| | d| d }| d }|| d| }|||< | }|t | k rt j d| |S )zParse an hstore representation in a Python string.
The hstore is represented as something like::
"a"=>"1", "b"=>"2"
with backslash-escaped strings.
Nr z"error parsing hstore pair at char \1r z/error parsing hstore: unparsed data after char )
_re_hstorefinditerstartr InterfaceErrorr groupendr_ ) r( r cur_bsdecr r mr6 rn s r, parsezHstoreAdapter.parseR s
94
))!,,
AyAGGII . . -@@@B B B
5!''!**--A
A}JJua((BqEEEGGEE 3q66>>)I%IIK K
K r- c |dS | t j |j j }| || S )z2Parse an hstore returning unicode keys and values.N)r r r r r r )r( r r s r,
parse_unicodezHstoreAdapter.parse_unicodes s?
94
HHT^CN$;< = =zz!S!!!r- c t | \ }}|j }|j j dk rdpd}g g }}| d| d |D ]8}| |d | |d 9|t j k r|j s| t | t | fS )zBReturn the lists of OID of the hstore and hstore[] types.
9 typarrayNULLzSELECT t.oid, z\
FROM pg_type t JOIN pg_namespace ns
ON typnamespace = ns.oid
WHERE typname = 'hstore';
r r ) r statusr r rT r r STATUS_IN_TRANSACTION
autocommitrollbackr ) r( r[ rh r conn_statusr rv0rv1rZ s r, get_oidszHstoreAdapter.get_oids| s &l33
d k 9+u4CMvrS
D JJtAw JJtAw
45 5 5
6 MMOOOSzz5::%%r- N)r? r@ rA rB r% ri r r rN _recompileVERBOSEr r r r r r r- r, r r s 33 / / /, , ,0 I I I I
J #.3;x#8#8 [ @ " " [" & & [ & & &r- r Fc L |Ht | }||d st j d |d }|d }t |t
r|f}|2t |t
r|f}nt
d |D }t j |dt j }t j
|| r| pd t j t t |r2t j
|d| }t j
|| r| pd dS dS )a Register adapter and typecaster for `!dict`\-\ |hstore| conversions.
:param conn_or_curs: a connection or cursor: the typecaster will be
registered only on this object unless *globally* is set to `!True`
:param globally: register the adapter globally, not only on *conn_or_curs*
:param unicode: if `!True`, keys and values returned from the database
will be `!unicode` instead of `!str`. The option is not available on
Python 3
:param oid: the OID of the |hstore| type if known. If not, it will be
queried on *conn_or_curs*.
:param array_oid: the OID of the |hstore| array type if known. If not, it
will be queried on *conn_or_curs*.
The connection or cursor passed to the function will be used to query the
database and look for the OID of the |hstore| type (which may be different
across databases). If querying is not desirable (e.g. with
:ref:`asynchronous connections `) you may specify it in the
*oid* parameter, which can be found using a query such as :sql:`SELECT
'hstore'::regtype::oid`. Analogously you can obtain a value for *array_oid*
using a query such as :sql:`SELECT 'hstore[]'::regtype::oid`.
Note that, when passing a dictionary from Python to the database, both
strings and unicode keys and values are supported. Dictionaries returned
from the database have keys/values according to the *unicode* parameter.
The |hstore| contrib module must be already installed in the database
(executing the ``hstore.sql`` script in your ``contrib`` directory).
Raise `~psycopg2.ProgrammingError` if the type is not found.
Nr z\hstore type not found in the database. please install it from your 'contrib/hstore.sql' filer c g | ]}||S r r )rr rk s r, r z#register_hstore.. s 9 9 9Qq 9q 9 9 9r- HSTORE HSTOREARRAY)r r r r rg rh r r rU r rX rY dictrV )r[ globallyunicoderu array_oidr r s r, register_hstorer sI > {$$\22 ;c!f;+HI I
I AIa&C#s f i % % ;" II 9 9) 9 9 9::I ]3-*=
>
>Fv8| < DEEE$
... O))]FKK ;H (E (MNNNNNO Or- c e Zd ZdZd dZd Zd Z ej dej Z
ej d Z e d Z
d Ze d
ZdS ) CompositeCasteram Helps conversion of a PostgreSQL composite type into a Python object.
The class is usually created by the `register_composite()` function.
You may want to create and register manually instances of the class if
querying the database at registration time is not desirable (such as when
using an :ref:`asynchronous connections `).
Nc X || _ || _ || _ || _ d |D | _ d |D | _ | || j t j |f|| j | _
|r&t j |f| d| j
| _ d S d | _ d S )Nc g | ]
}|d S r r rr as r, r z,CompositeCaster.__init__.. ---!1---r- c g | ]
}|d S )r r r s r, r z,CompositeCaster.__init__.. r r- ARRAY)
nameschemaru r attnamesatttypes _create_typer rU r
typecasterrV array_typecaster)r( r ru attrsr r s r, r% zCompositeCaster.__init__ s "--u---
--u---
$
... -djAA )$($7 nnndo%? %?D ! ! ! %)D ! ! !r- c h |d S | | }t | t | j k r>t j dt | j | j t | fz fdt
| j | D }| | S )Nz9expecting %d components for the type %s, %d found insteadc B g | ]\ }} || S r )cast)rr ru tokenr s r, r z)CompositeCaster.parse.. s; : : :U ))C'' : : :r- )tokenizer_ r r DataErrorr zipmake)r( r r tokensr s ` r, r zCompositeCaster.parse s 94 q!! v;;#dm,, , ,$KT]##TYF <=> >
>: : : : !$- 8 8: : : yy r- c , | | S )a Return a new Python object representing the data being casted.
*values* is the list of attributes, already casted into their Python
representation.
You can subclass this method to :ref:`customize the composite cast
`.
)_ctor)r( r s r, r zCompositeCaster.make s zz&!!!r- z
\(? ([,)]) # an empty token, representing NULL
| \(? " ((?: [^"] | "")*) " [,)] # or a quoted string
| \(? ([^",)]+) [,)] # or an unquoted string
z (["\\])\1c g }| j | D ]}|t j d| | d | d F| d B| | j d| d | | d |S )Nzcan't parse type: r r r ) _re_tokenizer r r r r _re_undoubler )r( r r r s r, r zCompositeCaster.tokenize s
"++A.. & &Ay -.H1.H.HIIIwwqzz% $' $+//qwwqzzBBCCCC !''!**%%%% r- c t d| }t || | _ | j j | _ d S )Nr )r r r r, r r )r( r r s r, r z CompositeCaster._create_type# s5 }}S$'' tX.. Y_
r- c t | \ }}|j }d|v r| dd \ }}n|}d}|j j dk rdpd}| d|z ||f | } | s d}
|j t j k r| d d
}
| d |z |f | } | r | d d
}| d d }n# t j
$ r Y nw xY w|
r| d n # |
r| d w w xY w|t j k r|j s| | st j
d| d | d d } | d d } d | D }
| || |
| | S )z}Return a `CompositeCaster` instance for the type *name*.
Raise `ProgrammingError` if the type is not found.
.r publicr r r zSELECT t.oid, %s, attname, atttypid
FROM pg_type t
JOIN pg_namespace ns ON typnamespace = ns.oid
JOIN pg_attribute a ON attrelid = typrelid
WHERE typname = %%s AND nspname = %%s
AND attnum > 0 AND NOT attisdropped
ORDER BY attnum;
FzSAVEPOINT register_typeTzSELECT t.oid, %s, attname, atttypid, typname, nspname
FROM pg_type t
JOIN pg_namespace ns ON typnamespace = ns.oid
JOIN pg_attribute a ON attrelid = typrelid
WHERE t.oid = %%s::regtype
AND attnum > 0 AND NOT attisdropped
ORDER BY attnum;
r z#ROLLBACK TO SAVEPOINT register_typezPostgreSQL type 'z ' not foundc . g | ]}|d |d fS )r r r )rr rs r, r z,CompositeCaster._from_db..u s$ 111qqtQqTl111r- )r r )
r r r. r r rT r9 r r r r r r )r( r r[ rh r r r tnamer recs savepointtype_oidr
type_attrss r, _from_dbzCompositeCaster._from_db( sS &l33
d k $;; JJsA . .MFEEE F 9+u4CMv
! ! ! }} H
H ! ;$"<<<LL!:;;; $I }} ( GAJE !!WQZF
,
HLL!FGGG HLL!FGGGGH $4 4 4T_ 4 MMOOO 7+5D5557 7
7 71:GAJ 11D111
tE8Z 0 0 0 0s+ AC=
3D* =D D* DD* *Er: )r? r@ rA rB r% r r r r r r r r r r r r r- r, r r s ) ) ) )
!
!
!
"
"
" 3;
L 3;|,,L [ % % %
O0 O0 [O0 O0 O0r- r c |t }| | | }t j |j | r|pd |j t j |j | r|pd |S )a Register a typecaster to convert a composite type into a tuple.
:param name: the name of a PostgreSQL composite type, e.g. created using
the |CREATE TYPE|_ command
:param conn_or_curs: a connection or cursor used to find the type oid and
components; the typecaster is registered in a scope limited to this
object, unless *globally* is set to `!True`
:param globally: if `!False` (default) register the typecaster only on
*conn_or_curs*, otherwise register it globally
:param factory: if specified it should be a `CompositeCaster` subclass: use
it to :ref:`customize how to cast composite types `
:return: the registered `CompositeCaster` or *factory* instance
responsible for the conversion
N)r r r rX r r )r r[ r factorycasters r, register_compositer { s !
dL
1
1Fv(h,*G<*O4PPP
* #\%Bl%Jd L L L Mr- c # K g }t | } t | D ]$}| t | %|V g }n# t $ r
|r|V Y dS w xY wT)zwConsume an iterable and return it in chunks.
Every chunk is at most `page_size`. Never return an empty chunk.
TN)rz r^ r r< r= )seq page_sizepager r` s r, _paginater s
D
cB 9%%
&
& DHH%%%%JJJDD
FF
s :A A%$A%d c t || D ]9} fd|D } d | :dS )a Execute groups of statements in fewer server roundtrips.
Execute *sql* several times, against all parameters set (sequences or
mappings) found in *argslist*.
The function is semantically similar to
.. parsed-literal::
*cur*\.\ `~cursor.executemany`\ (\ *sql*\ , *argslist*\ )
but has a different implementation: Psycopg will join the statements into
fewer multi-statement commands, each one containing at most *page_size*
statements, resulting in a reduced number of server roundtrips.
After the execution of the function the `cursor.rowcount` property will
**not** contain a total result.
r c < g | ]} | S r )mogrify)rr r) r sqls r, r z!execute_batch.. s' 8884 C&&888r- ;N)r rT r )r r argslistr r sqlss `` r,
execute_batchr sg ( (i888 % %888884888 DIIdOO$$$$% %r- c ddl m} t || r| | }t |t s/| t j | j j }t | \ }}|rg nd} t || D ]}
|2dd dgt |
d z z dz }|dd } |
D ]@} | | || | d A|| d d< | d
| |r'| | | S ) aA Execute a statement using :sql:`VALUES` with a sequence of parameters.
:param cur: the cursor to use to execute the query.
:param sql: the query to execute. It must contain a single ``%s``
placeholder, which will be replaced by a `VALUES list`__.
Example: ``"INSERT INTO mytable (id, f1, f2) VALUES %s"``.
:param argslist: sequence of sequences or dictionaries with the arguments
to send to the query. The type and content must be consistent with
*template*.
:param template: the snippet to merge to every item in *argslist* to
compose the query.
- If the *argslist* items are sequences it should contain positional
placeholders (e.g. ``"(%s, %s, %s)"``, or ``"(%s, %s, 42)``" if there
are constants value...).
- If the *argslist* items are mappings it should contain named
placeholders (e.g. ``"(%(id)s, %(f1)s, 42)"``).
If not specified, assume the arguments are sequence and use a simple
positional template (i.e. ``(%s, %s, ...)``), with the number of
placeholders sniffed by the first element in *argslist*.
:param page_size: maximum number of *argslist* items to include in every
statement. If there are more items the function will execute more than
one statement.
:param fetch: if `!True` return the query results into a list (like in a
`~cursor.fetchall()`). Useful for queries with :sql:`RETURNING`
clause.
.. __: https://www.postgresql.org/docs/current/static/queries-values.html
After the execution of the function the `cursor.rowcount` property will
**not** contain a total result.
While :sql:`INSERT` is an obvious candidate for this function it is
possible to use it with other statements, for example::
>>> cur.execute(
... "create table test (id int primary key, v1 int, v2 int)")
>>> execute_values(cur,
... "INSERT INTO test (id, v1, v2) VALUES %s",
... [(1, 2, 3), (4, 5, 6), (7, 8, 9)])
>>> execute_values(cur,
... """UPDATE test SET v1 = data.v1 FROM (VALUES %s) AS data (id, v1)
... WHERE test.id = data.id""",
... [(1, 20), (4, 50)])
>>> cur.execute("select * from test order by id")
>>> cur.fetchall()
[(1, 20, 3), (4, 50, 6), (7, 8, 9)])
r )
ComposableNr r ,s %sr r- ) psycopg2.sqlr rg as_stringr rM r r r r
_split_sqlr r r_ r r rT extendr9 )
r r r templater fetchr prepostresultr partsr) s
r, execute_valuesr sy x ('''''#z"" !mmC
c5 ! ! Bjj (?@AA3IC
"RRdF(i888
*
* dii#d1g,,(> ? ??$FHAAA D LLXt44 5 5 5 LL bcc
CHHUOO$$$ * MM#,,.. ) ) ) Mr- c g x}}g }t j d| }|D ]}t | dk s|dd dk r| | 9|dd dk r||u r|}Nt d |dd dk r| d t d|dd d d
z ||u rt d ||fS ) zSplit *sql* on a single ``%s`` placeholder.
Split on the %s, perform %% replacement and return pre, post lists of
snippets.
s (%.)r Nr % sz1the query contains more than one '%s' placeholderz"unsupported format character: '%s'asciir z.the query doesn't contain any '%s' placeholder)r r. r_ r
ValueErrorr )r currr r r r s r, r r s$ OD3
D
Yx
%
%F 8 8 u::??eBQBi4 / / KK 9 s{{ GI I I
122Y$
KK A)""7I66 7 8 8
8 s{{IJJJ 9r- [z !"#$%&'()*+,-./:;<=>?@[\]^`{|}~]r: )FFNN)FN)r )Nr F)VrB osr r r rer collectionsr r loggingr r r r r r r _connectionr r0 r
functoolsr psycopg2._psycopgr r
r _replicationConnectionr _replicationCursorr psycopg2._jsonr r r r r psycopg2._ranger r r r r r r r psycopg2._ipaddressr r rF rK r rP r r r r r r r r r r r r r r r r< r^ r` rv r r r r r r r r r r r escaper r r- r, r! s 8 / / / / / / / / ' ' ' ' ' ' ) ) ) ) ) ) 1 1 1 1 1 1 0 0 0 0 0 0 0 0 N N N N N N N N N N N N N N
/ / / / / / / / / / / / / / / / / / / / 3 2 2 2 2 29 9 9 9 9W 9 9 9x/ / / / /[ / / /) ) ) ) ) ) ) )83 3 3 3 3 d 3 3 3 l/ / / / / / / /) ) ) ) )^ ) ) )8 ( ( ( ( (+ ( ( (F/ / / / /; / / /U U U U Uw U U Up 3 $/;#?#? 0/ 0/ 0/ 0/ 0/ 0/ 0/ 0/f
2
2
2
2
2G
2
2
2 ? ? ? ? ? 0 ? ? ?8 < < < < <= < < <* * * * *#9 * * ** * * * *$: * * * i c( c( c( c( c(* c( c( c(P' ' ' ' ' ' ' '* D > B 8 E& E& E& E& E& E& E& E&P ;@(,9O 9O 9O 9Ox`0 `0 `0 `0 `0 `0 `0 `0F 8 &% % % %2T T T Tn B
CK*#*9
:
::S@
B
B r-