Your IP : 216.73.216.213
�(Z+ � @ s� d d l m Z m Z d d g Z Gd d � d � Z Gd d � d � Z y d d l Z e j Z Wn e k
r{ e Z Yn Xd S) � )�
ProtocolError�
ReplyError�Reader�PyReaderc @ sa e Z d Z d Z e e d d d � Z d d d d � Z d d
� Z d d � Z d
d � Z
d S)r znPure-Python Redis protocol parser that follows hiredis.Reader
interface (except setmaxbuf/getmaxbuf).
Nc C sI t | � s t d � � t | � s0 t d � � t | | | � | _ d S)NzExpected a callable)�callable� TypeError�Parser�_parser)�self�
protocolError�
replyError�encoding� r �@/opt/alt/python35/lib/python3.5/site-packages/aioredis/parser.py�__init__ s
zPyReader.__init__� r c C s� | d k r t | � | } | d k s4 | d k r@ t d � � | | t | � k rb t d � � | j j j | | | | � � d S)zFeed data to parser.r r znegative inputz input is larger than buffer sizeN���)�len�
ValueErrorr �buf�extend)r
�data�o�lr r r �feed s z
PyReader.feedc C s
| j j � S)z�Get parsed value or False otherwise.
Error replies are return as replyError exceptions (not raised).
Protocol errors are raised.
)r � parse_one)r
r r r �gets s z
PyReader.getsc C s d S)zNo-op.Nr )r
�sizer r r � setmaxbuf& s zPyReader.setmaxbufc C s d S)zNo-op.r r )r
r r r � getmaxbuf* s zPyReader.getmaxbufr ) �__name__�
__module__� __qualname__�__doc__r r r r r r r r r r r r s
c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
d � Z d d
� Z d d � Z d d d � Z
d d � Z d S)r c C sF t � | _ d | _ | | _ | | _ | | _ d | _ d | _ d S)Nr )� bytearrayr �posr r r
�_err�_gen)r
r r r
r r r r 0 s zParser.__init__c c s, x% t | j � | j | k r' d Vq Wd S)NF)r r r% )r
r r r r �waitsome9 s zParser.waitsomec c s# | j t | j � d � Ed Hd S)Nr )r( r r )r
r r r �waitany> s zParser.waitanyc c sN | j | j d � s% | j � Ed H| j | j d � } | j d 7_ | S)Nr )r r% r) )r
�valr r r �readoneA s
zParser.readoneNc c s | d k r~ t | j � | d | j k rB | j | d � Ed H| j | } | j | | d � d k r� | j d � � nR | j j d | j � } x7 | d k r� | j � Ed H| j j d | j � } q� W| j | j | � } d | _ | j d | d � =| S)N� s
zExpected b'
'r )r r r% r( �error�findr) )r
r �offsetr* r r r �readlineH s
zParser.readlinec c sT y t | j � Ed H� SWn4 t k
rO } z | j | � � WYd d } ~ Xn Xd S)N)�intr0 r r- )r
�excr r r �readintY s zParser.readintc C s | j | � | _ | j S)N)r r& )r
�msgr r r r- _ s z Parser.errorFc c s1 | j d k r | j � | j � Ed H} | d k r� | j � Ed H} | j d k r~ y | j | j � SWn t k
r} Yn Xt | � S| d k r� | j � Ed H} | j | j d � � S| d k r� | j � Ed HS| d k rS| j � Ed H} | d k rd S| j | � Ed H} | j rIy | j | j � SWn t k
rHYn Xt | � S| d k r| j � Ed H} | d k r�d Sg } d } xp t | � D]b } y" | j
| j d d � Ed H� Wq�t k
r�} z | d k r�| } WYd d } ~ Xq�Xq�W| d k r| � | S| j
d
j | � � � d S)
N� +� -zutf-8� :� $r � *�is_bulkTzInvalid first byte: {!r}r r )r& r+ r0 r
�decode�UnicodeDecodeError�bytesr r3 �range�append�parse� LookupErrorr- �format)r
r: Zctlr* Z
bulk_arrayr- �_�errr r r r@ c sV
" z Parser.parsec C s� | j d k r | j � | _ y | j j d � WnR t k
ri } z d | _ | j SWYd d } ~ Xn" t k
r� d | _ � Yn Xd Sd S)NF)r' r@ �send�
StopIteration�value� Exception)r
r2 r r r r � s
zParser.parse_one) r r! r" r r( r) r+ r0 r3 r- r@ r r r r r r / s .r r N) �errorsr r �__all__r r Zhiredisr � ImportErrorr r r r �<module> s 'q