o `^h:@slddlZddlmZmZddlmZmZddlmZddZ Gdd d eZ Gd d d Z Gd d d Z dS)N)MappingSequence) expressionsqltypes)exccs"t||||}|IdH|SN) ResultProxy_prepare) connectioncursordialect result_map result_proxyrP/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/aiomysql/sa/result.pycreate_result_proxy src@sdeZdZdZddZddZddZdd Zd d Zd d Z dZ ddZ ddZ ddZ ddZdS)RowProxy _result_proxy_row _processors_keymapcCs||_||_||_||_dS)z8RowProxy objects are constructed by ResultProxy objects.Nr)selfrrow processorskeymaprrr__init__s zRowProxy.__init__cCs t|jjSr)iterrkeysrrrr__iter__ zRowProxy.__iter__cCs t|jSr)lenrr rrr__len__  zRowProxy.__len__cCspz |j|\}}}Wnty|j|\}}}Ynw|dur(td||dur3||j|S|j|S)NzVAmbiguous column name '%s' in result set! try 'use_labels' option on select statement.)rKeyErrorr _key_fallbackrInvalidRequestErrorr)rkey processorobjindexrrr __getitem__#s  zRowProxy.__getitem__c Cs2z||WSty}zt|jdd}~ww)Nr)r&AttributeErrorargs)rnameerrr __getattr__>s  zRowProxy.__getattr__cCs|j|j|Sr)r_has_keyr)rr)rrr __contains__DszRowProxy.__contains__NcCs4t|tr ||kSt|tr||kStSr) isinstanceras_tuplerNotImplementedrotherrrr__eq__Is   zRowProxy.__eq__cCs ||k Srrr8rrr__ne__Qr%zRowProxy.__ne__cstfddDS)Nc3s|]}|VqdSrr).0kr rr Usz$RowProxy.as_tuple..)tupler rr rr6TszRowProxy.as_tuplecCs t|Sr)reprr6r rrr__repr__Wr"zRowProxy.__repr__)__name__ __module__ __qualname__ __slots__rr!r$r-r2r4__hash__r:r;r6rArrrrrs rc@s*eZdZdZddZd ddZddZd S) ResultMetaDatazRHandle cursor.description, applying additional info from an execution context.c Cs$g|_}i}|jrdd|jD}i|_}g|_|j}t|di}|js*Jdi}|jr3Jdt|D]S\} } | d} | d} | d| | | | t j } }}| || }| |||| f} | || <|| | | uryd|df|| <} |j | |r|D]}| ||<qq7||dS)NcSsi|] }|d|dqS)rr)r<elemrrr esz+ResultMetaData.__init__..dbapi_type_mapz4Doesn't support case insensitive database connectionz#psycopg in py3k should not use thisrr)r _result_maprrr getattrcase_sensitivedescription_encoding enumerategetrNULLTYPE_cached_result_processorappend setdefaultupdate)rrmetadatarrrr typemapprimary_keymapireccolnamecoltyper0r+type_r*orrrr_sN         zResultMetaData.__init__TcCs|j}d}t|tr||}n=t|tjrM|jr$|j|vr$||j}nt|dr3|j|vr3||j}|durM|ddurM|dD] }| |rJnqAd}|dur_|r]t dt |dS|||<|S)Nr0rz.Could not locate column in row for column '%s') rr5strrQr ColumnElement_labelhasattrr0_compare_name_for_resultrNoSuchColumnError_string_or_unprintable)rr)raiseerrmapresultr+rrrr's4        zResultMetaData._key_fallbackcCs||jvrdS||dduS)NTF)rr')rrr)rrrr3s zResultMetaData._has_keyN)T)rBrCrD__doc__rr'r3rrrrrG[s  N %rGc@seZdZdZddZddZeddZedd Zd d Z ed d Z eddZ eddZ eddZ ddZddZddZddZddZd)dd Zd!d"Zd#d$Zd%d&Zd'd(ZdS)*r acWraps a DB-API cursor object to provide easier access to row columns. Individual columns may be accessed by their integer position, case-insensitive column name, or by sqlalchemy schema.Column object. e.g.: row = fetchone() col1 = row[0] # access via integer position col2 = row['col2'] # access via name col3 = row[mytable.c.mycol] # access via Column object. ResultProxy also handles post-processing of result column data using sqlalchemy TypeEngine objects, which are referenced from the originating SQL statement that produced this result set. cCs2||_d|_||_||_|j|_|j|_||_dS)NF) _dialect_closed_cursor _connectionrowcount _rowcount lastrowid _lastrowidrL)rr r r rrrrrs zResultProxy.__init__csh|jjj|jjdur%t|j|_fdd}t|||_ dSd|_| IdHd|_ dS)NcsdSr) create_taskclose)wrr looprrcallbacksz&ResultProxy._prepare..callback) rnr rwrm descriptionrG _metadataweakrefref_weakrt)rrxrrvrr s   zResultProxy._preparecC|jS)zSQLAlchemy dialect.)rkr rrrr szResultProxy.dialectcCr~r)rmr rrrr zResultProxy.cursorcCs|jr t|jjSdS)z/Return the current set of string keys for rows.r)rzr?rr rrrr s zResultProxy.keyscCr~)aEReturn the 'rowcount' for this result. The 'rowcount' reports the number of rows *matched* by the WHERE criterion of an UPDATE or DELETE statement. .. note:: Notes regarding .rowcount: * This attribute returns the number of rows *matched*, which is not necessarily the same as the number of rows that were actually *modified* - an UPDATE statement, for example, may have no net change on a given row if the SET values given are the same as those present in the row already. Such a row would be matched but not modified. * .rowcount is *only* useful in conjunction with an UPDATE or DELETE statement. Contrary to what the Python DBAPI says, it does *not* return the number of rows available from the results of a SELECT statement as DBAPIs cannot support this functionality when rows are unbuffered. * Statements that use RETURNING may not return a correct rowcount. )rpr rrrroszResultProxy.rowcountcCr~)zReturns the 'lastrowid' accessor on the DBAPI cursor. This is a DBAPI specific method and is only functional for those backends which support it, for statements where it is appropriate. )rrr rrrrq3szResultProxy.lastrowidcCs |jduS)zTrue if this ResultProxy returns rows. I.e. if it is legal to call the methods .fetchone(), .fetchmany() and .fetchall()`. N)rzr rrr returns_rows=s zResultProxy.returns_rowscCr~r)rlr rrrclosedFrzResultProxy.closedcs2|jsd|_|jIdHd|_d|_dSdS)aeClose this ResultProxy. Closes the underlying DBAPI cursor corresponding to the execution. Note that any data cached within this ResultProxy is still available. For some types of results, this may include buffered rows. If this ResultProxy was generated from an implicit execution, the underlying Connection will also be closed (returns the underlying DBAPI connection to the connection pool.) This method is called automatically when: * all result rows are exhausted using the fetchXXX() methods. * cursor.description is None. TN)rlrmrtr}r rrrrtJs zResultProxy.closecCs|jdur tdtd)NzJThis result object does not return rows. It has been closed automatically.zThis result object is closed.)rzrResourceClosedErrorr rrr _non_resultks  zResultProxy._non_resultcs.t|jjjfdd|DS)Ncsg|] }|qSrr)r<rrrW process_rowrrr xsz-ResultProxy._process_rows..)rrzrr)rrowsrrr _process_rowssszResultProxy._process_rowscsPz |jIdH}Wnty|YdSw||}|IdH|S)z3Fetch all rows, just like DB-API cursor.fetchall().N)rmfetchallr.rrrt)rrretrrrr{s  zResultProxy.fetchallcs^z |jIdH}Wnty|YdSw|dur&||gdS|IdHdS)zFetch one row, just like DB-API cursor.fetchone(). If a row is present, the cursor remains open after this is called. Else the cursor is automatically closed and None is returned. Nr)rmfetchoner.rrrtrrrrrrs zResultProxy.fetchoneNcsxz|dur|jIdH}n |j|IdH}Wnty'|YdSw||}t|dkr:|IdH|S)zFetch many rows, just like DB-API cursor.fetchmany(size=cursor.arraysize). If rows are present, the cursor remains open after this is called. Else the cursor is automatically closed and an empty list is returned. Nr)rm fetchmanyr.rrr#rt)rsizerrrrrrs   zResultProxy.fetchmanycsD|jdur |z|IdHW|IdHS|IdHw)zwFetch the first row and then close the result set unconditionally. Returns None if no row is present. N)rzrrrtr rrrfirsts   zResultProxy.firstcs$|IdH}|dur|dSdS)zwFetch the first column of the first row, and close the result set. Returns None if no row is present. Nr)rrrrrscalars zResultProxy.scalarcCs|Srrr rrr __aiter__szResultProxy.__aiter__cs |IdH}|dur|Str)rStopAsyncIteration)rdatarrr __anext__s zResultProxy.__anext__r)rBrCrDrjrr propertyr r rrorqrrrtrrrrrrrrrrrrrr s6       !   r ) r{collections.abcrrsqlalchemy.sqlrrrrrrGr rrrrs I~