�
�-:Zc @@ s d d l m Z m Z d d l m Z d d l Z d d l Z d d l m Z m Z m
Z
d d l m Z e j
d d d e j e j B� Z d
e f d � � YZ d e f d
� � YZ d e e f d � � YZ d e f d � � YZ d e e f d � � YZ d S( i ( t print_functiont absolute_import( t partialNi ( t
range_typet text_typet PY2( t errs' \s*((?:INSERT|REPLACE)\b.+\bVALUES?\s*)s6 (\(\s*(?:%s|%\(.+\)s)\s*(?:,\s*(?:%s|%\(.+\)s)\s*)*\))s (\s*(?:ON DUPLICATE.*)?);?\s*\Zt Cursorc B@ ss e Z d Z d Z e Z d � Z d � Z d � Z d � Z d � Z
d � Z d � Z d � Z
d
� Z e d � Z d � Z d d
� Z d � Z d d � Z d d � Z d � Z d � Z d d � Z d � Z d d � Z d � Z d d � Z d � Z d � Z d � Z d � Z e! j" Z" e! j# Z# e! j$ Z$ e! j% Z% e! j& Z& e! j' Z' e! j( Z( e! j) Z) e! j* Z* e! j+ Z+ RS( s�
This is the object you use to interact with the database.
Do not create an instance of a Cursor yourself. Call
connections.Connection.cursor().
i � c C@ sU | | _ d | _ d | _ d | _ d | _ d | _ d | _ d | _ t | _
d S( Ni i����i ( t
connectiont Nonet descriptiont rownumbert rowcountt arraysizet _executedt _resultt _rowst Falset _warnings_handled( t selfR ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt __init__% s c C@ sA | j } | d k r d Sz x | j � r. q WWd d | _ Xd S( sD
Closing a cursor just exhausts all remaining data.
N( R R t nextset( R t conn( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt close0 s c C@ s | S( N( ( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt __enter__= s c G@ s ~ | j � d S( N( R ( R t exc_info( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt __exit__@ s c C@ s" | j s t j d � � n | j S( Ns
Cursor closed( R R t ProgrammingError( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt _get_dbD s c C@ s | j s t j d � � n d S( Ns execute() first( R R R ( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt _check_executedI s c C@ s | S( N( ( R t row( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt _conv_rowM s c G@ s d S( s! Does nothing, required by DB API.N( ( R t args( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt
setinputsizesP t c G@ s d S( s! Does nothing, required by DB API.N( ( R R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt setoutputsizesS R" c C@ sr | j � } | j } | r( | j � n | d k sC | | j k rG d S| j sT d S| j d | � | j � t S( s Get the next query sett
unbufferedN( R R t _show_warningsR t has_nextt next_resultt _do_get_resultt True( R R$ R t current_result( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt _nextsetV s
c C@ s
| j t � S( N( R+ R ( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR e s c @ sb t | t � r! | j � � } n= t | t t f � r^ t | � � � f d � | D� � } n | S( Nc 3@ s$ | ] } � j | d � �Vq d S( t encodingN( t
_ensure_bytes( t .0t v( R, R ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pys <genexpr>l s ( t
isinstanceR t encodet tuplet listt type( R t xR, ( ( R, R s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR- h s
(c @ s� t | j d � j �� t | t t f � re t rK t t � | � � } n t � f d � | D� � St | t � r� t r� t � f d � | j � D� � } n t � f d � | j � D� � St r� � | � } n � j
| � Sd S( NR, c 3@ s | ] } � j | � Vq d S( N( t literal( R. t arg( R ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pys <genexpr>u s c 3@ s- | ]# \ } } � | � � | � f Vq d S( N( ( R. t keyt val( t ensure_bytes( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pys <genexpr>x s c 3@ s* | ] \ } } | � j | � f Vq d S( N( R6 ( R. R8 R9 ( R ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pys <genexpr>z s ( R R- R, R0 R2 R3 R t mapt dictt itemst escape( R R R ( ( R R: s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt _escape_argso s c C@ sV | j � } t r- | j | d | j �} n | d k rR | | j | | � } n | S( s�
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.
R, N( R R R- R, R R? ( R t queryR R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt mogrify� s c C@ sA x | j � r q W| j | | � } | j | � } | | _ | S( s� Execute a query
:param str query: Query to execute.
:param args: parameters used with query. (optional)
:type args: tuple, list or dict
:return: Number of affected rows
:rtype: int
If args is a list or tuple, %s can be used as a placeholder in the query.
If args is a dict, %(name)s can be used as a placeholder in the query.
( R RA t _queryR ( R R@ R t result( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt execute� s c @ s� | s
d St j � � } | r� | j d � d } | j d � j � } | j d � pY d } � j | | | | � j � j � j � St � � f d � | D� � � _ � j S( s� Run several data against one query
:param query: query to execute on server
:param args: Sequence of sequences or mappings. It is used as parameter.
:return: Number of rows affected, if any.
This method improves performance on multiple-row INSERT and
REPLACE. Otherwise it is equivalent to looping over args with
execute().
Ni i i R" c 3@ s! | ] } � j � | � Vq d S( N( RD ( R. R7 ( R@ R ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pys <genexpr>� s ( (
t RE_INSERT_VALUESt matcht groupt rstript _do_execute_manyt max_stmt_lengthR R, t sumR ( R R@ R t mt q_prefixt q_valuest q_postfix( ( R@ R s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt executemany� s "c
C@ s� | j � } | j } t | t � r6 | j | � } n t r] t | t � r] | j | � } n t | t � r~ | j | � } n t | � } t | � } | | t | � | � }
t |
t � r� t r� |
j | � }
q� |
j | d � }
n | |
7} d } x� | D]� } | | | | � }
t |
t � rWt rB|
j | � }
qW|
j | d � }
n t | � t |
� t | � d | k r�| | j
| | � 7} t | � } n
| d 7} | |
7} qW| | j
| | � 7} | | _ | S( Nt surrogateescapei i t ,( R R? R0 R R1 R t bytearrayt itert nextt lenRD R (
R t prefixt valuest postfixR RJ R, R R> t sqlR/ t rowsR7 ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyRI � s>
*
c C@ s� | j � } xM t | � D]? \ } } d | | | j | � f } | j | � | j � q Wd | d j g t t | � � D] } d | | f ^ q{ � f } | j | � | | _ | S( s Execute stored procedure procname with args
procname -- string, name of procedure to execute on server
args -- Sequence of parameters to use with procedure
Returns the original 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.
s SET @_%s_%d=%ss CALL %s(%s)RR s @_%s_%d( R t enumerateR> RB R t joinR RV R ( R t procnameR R t indexR7 t qt i( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt callproc� s
3
c C@ sX | j � | j d k s1 | j t | j � k r5 d S| j | j } | j d 7_ | S( s Fetch the next rowi N( R R R R RV ( R RC ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt fetchone s
'c C@ se | j � | j d k r d S| j | p/ | j } | j | j | !} t | t | j � � | _ | S( s Fetch several rowsN( ( R R R R R
t minRV ( R t sizet endRC ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt fetchmany s
c C@ sX | j � | j d k r d S| j r9 | j | j } n | j } t | j � | _ | S( s Fetch all the rowsN( ( R R R R RV ( R RC ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt fetchall% s
t relativec C@ s� | j � | d k r& | j | } n( | d k r; | } n t j d | � � d | k on t | j � k n s� t d � � n | | _ d S( NRi t absolutes unknown scroll mode %si s out of range( R R R R RV R t
IndexError( R t valuet modet r( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt scroll1 s
%c C@ s3 | j � } | | _ | j | � | j � | j S( N( R t _last_executedR@ R( R ( R R` R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyRB > s
c C@ sx | j � } d | _ | j | _ } | j | _ | j | _ | j | _ | j | _ t
| _ | j st | j
� n d S( Ni ( R R R t
affected_rowsR R
t insert_idt lastrowidR[ R R R t _defer_warningsR% ( R R RC ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR( E s c C@ s� | j r
d St | _ | j r< | j j s8 | j j r< d S| j � j � } | d k r^ d Sxh | D]` } | d } t r� t | t
� r� | j d d � } q� n t j
t j | d d !� d d �qe Wd S( Ni����s utf-8t replacei i t
stackleveli ( R R) R R& t
warning_countR t
show_warningsR R R0 t unicodeR1 t warningst warnR t Warning( R t wst wt msg( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR% T s "
c C@ s t | j d � S( N( RT Rc R ( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt __iter__d s N( (, t __name__t
__module__t __doc__RJ R Rt R R R R R R R R! R# R+ R R R- R? RA RD RP RI Rb Rc Rg Rh Ro RB R( R% R� R R| t Errort InterfaceErrort
DatabaseErrort DataErrort OperationalErrort IntegrityErrort
InternalErrorR t NotSupportedError( ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR sN
$ )
t DictCursorMixinc B@ s e Z e Z d � Z d � Z RS( c C@ s� t t | � j � g } | j r{ xJ | j j D]< } | j } | | k r^ | j d | } n | j | � q/ W| | _ n | r� | j
r� g | j
D] } | j | � ^ q� | _
n d S( Nt .( t superR� R( R
R t fieldst namet
table_namet appendt _fieldsR R ( R R� t fR� Rn ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR( w s c C@ s) | d k r d S| j t | j | � � S( N( R t dict_typet zipR� ( R R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR � s ( R� R� R< R� R( R ( ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR� s s t
DictCursorc B@ s e Z d Z RS( s. A cursor which returns results as a dictionary( R� R� R� ( ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR� � s t SSCursorc B@ s} e Z d Z e Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z d � Z d � Z
d
d
� Z d d � Z RS( s�
Unbuffered 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.
c C@ s | S( N( ( R R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR � s c C@ sr | j } | d k r d S| j d k rJ | j | j k rJ | j j � n z x | j � r_ qP WWd d | _ Xd S( N( R R R t _finish_unbuffered_queryR ( R R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR � s !c C@ s9 | j � } | | _ | j | d t �| j � | j S( NR$ ( R Rp R@ R) R( R ( R R` R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyRB � s
c C@ s | j d t � S( NR$ ( R+ R) ( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR � s c C@ s | j | j j � � S( s
Read next row( R R t _read_rowdata_packet_unbuffered( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt read_next� s c C@ sC | j � | j � } | d k r0 | j � d S| j d 7_ | S( s Fetch next rowi N( R R� R R% R ( R R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyRc � s
c C@ s t | j � � S( s�
Fetch all, as per MySQLdb. Pretty useless for large queries, as
it is buffered. See fetchall_unbuffered(), if you want an unbuffered
generator version of this method.
( R3 t fetchall_unbuffered( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyRh � s c C@ s t | j d � S( s�
Fetch all, implemented as a generator, which isn't to standard,
however, it doesn't make sense to return everything in a list, as that
would use ridiculous memory for large result sets.
N( RT Rc R ( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR� � s c C@ s
| j � S( N( R� ( R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR� � s c C@ s� | j � | d k r" | j } n g } xV t | � D]H } | j � } | d k ra | j � Pn | j | � | j d 7_ q5 W| S( s
Fetch manyi N( R R R
R R� R% R� R ( R Re R[ Ra R ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyRg � s
Ri c C@ s� | j � | d k rg | d k r4 t j d � � n x t | � D] } | j � qA W| j | 7_ nz | d k r� | | j k r� t j d � � n | | j } x t | � D] } | j � q� W| | _ n t j d | � � d S( NRi i s0 Backwards scrolling not supported by this cursorRj s unknown scroll mode %s( R R R� R R� R R ( R Rl Rm t _Rf ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyRo � s"
N( R� R� R� R) Rt R R RB R R� Rc Rh R� R� R Rg Ro ( ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR� � s
t SSDictCursorc B@ s e Z d Z RS( s; An unbuffered cursor, which returns results as a dictionary( R� R� R� ( ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyR� s ( t
__future__R R t functoolsR t reRz t _compatR R R R" R t compilet
IGNORECASEt DOTALLRE t objectR R� R� R� R� ( ( ( s3 /usr/lib/python2.7/site-packages/pymysql/cursors.pyt <module> s � _v