$g9dZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z ddl mZddl mZdd lmZmZmZmZdd lmZmZmZdd lmZdd lmZdd lmZddl m!Z!ddl"m#Z#er ddl$m%Z%ddl&m'Z'dZ(e j)j*Z*e j)j+Z+e j,j-Z-e j,j.Z.e j/j0Z0e j/j1Z1Gdde!eefZ2Gdde2defeeZ3Gdde2defe#eZ4dS)z% psycopg server-side cursor objects. ) annotations)Any AsyncIteratorIterableIterator TYPE_CHECKINGoverload)warn)pq)sql)errors)ConnectionTypeQueryParamsPQGen)Row RowFactoryAsyncRowFactory)Cursor)Self)execute) BaseCursor) AsyncCursor) Connection)AsyncConnectiondceZdZdZdZd$d Zd%fd Zed%d Z ed&d Z ed'dZ ed(dZ d)d*dZ d+dZd+dZd,dZd-d!Zd.d#ZxZS)/ServerCursorMixinzDMixin to add ServerCursor behaviour and implementation a BaseCursor.z7_name _scrollable _withhold _described itersize _formatnamestr scrollable bool | Nonewithholdboolcn||_||_||_d|_t|_t |_dS)NF)_name _scrollable _withhold _describedDEFAULT_ITERSIZEitersizeTEXT_format)selfr r"r$s U/var/www/html/netbox-4.1.3/venv/lib/python3.11/site-packages/psycopg/server_cursor.py__init__zServerCursorMixin.__init__,s3  %!-  returnctdd}|d|jd|S)Nr  )super__repr__splitinsertr'join)r/parts __class__s r0r7zServerCursorMixin.__repr__9sQ  ""((q11 Q4:)***xxr2c|jS)zThe name of the cursor.)r'r/s r0r zServerCursorMixin.name?s zr2c|jS)z Whether the cursor is scrollable or not. If `!None` leave the choice to the server. Use `!True` if you want to use `scroll()` on the cursor. )r(r>s r0r"zServerCursorMixin.scrollableDs r2c|jS)zY If the cursor can be used after the creating transaction has committed. )r)r>s r0r$zServerCursorMixin.withholdNs ~r2 int | Nonec`|j}|o|jtkp |jdk}|r|jndS)zoIndex of the next row to fetch in the current result. `!None` if there is no result to fetch. r2N)pgresultstatus TUPLES_OKcommand_status_pos)r/restupless r0 rownumberzServerCursorMixin.rownumberUs< mO#* 1NS5G35N",tyy,r2Nqueryrparams Params | Nonebinary PQGen[None]c#0K||}|jr!|Ed{Vd|_||Ed{V|||}||dt |jjEd{V}|dj tkr| |d| |j |_ n|rtnt|_ |Ed{VdS)z0Generator implementing `ServerCursor.execute()`.NFT)force_extended)_make_declare_statementr* _close_gen _start_query_convert_query _execute_sendr_connpgconnrD COMMAND_OK_raise_for_resultformatr.BINARYr- _describe_gen)r/rKrLrNpgqresultss r0 _declare_genzServerCursorMixin._declare_genbs@,,U33 ? $(( ( ( ( ( ( ( (#DO$$U+++++++++!!%00 3t444$TZ%677777777 2;  + +  " "72; / / / >;DLL%+566DL%%'''''''''''r2c#*K|j|j|jt |jEd{V}||||_|d|j d|_ dS)Nr)r\T) _pgconnsend_describe_portalr'encode _encodingr_check_results_results_select_current_resultr.r*)r/r`s r0r^zServerCursorMixin._describe_gens ))$**;*;DN*K*KLLL$T\22222222 G$$$  ##Adl#;;;r2c#,K|jjj}|tkr |tkrdS|js |tkrdS|jsotjd tj |j }|j |Ed{V}|J|j dkrdStjd tj|j }|j |Ed{VdS)Nz3SELECT 1 FROM pg_catalog.pg_cursors WHERE name = {}rzCLOSE {})rXrYtransaction_statusIDLEINTRANSr)r*r SQLr\Literalr' _exec_commandntuples Identifier)r/tsrKrHs r0rTzServerCursorMixin._close_gens Z  1 ::"-- F~ "** F GEfS[,,-- "Z55e<<<<<<<$*+E+EFF:++E22222222222r2numPQGen[list[Row]]c#hK|jrtjd|js4|Ed{V|Ed{Vt jd|t jdnt j |t j |j }|j ||jEd{V}|J||_|j|d|jd|j|jS)Nzthe cursor is closedzFETCH FORWARD {} FROM {}ALL) result_formatF) set_loadersr)closedeInterfaceErrorr*rUr^r rnr\rorrr'rXrpr.rC_tx set_pgresult load_rowsrq _make_row)r/rtrKrHs r0 _fetch_genzServerCursorMixin._fetch_gens9 ; ;"#9:: : ,((** * * * * * * *))++ + + + + + + +233::!kCGENNNs{3/?/? N4: & &  11%t|1TTTTTTTT  cu555x!!!S[$.AAAr2valueintmodec#PK|dvrtd|dtjdtj|dkrdndtj|tj|j}|j|Ed{VdS)N)relativeabsolutez bad mode: z'. It should be 'relative' or 'absolute'zMOVE{} {} FROM {}rz ABSOLUTE) ValueErrorr rnr\rorrr'rXrp)r/rrrKs r0 _scroll_genzServerCursorMixin._scroll_gens / / /W$WWWXX X+,,33 G4:#5#5KK2 > > K   N4: & &   :++E22222222222r2 sql.Composedct|tr||j}t|tjst j|}t jdt j|jg}|j 0| t j|j rdnd| t jd|j r'| t jd| t jd| |t jd |S)NDECLARESCROLLz NO SCROLLCURSORz WITH HOLDFORr5) isinstancebytesdecoderfr Composablernrrr'r(appendr)r:)r/rKr;s r0rSz)ServerCursorMixin._make_declare_statements* eU # # 1LL00E%00 #GENNE GI   N4: & &    ' LLT-=!N;OO P P P SWX&&''' > / LL-- . . . SWU^^$$$ Uws||  '''r2)r r!r"r#r$r%)r3r!)r3r#)r3r%)r3rA)NN)rKrrLrMrNr#r3rO)r3rO)rtrAr3ru)rrrr!r3rO)rKrr3r)__name__ __module__ __qualname____doc__r8 __slots__r1r7propertyr r"r$rJrar^rTrrrS __classcell__r<s@r0rr'sqNNIOOQQI     X   X X  - - -X -!%" (((((>33334BBBB(3333((((((((r2rceZdZdZdZedddd8dZedddd9dZddddd:dZd;dZd;fd Z dd)Z d?d+Z d@dAd0Z dBd1Z dCd3Z dDdEd7ZxZS)F ServerCursorpsycopgNFr"r$ connectionConnection[Row]r r!r"r#r$r%cdSNrr/rr r"r$s r0r1zServerCursor.__init__  sr2Connection[Any] row_factoryRowFactory[Row]cdSrrr/rr rr"r$s r0r1zServerCursor.__init__  sr2rr"r$RowFactory[Row] | Nonec|tj|||p|jt||||dSN)r)rr1rrrs r0r1zServerCursor.__init__sN  *+*O9O     ""4z8DDDDDr2r3NonecJ|jstd|dtdSdSNzthe server-side cursor z[ was deleted while still open. Please use 'with' or '.close()' to close the cursor properlyrzr ResourceWarningr>s r0__del__zServerCursor.__del__O{  P$PPP       r2c,|jj5|jr ddddS|jjs,|j|t ddddS#1swxYwYdS)zI Close the current cursor and free associated resources. NrXlockrzwaitrTr6closer/r<s r0rzServerCursor.closesZ_  {         :$ 3  1 1222 GGMMOOO                   s B AB  B B rNrKrrLrMrNkwargsrrc |r%tdt|d|jjrt jd |jj5|j| |||dddn #1swxYwYn,#tj $r}| dd}~wwxYw|S)zC Open a cursor to execute a query to the database. keyword not supported: r2server-side cursors not supported in pipeline modeN TypeErrorlistrcpipeline_statusr{NotSupportedErrorrXrrra _NO_TRACEBACKwith_tracebackr/rKrLrNrexs r0rzServerCursor.executes+  IGd6ll1oGGHH H < ' %D  * J J  1 1% H HIII J J J J J J J J J J J J J J J * * *##D)) ) * s< B0B BBBBBC,CCT returning params_seqIterable[Params]rc*tjd)z/Method not implemented for server-side cursors.0executemany not supported on server-side cursorsr{rr/rKrrs r0 executemanyzServerCursor.executemany*s!"TUUUr2 Row | Nonec|jj5|j|d}dddn #1swxYwY|r|xjdz c_|dSdSNr rrXrrrrGr/recss r0fetchonezServerCursor.fetchone4s Z_ 7 7:??4??1#5#566D 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7   IINII7N4.AA A rsizer list[Row]c|s|j}|jj5|j||}dddn #1swxYwY|xjt |z c_|Sr arraysizerXrrrrGlenr/rrs r0 fetchmanyzServerCursor.fetchmany=s ">D Z_ : ::??4??4#8#899D : : : : : : : : : : : : : : : SYY  s.AAAc|jj5|j|d}dddn #1swxYwY|xjt |z c_|SrrXrrrrGrrs r0fetchallzServerCursor.fetchallEs Z_ : ::??4??4#8#899D : : : : : : : : : : : : : : : SYY  r Iterator[Row]c#K |jj5|j||j}dddn #1swxYwY|D]}|xjdz c_|Vt ||jkrdSNTr rXrrrr,rGrr/rrecs r0__iter__zServerCursor.__iter__Ks  G Gztt}'E'EFF G G G G G G G G G G G G G G G   Q  4yy4=(( s3AAArrrc|jj5|j|||dddn #1swxYwY|dkr|xj|z c_dS||_dS)Nr)rXrrrrGr/rrs r0scrollzServerCursor.scrollUs Z_ ; ; JOOD,,UD99 : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; :   II IIIIDIIIs/AA A )rrr r!r"r#r$r%) rrr r!rrr"r#r$r%) rrr r!rrr"r#r$r%r3rr rKrrLrMrNr#rrr3rrKrrrrr%r3rr3rrrrr3rr3r)r3rrrrrr!r3r)rrrrr r1rrrrrrrrrrrs@r0rrsJI #'      X #'     X /3"& E E E E E E      !% # > VVVVVV r2rrceZdZdZdZedddd8dZedddd9dZddddd:dZd;dZd;fd Z dd)Z d?d+Z d@dAd0Z dBd1Z dCd3Z dDdEd7ZxZS)FAsyncServerCursorrrNFrrAsyncConnection[Row]r r!r"r#r$r%cdSrrrs r0r1zAsyncServerCursor.__init__err2AsyncConnection[Any]rAsyncRowFactory[Row]cdSrrrs r0r1zAsyncServerCursor.__init__orr2rAsyncRowFactory[Row] | Nonec|tj|||p|jt||||dSr)rr1rrrs r0r1zAsyncServerCursor.__init__zsO  *+*O9O     ""4z8DDDDDr2r3rcJ|jstd|dtdSdSrrr>s r0rzAsyncServerCursor.__del__rr2cxK|jj4d{V|jr dddd{VdS|jjs2|j|d{Vt d{Vdddd{VdS#1d{VswxYwYdSrrrs r0rzAsyncServerCursor.closes:? " " " " " " " "{  " " " " " " " " " " " " " ":$ 9joodoo&7&7888888888''--// ! ! ! ! ! ! !  " " " " " " " " " " " " " " " " " " " " " " " " " " " " " "s B)A$B)) B36B3rrKrrLrMrNrrrcK|r%tdt|d|jjrt jd |jj4d{V|j| |||d{Vdddd{Vn#1d{VswxYwYn,#tj $r}| dd}~wwxYw|S)Nrrrrrs r0rzAsyncServerCursor.executes  IGd6ll1oGGHH H < ' %D  *z P P P P P P P Pjood&7&7vv&N&NOOOOOOOOO P P P P P P P P P P P P P P P P P P P P P P P P P P P * * *##D)) ) * s< B76B%B7% B//B72B/3B77C CC Trrrrc.Ktjd)Nrrrs r0rzAsyncServerCursor.executemanys!"TUUUr2rcK|jj4d{V|j|dd{V}dddd{Vn#1d{VswxYwY|r|xjdz c_|dSdSrrrs r0rzAsyncServerCursor.fetchones:? = = = = = = = =););<<<<<<<D:? @ @ @ @ @ @ @ @)>)>????????D @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ SYY  s4A$$ A.1A.cK|jj4d{V|j|dd{V}dddd{Vn#1d{VswxYwY|xjt |z c_|Srrrs r0rzAsyncServerCursor.fetchalls.:? @ @ @ @ @ @ @ @)>)>????????D @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ SYY  r AsyncIterator[Row]cNK |jj4d{V|j||jd{V}dddd{Vn#1d{VswxYwY|D]}|xjdz c_|WVt ||jkrdSrrrs r0 __aiter__zAsyncServerCursor.__aiter__sb z M M M M M M M M!Z__T__T]-K-KLLLLLLLL M M M M M M M M M M M M M M M M M M M M M M M M M M M   Q  4yy4=(( s9A!! A+.A+rrrcK|jj4d{V|j|||d{Vdddd{VdS#1d{VswxYwYdSr)rXrrrrs r0rzAsyncServerCursor.scrolls*:? A A A A A A A A*//$"2"25$"?"?@@ @ @ @ @ @ @ @ A A A A A A A A A A A A A A A A A A A A A A A A A A A A A As5A A'*A')rrr r!r"r#r$r%) rrr r!rrr"r#r$r%) rrr r!rrr"r#r$r%rrrrrrrr)r3rrr)rrrrr r1rrrrrrrrrrrs@r0rr_sJI #'      X #'     X 48"& E E E E E E""""""!% # 8 VVVVVV AAAAAAAAAr2rrN)5r __future__rtypingrrrrrr warningsr rr r rr{abcrrrrrowsrrrcursorr_compatr generatorsr _cursor_baser cursor_asyncrrrconnection_asyncrr+Formatr-r] ExecStatusrZrETransactionStatusrlrmrrrrr2r0r s #"""""RRRRRRRRRRRRRRRR5555555555552222222222$$$$$$%%%%%%2&&&&&&111111 y~   ] % M #     &l(l(l(l(l( >3#67l(l(l(^FFFFF$%6%;