관리-도구
편집 파일: result.cpython-37.pyc
B ��4]�� � @ s` d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ yddl m Z d d � Z W n e k r� d d � Z Y nX ydd l mZ d ZW n( e k r� dZG dd� de�ZY nX G dd� de�Zyejj�e� W n e k r� Y nX G dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�ZdS ) zSDefine result set constructs including :class:`.ResultProxy` and :class:`.RowProxy.� N� )�exc)�util)� expression)�sqltypes)�safe_rowproxy_reconstructorc C s t | |�S )N)r )�cls�state� r �K/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/engine/result.py�rowproxy_reconstructor! s r c C s | � | �}|�|� |S )N)�__new__� __setstate__)r r �objr r r r ' s )� BaseRowProxyTFc @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )r )�_parent�_row� _processors�_keymapc C s || _ || _|| _|| _dS )z8RowProxy objects are constructed by ResultProxy objects.N)r r r r )�self�parent�row� processors�keymapr r r �__init__7 s zBaseRowProxy.__init__c C s t | j| �� ffS )N)r � __class__� __getstate__)r r r r � __reduce__? s zBaseRowProxy.__reduce__c C s t | �S )z9Return the values represented by this RowProxy as a list.)�list)r r r r �valuesE s zBaseRowProxy.valuesc c s: x4t | j| j�D ]"\}}|d kr(|V q||�V qW d S )N)�zipr r )r � processor�valuer r r �__iter__I s zBaseRowProxy.__iter__c C s t | j�S )N)�lenr )r r r r �__len__P s zBaseRowProxy.__len__c C s� y| j | \}}}W n� tk r: | j�|�\}}}Y nr tk r� t|t�r�g }xDt| j| | j | �D ]*\}}|d kr�|� |� qn|� ||�� qnW t |�S � Y nX |d kr�t � d| ��|d k r�|| j | �S | j | S d S )Nz<Ambiguous column name '%s' in result set column descriptions)r �KeyErrorr � _key_fallback� TypeError� isinstance�slicer r r �append�tupler �InvalidRequestError)r �keyr! r �index�lr" r r r � __getitem__S s* zBaseRowProxy.__getitem__c C s> y| | S t k r8 } zt|jd ��W d d }~X Y nX d S )Nr )r& �AttributeError�args)r �name�er r r � __getattr__o s zBaseRowProxy.__getattr__N) �__name__� __module__� __qualname__� __slots__r r r r# r% r1 r6 r r r r r 4 s r c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd S )$�RowProxyax Proxy values from a single cursor row. Mostly follows "ordered dictionary" behavior, mapping result values to the string-based column name, the integer position of the result in the row, as well as Column instances which can be mapped to the original Columns that produced this result set (for results that correspond to constructed SQL expressions). r c C s | j �|�S )N)r �_has_key)r r. r r r � __contains__� s zRowProxy.__contains__c C s | j t| �d�S )N)r r )r r, )r r r r r � s zRowProxy.__getstate__c C s, |d | _ }|d | _|j| _|j| _d S )Nr r )r r r r )r r r r r r r � s zRowProxy.__setstate__Nc C s* t |t�r |t| �t|��S |t| �|�S )N)r) r; r, )r �other�opr r r �_op� s z RowProxy._opc C s | � |tj�S )N)r@ �operator�lt)r r> r r r �__lt__� s zRowProxy.__lt__c C s | � |tj�S )N)r@ rA �le)r r> r r r �__le__� s zRowProxy.__le__c C s | � |tj�S )N)r@ rA �ge)r r> r r r �__ge__� s zRowProxy.__ge__c C s | � |tj�S )N)r@ rA �gt)r r> r r r �__gt__� s zRowProxy.__gt__c C s | � |tj�S )N)r@ rA �eq)r r> r r r �__eq__� s zRowProxy.__eq__c C s | � |tj�S )N)r@ rA �ne)r r> r r r �__ne__� s zRowProxy.__ne__c C s t t�| ��S )N)�repr�sql_util� _repr_row)r r r r �__repr__� s zRowProxy.__repr__c C s | j �|�S )z4Return True if this RowProxy contains the given key.)r r< )r r. r r r �has_key� s zRowProxy.has_keyc s � fdd�� � � D �S )z@Return a list of tuples, each tuple containing a key/value pair.c s g | ]}|� | f�qS r r )�.0r. )r r r � <listcomp>� s z"RowProxy.items.<locals>.<listcomp>)�keys)r r )r r �items� s zRowProxy.itemsc C s | j jS )z@Return the list of keys as strings represented by this RowProxy.)r rU )r r r r rU � s z RowProxy.keysc C s t | jj�S )N)�iterr rU )r r r r �iterkeys� s zRowProxy.iterkeysc C s t | �S )N)rW )r r r r � itervalues� s zRowProxy.itervalues)r7 r8 r9 �__doc__r: r= r r �__hash__r@ rC rE rG rI rK rM rQ rR rV rU rX rY r r r r r; v s& r; c @ s~ e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z e d dd��Z d dd�Z dd� Zd dd�Zdd� Zdd� Zd S ) �ResultMetaDatazRHandle cursor.description, applying additional info from an execution context.)r �case_sensitive�matched_on_namer rU �_orig_processorsc s� |j }|j}|j�_d�_d �_|jr<|j\}}}t|�}nd } } }}��||||||�} i �_t s�t| ���j� dd� | D ��fdd�| D � � dd� | D ��_ t dd� | D ��� |�rrt� �|k�r\t � } xJ| D ]B} | d } | | k�r�j�s| �� n| } d | d f� | < | �| � q�W �j� � fdd�| D �� �j�sr�j� d d� | D �� n�j� d d� | D �� �j� � � |�s�|j�r��j� �fd d�| D �� d S ) NFc S s* g | ]"}|d |d |d |d ff�qS )r � � r )rS �elemr r r rT � s z+ResultMetaData.__init__.<locals>.<listcomp>c s. g | ]&}|d � |d |d |d ff�qS )r r` ra r )rS rb )�len_rawr r rT s c S s g | ] }|d �qS )r` r )rS rb r r r rT s c S s* g | ]"}|d |d |d |d ff�qS )r r` ra r r )rS rb r r r rT s � c s2 g | ]*}|d r|d D ]}|� |d f�qqS )ra r r )rS rb �obj_elem)�by_keyr r rT % s c S s6 g | ].}|d r|d d |d |d |d ff�qS )ra r r` r )rS rb r r r rT 3 s c S s< g | ]4}|d r|d D ] }||d |d |d ff�qqS )ra r` r r )rS rb re r r r rT = s c s* g | ]"}|d r|d � j |d f�qS )� r )r )rS rb )r r r rT K s )�context�dialectr] r^ r_ Zresult_column_structr$ �_merge_cursor_descriptionr �_baserowproxy_usecext�updater �dict�set�lower�add�_translate_colname) r r �cursor_descriptionrh ri �result_columns�cols_are_ordered�textual_ordered� num_ctx_cols�raw�seen�recr. r )rf rc r r r � s` zResultMetaData.__init__c s� �j j� |rJ|rJ|sJ|t��krJdd� |D �| _� ��fdd�t|�D �S |r^| ���|�}n |rr| ���|�}n | ����}�fdd�|D �S dS )a� Merge a cursor.description with compiled result column information. There are at least four separate strategies used here, selected depending on the type of SQL construct used to start with. The most common case is that of the compiled SQL expression construct, which generated the column names present in the raw SQL string and which has the identical number of columns as were reported by cursor.description. In this case, we assume a 1-1 positional mapping between the entries in cursor.description and the compiled object. This is also the most performant case as we disregard extracting / decoding the column names present in cursor.description since we already have the desired name we generated in the compiled SQL construct. The next common case is that of the completely raw string SQL, such as passed to connection.execute(). In this case we have no compiled construct to work with, so we extract and decode the names from cursor.description and index those as the primary result row target keys. The remaining fairly common case is that of the textual SQL that includes at least partial column information; this is when we use a :class:`.TextAsFrom` construct. This construct may have unordered or ordered column information. In the ordered case, we merge the cursor.description and the compiled construct's information positionally, and warn if there are additional description names present, however we still decode the names in cursor.description as we don't have a guarantee that the names in the columns match on these. In the unordered case, we match names in cursor.description to that of the compiled construct based on name matching. In both of these cases, the cursor.description names and the column expression objects and names are indexed as result row target keys. The final case is much less common, where we have a compiled non-textual SQL expression construct, but the number of columns in cursor.description doesn't match what's in the compiled construct. We make the guess here that there might be textual column expressions in the compiled construct that themselves include a comma in them causing them to split. We do the same name-matching as with textual non-ordered columns. The name-matched system of merging is the same as that used by SQLAlchemy for all cases up through te 0.9 series. Positional matching for compiled SQL expressions was introduced in 1.0 as a major performance feature, and positional matching for textual :class:`.TextAsFrom` objects in 1.1. As name matching is no longer a common case, it was acceptable to factor it into smaller generator- oriented methods that are easier to understand, but incur slightly more performance overhead. c S s g | ] }|d �qS )r r )rS rb r r r rT � s z<ResultMetaData._merge_cursor_description.<locals>.<listcomp>c sF g | ]>\}\}}}}||� s$|� � n|��||�| d �|df�qS )rd N)ro �get_result_processor)rS �idxr. r4 r �type_)r] rh rr r r rT � s c s2 g | ]*\}}}}}}|||� � |||�||f�qS r )rz )rS r{ �colname� mapped_type�coltyper � untranslated)rh r r rT � s N)ri r] r$ rU � enumerate� _merge_textual_cols_by_position�_merge_cols_by_name�_merge_cols_by_none)r rh rr rs rv rt ru Z raw_iteratorr )r] rh rr r rj N s$ > z(ResultMetaData._merge_cursor_descriptionc c s� |j }|j}|j}|jr |jnd}|jr.|jnd}d}g | _xrt|�D ]f\} } | d } | d } |rj|| �} |rz|| �\} }|r�|| �} | j� | � |s�| � � } | | || fV qFW dS )z�Extract column names and data types from a cursor.description. Applies unicode decoding, column translation, "normalization", and case sensitivity rules to the names based on the dialect. Nr rd ) ri r] rq Zdescription_encodingZ_description_decoderZrequires_name_normalize�normalize_namerU r� r+ ro ) r rh rr ri r] Ztranslate_colnameZdescription_decoderr� r� r{ ry r} r r r r �_colnames_from_description� s* z)ResultMetaData._colnames_from_descriptionc c s� |r t |�nd }|t |�kr2t�d|t |�f � t� }x�| �||�D ]v\}}}} ||k r�|| } | d } | d } | d |kr�t�d| d ��|�| d � n tj } d } ||| | | |fV qFW d S )NzVNumber of columns in textual SQL (%d) is smaller than number of columns requested (%d)r r` r z8Duplicate column expression requested in textual SQL: %r) r$ r �warnrn r� r r- rp r �NULLTYPE) r rh rr rs rv rx r{ r} r� r �ctx_recr r~ r r r r� � s( z.ResultMetaData._merge_textual_cols_by_positionc c s� |j }|j}| �||�}d| _xj| �||�D ]Z\}}} } y || } W n tk rb tj} d } Y nX | d } | d } ||| | | | fV q,W d S )NTrd r )ri r] �_create_result_mapr^ r� r& r r� )r rh rr rs ri r] Z result_mapr{ r} r� r r� r~ r r r r r� s z"ResultMetaData._merge_cols_by_namec c s6 x0| � ||�D ] \}}}}||tj|d |fV qW d S )N)r� r r� )r rh rr r{ r} r� r r r r r� / s z"ResultMetaData._merge_cols_by_noneTc C sn i }xd|D ]\}|d |dd � }}|s0|� � }||kr^|| \}}} |||d | f||< q |||< q W |S )Nr rd )ro ) r rs r] �drb r. ry Ze_nameZe_objZe_typer r r r� 8 s z!ResultMetaData._create_result_mapc C s$ | j }d }t|tj�r0|�| jr$|n|�� �}n�t|tj�r�|j rx| jrN|j n|j �� |krx|| jrj|j n|j �� }n>t |d�r�| jr�|j n|j �� |kr�|| jr�|j n|j �� }|d k r�|d d k r�x |d D ]}|� |�r�P q�W d }|d k�r|�rt �dt�|� ��nd S n|||< |S )Nr4 rd z.Could not locate column in row for column '%s')r r) r � string_types�getr] ro r Z ColumnElementZ_label�hasattrr4 Z_compare_name_for_resultr ZNoSuchColumnErrorZ_string_or_unprintable)r r. �raiseerrZmap_�resultr r r r r' J s8 z ResultMetaData._key_fallbackc C s" || j krdS | �|d�d k S d S )NTF)r r' )r r. r r r r< { s zResultMetaData._has_keyc C s^ || j kr | j | \}}}n"| �||�}|d kr4d S |\}}}|d krTt�d| ��t�|�S )Nz<Ambiguous column name '%s' in result set column descriptions)r r' r r- rA � itemgetter)r r. r� r! r r/ �retr r r �_getter� s zResultMetaData._getterc C s( t dd� | j�� D ��| j| j| jd�S )Nc s s2 | ]*\}\}}}t |tjtj �r||fV qd S )N)r) r r� Z int_types)rS r. r! r r/ r r r � <genexpr>� s z.ResultMetaData.__getstate__.<locals>.<genexpr>)�_pickled_keymaprU r] r^ )rm r rV rU r] r^ )r r r r r � s zResultMetaData.__getstate__c C sp dd� t t|d ��D �| _i | _}x&|d �� D ]\}}d d |f||< q4W |d | _|d | _|d | _d S )Nc S s g | ]}d �qS )Nr )rS �_r r r rT � s z/ResultMetaData.__setstate__.<locals>.<listcomp>rU r� r] r^ )�ranger$ r r rV rU r] r^ )r r r r. r/ r r r r � s zResultMetaData.__setstate__N)T)T)T)r7 r8 r9 rZ r: r rj r� r� r� r� � classmethodr� r' r< r� r r r r r r r\ � s w*" 1 r\ c @ sV e Zd ZdZeZdZdZdZdZ dZ dd� Z dGdd�Z d d � Z d d � Zd d� Zejdd� �Zedd� �Zedd� �Zedd� �Zdd� Zdd� Zdd � Zd d � Zd d � ZeZ ejd!d"� �Z d#d$� Z d%d&� Z ed'd(� �Z d)d*� Z!d+d,� Z"d-d.� Z#d/d0� Z$d1d2� Z%d3d4� Z&dHd5d6�Z'd7d8� Z(d9d:� Z)d;d<� Z*d=d>� Z+dId?d@�Z,dAdB� Z-dCdD� Z.dEdF� Z/dS )J� ResultProxya[ Wraps 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 ``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 ``TypeEngine`` objects, which are referenced from the originating SQL statement that produced this result set. NFc C sD || _ |j| _|j | _| _|j| _| jjo4|j�� | _| � � d S )N) rh ri �cursor� _saved_cursorZroot_connection� connection�_echo�engineZ_should_log_debug�_init_metadata)r rh r r r r � s zResultProxy.__init__Tc C s6 y | j j}W n tk r& | �d �S X |||�S d S )N)� _metadatar� r2 � _non_result)r r. r� �getterr r r r� � s zResultProxy._getterc C s4 y | j j}W n tk r& | �d �S X ||�S d S )N)r� r< r2 r� )r r. rR r r r r< � s zResultProxy._has_keyc C s� | � � }|d k r�| jjrTd| jjkrT| jjjr<| jjj| _q`t| |� | _| jj_n t| |�| _| jr�| jjj � dt dd� |D ��� d S )NZcompiled_cachezCol %rc s s | ]}|d V qdS )r Nr )rS �xr r r r� � s z-ResultProxy._init_metadata.<locals>.<genexpr>) �_cursor_descriptionrh �compiled�execution_optionsZ_cached_metadatar� r\ r� r� �logger�debugr, )r rr r r r r� � s zResultProxy._init_metadatac C s | j r| j jS g S dS )z/Return the current set of string keys for rows.N)r� rU )r r r r rU � s zResultProxy.keysc C sH y| j jS tk rB } z | j�|dd| j| j � W dd}~X Y nX dS )a� Return 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 :attr:`.ResultProxy.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. On backends that feature both styles, such as MySQL, rowcount is configured by default to return the match count in all cases. * :attr:`.ResultProxy.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. * :attr:`.ResultProxy.rowcount` may not be fully implemented by all dialects. In particular, most DBAPIs do not support an aggregate rowcount result from an executemany call. The :meth:`.ResultProxy.supports_sane_rowcount` and :meth:`.ResultProxy.supports_sane_multi_rowcount` methods will report from the dialect if each usage is known to be supported. * Statements that use RETURNING may not return a correct rowcount. N)rh �rowcount� BaseExceptionr� �_handle_dbapi_exceptionr� )r r5 r r r r� � s )zResultProxy.rowcountc C sH y| j jS tk rB } z | j�|dd| j | j� W dd}~X Y nX dS )a/ return 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. It's behavior is not consistent across backends. Usage of this method is normally unnecessary when using insert() expression constructs; the :attr:`~ResultProxy.inserted_primary_key` attribute provides a tuple of primary key values for a newly inserted row, regardless of database backend. N)r� � lastrowidr� r� r� rh )r r5 r r r r� - s zResultProxy.lastrowidc C s | j dk S )z�True if this :class:`.ResultProxy` returns rows. I.e. if it is legal to call the methods :meth:`~.ResultProxy.fetchone`, :meth:`~.ResultProxy.fetchmany` :meth:`~.ResultProxy.fetchall`. N)r� )r r r r � returns_rowsD s zResultProxy.returns_rowsc C s | j jS )ad True if this :class:`.ResultProxy` is the result of a executing an expression language compiled :func:`.expression.insert` construct. When True, this implies that the :attr:`inserted_primary_key` attribute is accessible, assuming the statement did not include a user defined "returning" construct. )rh �isinsert)r r r r � is_insertP s zResultProxy.is_insertc C s | j jS )z May be overridden by subclasses.)r� � description)r r r r r� ^ s z ResultProxy._cursor_descriptionc C s< | j r dS d| _ | j}| j�|� | jr2| j�� d| _dS )al Soft close this :class:`.ResultProxy`. This releases all DBAPI cursor resources, but leaves the ResultProxy "open" from a semantic perspective, meaning the fetchXXX() methods will continue to return empty results. This method is called automatically when: * all result rows are exhausted using the fetchXXX() methods. * cursor.description is None. This method is **not public**, but is documented in order to clarify the "autoclose" process used. .. versionadded:: 1.0.0 .. seealso:: :meth:`.ResultProxy.close` NT)� _soft_closedr� r� Z_safe_close_cursor�_autoclose_connection�close)r r� r r r � _soft_closec s zResultProxy._soft_closec C s | j s| �� d| _ dS )a Close this ResultProxy. This closes out the underlying DBAPI cursor corresponding to the statement execution, if one is still present. Note that the DBAPI cursor is automatically released when the :class:`.ResultProxy` exhausts all available rows. :meth:`.ResultProxy.close` is generally an optional method except in the case when discarding a :class:`.ResultProxy` that still has additional rows pending for fetch. In the case of a result that is the product of :ref:`connectionless execution <dbengine_implicit>`, the underlying :class:`.Connection` object is also closed, which :term:`releases` DBAPI connection resources. After this method is called, it is no longer valid to call upon the fetch methods, which will raise a :class:`.ResourceClosedError` on subsequent use. .. versionchanged:: 1.0.0 - the :meth:`.ResultProxy.close` method has been separated out from the process that releases the underlying DBAPI cursor resource. The "auto close" feature of the :class:`.Connection` now performs a so-called "soft close", which releases the underlying DBAPI cursor, but allows the :class:`.ResultProxy` to still behave as an open-but-exhausted result set; the actual :meth:`.ResultProxy.close` method is never called. It is still safe to discard a :class:`.ResultProxy` that has been fully exhausted without calling this method. .. seealso:: :ref:`connections_toplevel` :meth:`.ResultProxy._soft_close` TN)�closedr� )r r r r r� � s %zResultProxy.closec c s$ x | � � }|dkrdS |V qW dS )z Implement iteration protocol.N)�fetchone)r r r r r r# � s zResultProxy.__iter__c C s | � � }|dkrt� �n|S dS )zGImplement the next() protocol. .. versionadded:: 1.2 N)r� � StopIteration)r r r r r �__next__� s zResultProxy.__next__c C sB | j jst�d��n&| j js(t�d��n| j jr:t�d��| j jS )a� Return the primary key for the row just inserted. The return value is a list of scalar values corresponding to the list of primary key columns in the target table. This only applies to single row :func:`.insert` constructs which did not explicitly specify :meth:`.Insert.returning`. Note that primary key columns which specify a server_default clause, or otherwise do not qualify as "autoincrement" columns (see the notes at :class:`.Column`), and were generated using the database-side default, will appear in this list as ``None`` unless the backend supports "returning" and the insert statement executed with the "implicit returning" enabled. Raises :class:`~sqlalchemy.exc.InvalidRequestError` if the executed statement is not a compiled expression construct or is not an insert() construct. z1Statement is not a compiled expression construct.z2Statement is not an insert() expression construct.z9Can't call inserted_primary_key when returning() is used.)rh r� r r- r� Z_is_explicit_returning�inserted_primary_key)r r r r r� � s z ResultProxy.inserted_primary_keyc C sH | j jst�d��n0| j js(t�d��n | j jr8| j jS | j jd S dS )a Return the collection of updated parameters from this execution. Raises :class:`~sqlalchemy.exc.InvalidRequestError` if the executed statement is not a compiled expression construct or is not an update() construct. z1Statement is not a compiled expression construct.z2Statement is not an update() expression construct.r N)rh r� r r- �isupdate� executemany�compiled_parameters)r r r r �last_updated_params� s z ResultProxy.last_updated_paramsc C sH | j jst�d��n0| j js(t�d��n | j jr8| j jS | j jd S dS )a Return the collection of inserted parameters from this execution. Raises :class:`~sqlalchemy.exc.InvalidRequestError` if the executed statement is not a compiled expression construct or is not an insert() construct. z1Statement is not a compiled expression construct.z2Statement is not an insert() expression construct.r N)rh r� r r- r� r� r� )r r r r �last_inserted_params s z ResultProxy.last_inserted_paramsc C s | j jS )a� Return the values of default columns that were fetched using the :meth:`.ValuesBase.return_defaults` feature. The value is an instance of :class:`.RowProxy`, or ``None`` if :meth:`.ValuesBase.return_defaults` was not used or if the backend does not support RETURNING. .. versionadded:: 0.9.0 .. seealso:: :meth:`.ValuesBase.return_defaults` )rh �returned_defaults)r r r r r� s z ResultProxy.returned_defaultsc C s | j �� S )z�Return ``lastrow_has_defaults()`` from the underlying :class:`.ExecutionContext`. See :class:`.ExecutionContext` for details. )rh �lastrow_has_defaults)r r r r r� . s z ResultProxy.lastrow_has_defaultsc C s6 | j jst�d��n| j js.| j js.t�d��| j jS )aM Return ``postfetch_cols()`` from the underlying :class:`.ExecutionContext`. See :class:`.ExecutionContext` for details. Raises :class:`~sqlalchemy.exc.InvalidRequestError` if the executed statement is not a compiled expression construct or is not an insert() or update() construct. z1Statement is not a compiled expression construct.z>Statement is not an insert() or update() expression construct.)rh r� r r- r� r� �postfetch_cols)r r r r r� 8 s zResultProxy.postfetch_colsc C s6 | j jst�d��n| j js.| j js.t�d��| j jS )aL Return ``prefetch_cols()`` from the underlying :class:`.ExecutionContext`. See :class:`.ExecutionContext` for details. Raises :class:`~sqlalchemy.exc.InvalidRequestError` if the executed statement is not a compiled expression construct or is not an insert() or update() construct. z1Statement is not a compiled expression construct.z>Statement is not an insert() or update() expression construct.)rh r� r r- r� r� � prefetch_cols)r r r r r� O s zResultProxy.prefetch_colsc C s | j jS )zxReturn ``supports_sane_rowcount`` from the dialect. See :attr:`.ResultProxy.rowcount` for background. )ri �supports_sane_rowcount)r r r r r� f s z"ResultProxy.supports_sane_rowcountc C s | j jS )z~Return ``supports_sane_multi_rowcount`` from the dialect. See :attr:`.ResultProxy.rowcount` for background. )ri � supports_sane_multi_rowcount)r r r r r� o s z(ResultProxy.supports_sane_multi_rowcountc C s* y | j �� S tk r$ | �d �S X d S )N)r� r� r2 r� )r r r r �_fetchone_implx s zResultProxy._fetchone_implc C sB y"|d kr| j �� S | j �|�S W n tk r< | �g �S X d S )N)r� � fetchmanyr2 r� )r �sizer r r �_fetchmany_impl~ s zResultProxy._fetchmany_implc C s* y | j �� S tk r$ | �g �S X d S )N)r� �fetchallr2 r� )r r r r �_fetchall_impl� s zResultProxy._fetchall_implc C s0 | j d krt�d��n| jr(t�d��n|S d S )NzJThis result object does not return rows. It has been closed automatically.z This result object is closed.)r� r ZResourceClosedErrorr� )r �defaultr r r r� � s zResultProxy._non_resultc s� | j �| j��j� �j�| jrd| jjjj}g }x0|D ](}|dt � |�� |� ��|�� �� q4W |S � ���fdd�|D �S d S )NzRow %rc s g | ]}��|�� ��qS r r )rS r )r �metadata� process_rowr r r rT � s z,ResultProxy.process_rows.<locals>.<listcomp>) � _process_rowr� r r r� rh r� r� r� rO rP r+ )r Zrows�logr0 r r )r r� r� r r � process_rows� s zResultProxy.process_rowsc C sZ y| � | �� �}| �� |S tk rT } z | j�|dd| j| j� W dd}~X Y nX dS )av Fetch all rows, just like DB-API ``cursor.fetchall()``. After all rows have been exhausted, the underlying DBAPI cursor resource is released, and the object may be safely discarded. Subsequent calls to :meth:`.ResultProxy.fetchall` will return an empty list. After the :meth:`.ResultProxy.close` method is called, the method will raise :class:`.ResourceClosedError`. .. versionchanged:: 1.0.0 - Added "soft close" behavior which allows the result to be used in an "exhausted" state prior to calling the :meth:`.ResultProxy.close` method. N)r� r� r� r� r� r� r� rh )r r0 r5 r r r r� � s zResultProxy.fetchallc C sh y(| � | �|��}t|�dkr&| �� |S tk rb } z | j�|dd| j| j� W dd}~X Y nX dS )a� Fetch many rows, just like DB-API ``cursor.fetchmany(size=cursor.arraysize)``. After all rows have been exhausted, the underlying DBAPI cursor resource is released, and the object may be safely discarded. Calls to :meth:`.ResultProxy.fetchmany` after all rows have been exhausted will return an empty list. After the :meth:`.ResultProxy.close` method is called, the method will raise :class:`.ResourceClosedError`. .. versionchanged:: 1.0.0 - Added "soft close" behavior which allows the result to be used in an "exhausted" state prior to calling the :meth:`.ResultProxy.close` method. r N) r� r� r$ r� r� r� r� r� rh )r r� r0 r5 r r r r� � s zResultProxy.fetchmanyc C sp y0| � � }|dk r"| �|g�d S | �� dS W n: tk rj } z | j�|dd| j| j� W dd}~X Y nX dS )a� Fetch one row, just like DB-API ``cursor.fetchone()``. After all rows have been exhausted, the underlying DBAPI cursor resource is released, and the object may be safely discarded. Calls to :meth:`.ResultProxy.fetchone` after all rows have been exhausted will return ``None``. After the :meth:`.ResultProxy.close` method is called, the method will raise :class:`.ResourceClosedError`. .. versionchanged:: 1.0.0 - Added "soft close" behavior which allows the result to be used in an "exhausted" state prior to calling the :meth:`.ResultProxy.close` method. Nr )r� r� r� r� r� r� r� rh )r r r5 r r r r� � s zResultProxy.fetchonec C s� | j dkr| �d�S y | �� }W n: tk rZ } z | j�|dd| j| j� W dd}~X Y nX z |dk rv| �|g�d S dS W d| � � X dS )a Fetch the first row and then close the result set unconditionally. Returns None if no row is present. After calling this method, the object is fully closed, e.g. the :meth:`.ResultProxy.close` method will have been called. Nr ) r� r� r� r� r� r� r� rh r� r� )r r r5 r r r �first� s $zResultProxy.firstc C s | � � }|dk r|d S dS dS )a Fetch the first column of the first row, and close the result set. Returns None if no row is present. After calling this method, the object is fully closed, e.g. the :meth:`.ResultProxy.close` method will have been called. Nr )r� )r r r r r �scalar s zResultProxy.scalar)T)N)N)0r7 r8 r9 rZ r; r� Zout_parametersr� r� r� r� r r� r< r� rU r Zmemoized_propertyr� �propertyr� r� r� r� r� r� r# r� �nextr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � sR 0 ) , r� c sd e Zd ZdZ� fdd�Zdddddd d d d �Zd d� Z� fdd�Zdd� Zddd�Z dd� Z � Z S )�BufferedRowResultProxya� A ResultProxy with row buffering behavior. ``ResultProxy`` that buffers the contents of a selection of rows before ``fetchone()`` is called. This is to allow the results of ``cursor.description`` to be available immediately, when interfacing with a DB-API that requires rows to be consumed before this information is available (currently psycopg2, when used with server-side cursors). The pre-fetching behavior fetches only one row initially, and then grows its buffer size by a fixed amount with each successive need for additional rows up to a size of 1000. The size argument is configurable using the ``max_row_buffer`` execution option:: with psycopg2_engine.connect() as conn: result = conn.execution_options( stream_results=True, max_row_buffer=50 ).execute("select * from table") .. versionadded:: 1.0.6 Added the ``max_row_buffer`` option. .. seealso:: :ref:`psycopg2_execution_options` c s, | j j�dd �| _| �� tt| ��� d S )NZmax_row_buffer)rh r� r� �_max_row_buffer�$_BufferedRowResultProxy__buffer_rows�superr� r� )r )r r r r� F s z%BufferedRowResultProxy._init_metadatarg � � �2 �d � i� i� )rd rg r� r� r� r� r� i� c C s\ | j d krd S t| dd�}t�| j �|��| _| j�||�| _| j d k rXt | j | j�| _d S )N�_bufsizerd ) r� �getattr� collections�dequer� �"_BufferedRowResultProxy__rowbuffer� size_growthr� r� r� �min)r r� r r r Z __buffer_rows\ s z$BufferedRowResultProxy.__buffer_rowsc s | j �� tt| �jf |� d S )N)r� �clearr� r� r� )r �kw)r r r r� e s z"BufferedRowResultProxy._soft_closec C s6 | j d kr| �d �S | js,| �� | js,d S | j�� S )N)r� r� r� r� �popleft)r r r r r� i s z%BufferedRowResultProxy._fetchone_implNc C sH |d kr| � � S g }x.td|�D ] }| �� }|d kr6P |�|� q W |S )Nr )r� r� r� r+ )r r� r� r� r r r r r� r s z&BufferedRowResultProxy._fetchmany_implc C s: | j d kr| �g �S | j�| j �� � | j}t�� | _|S )N)r� r� r� �extendr� r� r� )r r� r r r r� } s z%BufferedRowResultProxy._fetchall_impl)N) r7 r8 r9 rZ r� r� r� r� r� r� r� � __classcell__r r )r r r� ( s r� c sN e Zd ZdZ� fdd�Zdd� Z� fdd�Zdd � Zdd d �Zd d� Z � Z S )�FullyBufferedResultProxyz�A result proxy that buffers rows fully upon creation. Used for operations where a result is to be delivered after the database conversation can not be continued, such as MSSQL INSERT...OUTPUT after an autocommit. c s t t| ��� | �� | _d S )N)r� r� r� � _buffer_rows�$_FullyBufferedResultProxy__rowbuffer)r )r r r r� � s z'FullyBufferedResultProxy._init_metadatac C s t �| j�� �S )N)r� r� r� r� )r r r r r� � s z%FullyBufferedResultProxy._buffer_rowsc s | j �� tt| �jf |� d S )N)r� r� r� r� r� )r r� )r r r r� � s z$FullyBufferedResultProxy._soft_closec C s | j r| j �� S | �d �S d S )N)r� r� r� )r r r r r� � s z'FullyBufferedResultProxy._fetchone_implNc C sH |d kr| � � S g }x.td|�D ] }| �� }|d kr6P |�|� q W |S )Nr )r� r� r� r+ )r r� r� r� r r r r r� � s z(FullyBufferedResultProxy._fetchmany_implc C s$ | j s| �g �S | j}t�� | _|S )N)r� r� r� r� r� )r r� r r r r� � s z'FullyBufferedResultProxy._fetchall_impl)N) r7 r8 r9 rZ r� r� r� r� r� r� r� r r )r r r� � s r� c s e Zd Z� fdd�Z� ZS )�BufferedColumnRowc s^ t |�}d}x.|jD ]$}|d k r0||| �||< |d7 }qW t|�}tt| ��||||� d S )Nr rd )r r_ r, r� r� r )r r r r r r/ r! )r r r r � s zBufferedColumnRow.__init__)r7 r8 r9 r r� r r )r r r� � s r� c s6 e Zd ZdZeZ� fdd�Zdd� Zd dd�Z� Z S ) �BufferedColumnResultProxya� A ResultProxy with column buffering behavior. ``ResultProxy`` that loads all columns into memory each time fetchone() is called. If fetchmany() or fetchall() are called, the full grid of results is fetched. This is to operate with databases where result rows contain "live" results that fall out of scope unless explicitly fetched. .. versionchanged:: 1.2 This :class:`.ResultProxy` is not used by any SQLAlchemy-included dialects. c sz t t| ��� | j}|jd krv|j|_dd� tt|j��D �|_i }x*|j � � D ] \}\}}}d ||f||< qPW ||_ d S )Nc S s g | ]}d �qS )Nr )rS r� r r r rT � s z<BufferedColumnResultProxy._init_metadata.<locals>.<listcomp>) r� r� r� r� r_ r r� r$ rU r rV )r r� r �k�funcr r/ )r r r r� � s z(BufferedColumnResultProxy._init_metadatac C s* g }x | � � }|d krP |�|� qW |S )N)r� r+ )r r0 r r r r r� � s z"BufferedColumnResultProxy.fetchallNc C sF |d kr| � � S g }x,t|�D ] }| �� }|d kr4P |�|� q W |S )N)r� r� r� r+ )r r� r0 �ir r r r r� � s z#BufferedColumnResultProxy.fetchmany)N) r7 r8 r9 rZ r� r� r� r� r� r� r r )r r r� � s r� )rZ r� rA � r r Zsqlr r rO Zsqlalchemy.cresultproxyr r � ImportErrorr rk �objectr; Zcollections_abc�Sequence�registerr\ r� r� r� r� r� r r r r �<module> sJ BL e ^-