o `^h[@sddlZddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZedejejBZGdddZGd d d ZGd d d eeZGd ddZGdddeeZGdddeZGdddeeZdS)N) WarningErrorInterfaceError DataError DatabaseErrorOperationalErrorIntegrityError InternalErrorNotSupportedErrorProgrammingError)logger) FIELD_TYPEz|\s*((?:INSERT|REPLACE)\s.+\sVALUES?\s+)(\(\s*(?:%s|%\(.+\)s)\s*(?:,\s*(?:%s|%\(.+\)s)\s*)*\))(\s*(?:ON DUPLICATE.*)?);?\s*\Zc@seZdZdZdZdLddZeddZedd Zed d Z ed d Z eddZ e j ddZ eddZ eddZeddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&ZdMd(d)ZdMd*d+Zd,d-Zd.d/ZdNd1d2Zd3d4ZdMd5d6Zd7d8ZdOd:d;Z dd?Z"d@dAZ#dBdCZ$e%Z%e&Z&e'Z'e(Z(e)Z)e*Z*e+Z+e,Z,e-Z-e.Z.dDdEZ/dFdGZ0dHdIZ1dJdKZ2d'S)PCursorz-Cursor is used to interact with the database.iFcCsJ||_|jj|_d|_d|_d|_d|_d|_d|_d|_ d|_ ||_ dS)zfDo not create an instance of a Cursor yourself. Call connections.Connection.cursor(). Nrr ) _connectionloop_loop _description _rownumber _rowcount _arraysize _executed_result_rows _lastrowid_echo)self connectionechor N/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/aiomysql/cursors.py__init__$s  zCursor.__init__cC|jS)zmThis read-only attribute return a reference to the Connection object on which the cursor was created.rrr r r!r4szCursor.connectioncCr#)aThis read-only attribute is a sequence of 7-item sequences. Each of these sequences is a collections.namedtuple containing information describing one result column: 0. name: the name of the column returned. 1. type_code: the type of the column. 2. display_size: the actual length of the column in bytes. 3. internal_size: the size in bytes of the column associated to this column on the server. 4. precision: total number of significant digits in columns of type NUMERIC. None for other types. 5. scale: count of decimal digits in the fractional part in columns of type NUMERIC. None for other types. 6. null_ok: always None as not easy to retrieve from the libpq. This attribute will be None for operations that do not return rows or if the cursor has not had an operation invoked via the execute() method yet. )rr%r r r! description:szCursor.descriptioncCr#)aReturns the number of rows that has been produced of affected. This read-only attribute specifies the number of rows that the last :meth:`execute` produced (for Data Query Language statements like SELECT) or affected (for Data Manipulation Language statements like UPDATE or INSERT). The attribute is -1 in case no .execute() has been performed on the cursor or the row count of the last operation if it can't be determined by the interface. )rr%r r r!rowcountRs zCursor.rowcountcCr#)zRow index. This read-only attribute provides the current 0-based index of the cursor in the result set or ``None`` if the index cannot be determined. )rr%r r r! rownumbera zCursor.rownumbercCr#)How many rows will be returned by fetchmany() call. This read/write attribute specifies the number of rows to fetch at a time with fetchmany(). It defaults to 1 meaning to fetch a single row at a time. rr%r r r! arraysizelr)zCursor.arraysizecCs ||_dS)r*Nr+)rvalr r r!r,ws cCr#)ajThis read-only property returns the value generated for an AUTO_INCREMENT column by the previous INSERT or UPDATE statement or None when there is no such value available. For example, if you perform an INSERT into a table that contains an AUTO_INCREMENT column, lastrowid returns the AUTO_INCREMENT value for the new row. )rr%r r r! lastrowidszCursor.lastrowidcCr#)zReturn echo mode status.)rr%r r r!rsz Cursor.echocCs|jsdSdS)zlThe readonly property that returns ``True`` if connections was detached from current cursor TFr$r%r r r!closedsz Cursor.closedcsT|j}|dur dSz|IdHr |IdHsWd|_dSWd|_dSd|_w)z2Closing a cursor just exhausts all remaining data.N)rnextsetrconnr r r!closes z Cursor.closecCs|jstd|jS)Nz Cursor closed)rr r%r r r!_get_dbszCursor._get_dbcCs|jstddS)Nzexecute() first)rr r%r r r!_check_executedszCursor._check_executedcCs|SNr rrowr r r! _conv_rowzCursor._conv_rowcGdSz!Does nothing, required by DB API.Nr rargsr r r! setinputsizeszCursor.setinputsizescGr;r<r r=r r r!setoutputsizesr@zCursor.setoutputsizescs^|}|j}|dus||jurdS|jsdSd|_||IdH|IdHdS)zGet the next query setNT)r4rhas_next _clear_result next_result_do_get_result)rr2current_resultr r r!r0szCursor.nextsetcsNt|ttfrtfdd|DSt|tr"fdd|DS|S)Nc3s|]}|VqdSr6escape).0argr2r r! sz&Cursor._escape_args..csi|] \}}||qSr rG)rIkeyr-rKr r! sz'Cursor._escape_args..) isinstancetuplelistdictitemsrH)rr>r2r rKr! _escape_argss   zCursor._escape_argsNcCs$|}|dur||||}|S)a, Returns the exact string that is sent to the database by calling the execute() method. This method follows the extension to the DB API 2.0 followed by Psycopg. :param query: ``str`` sql statement :param args: ``tuple`` or ``list`` of arguments for sql query N)r4rTrqueryr>r2r r r!mogrifyszCursor.mogrifycsx|}|IdHr |IdHs |dur ||||}||IdH||_|jr9t|td||jS)aExecutes the given operation Executes the given operation substituting any markers with the given parameters. For example, getting all rows where id is 5: cursor.execute("SELECT * FROM t1 WHERE id = %s", (5,)) :param query: ``str`` sql statement :param args: ``tuple`` or ``list`` of arguments for sql query :returns: ``int``, number of rows that has been produced of affected N%r) r4r0rT_queryrrr inforrUr r r!executes   zCursor.executec s|sdS|jrtd|td|t|}|rO|dd}|d}|dp/d}|d d kr<|d d ks>J||||||j| j IdHSd }|D]}| ||IdH||j 7}qS||_ |j S) aExecute the given operation multiple times The executemany() method will execute the operation iterating over the list of parameters in seq_params. Example: Inserting 3 new employees and their phone number data = [ ('Jane','555-001'), ('Joe', '555-001'), ('John', '555-003') ] stmt = "INSERT INTO employees (name, phone) VALUES ('%s','%s')" await cursor.executemany(stmt, data) INSERT or REPLACE statements are optimized by batching the data, that is using the MySQL multiple rows syntax. :param query: `str`, sql statement :param args: ``tuple`` or ``list`` of arguments for sql query NCALL %srXr r r(r)) rr rZRE_INSERT_VALUESmatchgrouprstrip_do_execute_manymax_stmt_lengthr4encodingr[r) rrVr>mq_prefixq_values q_postfixrowsrJr r r! executemanys,      zCursor.executemanycs&|}|j}t|tr||}t|tr||}t|} t|}||t||} t| tr8| |d} | | 7} d} |D]?} ||| |} t| trT| |d} t| t| t|d|krw| | |IdH} | | 7} t|} n| d7} | | 7} q@| | |IdH} | | 7} | |_ | S)Nsurrogateescaperr ,) r4rTrOstrencode bytearrayiternextlenr[r)rprefixvaluespostfixr>rgrhr2rHsqlvrmrJrr r r!rf$s8           zCursor._do_execute_manyr cs|}|jrtdtd|t|D]\}}d|||f}||IdH|IdHqdfddt t |D}dd |d }||IdH||_ |S) a+Execute stored procedure procname with args Compatibility warning: PEP-249 specifies that any modified parameters must be returned. This is currently impossible as they are only available by storing them in a server variable and then retrieved by a query. Since stored procedures return zero or more result sets, there is no reliable way to get at OUT or INOUT parameters via callproc. The server variables are named @_procname_n, where procname is the parameter above and n is the position of the parameter (from zero). Once all result sets generated by the procedure have been fetched, you can issue a SELECT @_procname_0, ... query using .execute() to get any OUT or INOUT values. Compatibility warning: The act of calling a stored procedure itself creates an empty result set. This appears after any result sets generated by the procedure. This is non-standard behavior with respect to the DB-API. Be sure to use nextset() to advance through all result sets; otherwise you may get disconnected. :param procname: ``str``, name of procedure to execute on server :param args: `sequence of parameters to use with procedure :returns: the original args. r\rXzSET @_%s_%d=%sN,c3s|] }d|fVqdS)z@_%s_%dNr )rIiprocnamer r!rLgsz"Cursor.callproc..zCALL r`ra) r4rr rZ enumeraterHrYr0joinrangervr)rrr>r2indexrJq_argsr rr!callprocCs   zCursor.callproccCsl||j}|jdus|jt|jkr|d|S|j|j}|jd7_|j}|||S)zFetch the next row Nr )r5r create_futurerrrv set_resultrfutresultr r r!fetchonems     zCursor.fetchonecCsj||j}|jdur|g|S|j|p|j}|j|j|}t|t|j|_|||S)nReturns the next set of rows of a query result, returning a list of tuples. When no more rows are available, it returns an empty list. The number of rows returned can be specified using the size argument, which defaults to one :param size: ``int`` number of rows to return :returns: ``list`` of fetched rows N) r5rrrrrrminrv)rsizerendrr r r! fetchmany|s    zCursor.fetchmanycCsb||j}|jdur|g|S|jr!|j|jd}n|j}t|j|_|||S)z[Returns all rows of a query result set :returns: ``list`` of fetched rows N)r5rrrrrrvrr r r!fetchalls     zCursor.fetchallrelativecCs~||dkr|j|}n |dkr|}ntd|d|kr+t|jks0tdtd||_|j}|d|S)aScroll the cursor in the result set to a new position according to mode. If mode is relative (default), value is taken as offset to the current position in the result set, if set to absolute, value states an absolute target position. An IndexError should be raised in case a scroll operation would leave the result set. In this case, the cursor position is left undefined (ideal would be to not move the cursor at all). :param int value: move cursor to next position according to mode. :param str mode: scroll mode, possible modes: `relative` and `absolute` rabsoluteunknown scroll mode %srz out of rangeN) r5rr rvr IndexErrorrrr)rvaluemoder|rr r r!scrolls    z Cursor.scrollcs:|}||_|||IdH|IdHdSr6)r4_last_executedrCrVrErrr2r r r!rYs z Cursor._querycCs(d|_d|_d|_d|_d|_d|_dSNr)rrrrrrr%r r r!rCs  zCursor._clear_resultcs^|}d|_|j|_}|j|_|j|_|j|_|j |_ |j dkr-| |IdHdSdSr) r4rr affected_rowsrr&r insert_idrrmr warning_count_show_warnings)rr2rr r r!rEs  zCursor._do_get_resultcsV|jr |jjr dS|IdH}|durdS|D]}|d}tt|tdqdS)Nr)rrB show_warningswarningswarnrqr)rr2wswmsgr r r!rszCursor._show_warningscCs|Sr6r r%r r r! __aiter__r:zCursor.__aiter__cs |IdH}|dur|Str6)rStopAsyncIteration)rretr r r! __anext__s zCursor.__anext__cs|Sr6r r%r r r! __aenter__szCursor.__aenter__cs|IdHdSr6)r3)rexc_typeexc_valexc_tbr r r! __aexit__szCursor.__aexit__)Fr6)r r)3__name__ __module__ __qualname____doc__rgr"propertyrr&r'r(r,setterr.rr/r3r4r5r9r?rAr0rTrWr[rnrfrrrrrrYrCrErrrrrrrrr r r rrrrr r r r!rsr            . *      rcs0eZdZfddZddZfddZZS)_DeserializationCursorMixincs6tIdHjrfddjD_dSdS)Ncg|]}|qSr )_deserialization_rowrIr|r%r r! z>_DeserializationCursorMixin._do_get_result..)superrErr% __class__r%r!rE s z*_DeserializationCursorMixin._do_get_resultc Cs|durdSt|trd}nt|}d}t|jD]1\}^}}}|tjkrJ|r)|n|}tt t t ||||<Wdn1sEwYq|rO|St |S)NTF)rOrRrQrrrJSON contextlibsuppress ValueError TypeErrorjsonloadsrP)rr8 dict_flagrname field_typenpointr r r!rs    z0_DeserializationCursorMixin._deserialization_rowcs"|durdSt|}||Sr6)rr9rr7rr r!r9 s  z%_DeserializationCursorMixin._conv_row)rrrrErr9 __classcell__r r rr!rs rc@eZdZdZdS)DeserializationCursorz6A cursor automatic deserialization of json type fieldsNrrrrr r r r!r'rcs,eZdZeZfddZfddZZS)_DictCursorMixincstIdHg}jr+jjD]}|j}||vr"|jd|}||q|_|r=j r?fddj D_ dSdSdS)N.crr )r9rr%r r!r;rz3_DictCursorMixin._do_get_result..) rrErrfieldsr table_nameappend_fieldsr)rrfrrr%r!rE/s   z_DictCursorMixin._do_get_resultcs*|durdSt|}|t|j|Sr6)rr9 dict_typeziprr7rr r!r9=s z_DictCursorMixin._conv_row)rrrrRrrEr9rr r rr!r+s rc@r) DictCursorz.A cursor which returns results as a dictionaryNrr r r r!rDrrc@sLeZdZdZddZddZddZdd Zd d Zdd dZ dddZ d S)SSCursoraUnbuffered Cursor, mainly useful for queries that return a lot of data, or for connections to remote servers over a slow network. Instead of copying every row of data into a buffer, this will fetch rows as needed. The upside of this, is the client uses much less memory, and rows are returned much faster when traveling over a slow network, or if the result set is very big. There are limitations, though. The MySQL protocol doesn't support returning the total number of rows, so the only way to tell how many rows there are is to iterate over every row returned. Also, it currently isn't possible to scroll backwards, as only the current row is held in memory. csz|j}|dur dS|jdur|j|jur|jIdHz|IdHr3 |IdHs%Wd|_dSWd|_dSd|_wr6)rr_finish_unbuffered_queryr0r1r r r!r3Ws zSSCursor.closecs8|}||_|j|ddIdH|IdH|jS)NT) unbuffered)r4rrVrErrr r r!rYes zSSCursor._querycs |jIdH}||}|S)zRead next row N)r_read_rowdata_packet_unbufferedr9r7r r r! _read_nextls zSSCursor._read_nextcs6||IdH}|durdS|jd7_|S)z Fetch next row Nr )r5rrr7r r r!rrszSSCursor.fetchonecs0g} |IdH}|dur |S||q)z`Fetch all, as per MySQLdb. Pretty useless for large queries, as it is buffered. TN)rr)rrmr8r r r!r{s zSSCursor.fetchallNcsb||dur |j}g}t|D]}|IdH}|dur"|S|||jd7_q|S)rNr )r5rrrrr)rrrmr~r8r r r!rs   zSSCursor.fetchmanyrcs||dkr(|dkrtdt|D] }|IdHq|j|7_dS|dkrM||jkr5td||j}t|D] }|IdHq>||_dStd|)aScroll the cursor in the result set to a new position according to mode . Same as :meth:`Cursor.scroll`, but move cursor on server side one by one row. If you want to move 20 rows forward scroll will make 20 queries to move cursor. Currently only forward scrolling is supported. :param int value: move cursor to next position according to mode. :param str mode: scroll mode, possible modes: `relative` and `absolute` rrz0Backwards scrolling not supported by this cursorNrr)r5r rrrr )rrr_rr r r!rs$       zSSCursor.scrollr6r) rrrrr3rYrrrrrr r r r!rHs rc@r) SSDictCursorzs*0   qy