o
1li@ @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ e e
Z G dd dej Z
G d d
d
e
ZG d d d e
ZG d
d de
ZG dd de
ZG dd dej Zdd ZdS )zBHI) joinstructpackr r lenr r FRAME_END)r piecespayloadr
r
r _marshal s
zFrame._marshalc C s t )zKTo be ended by child classes
:raises NotImplementedError
)NotImplementedErrorr r
r
r marshal) s z
Frame.marshalN)__name__
__module__ __qualname____doc__NAMEr r r r
r
r
r r
s
r c @ s$ e Zd ZdZdZdd Zdd ZdS )MethodzBase Method frame object mapping. AMQP method frames are mapped on top
of this class for creating or accessing their data and attributes.
METHODc C t | tj| || _dS )zCreate a new instance of a frame
:param int channel_number: The frame type
:param pika.Spec.Class.Method method: The AMQP Class.Method
N)r r r FRAME_METHODmethod)r r r" r
r
r r 9
zMethod.__init__c C s, | j }|dtd| j j | |S )PReturn the AMQP binary encoded value of the frame
:rtype: str
r >I)r" encodeinsertr r INDEXr r r r
r
r r C s
zMethod.marshalNr r r r r r r r
r
r
r r 2
r c @ $ e Zd ZdZd Zdd Zdd ZdS )HeaderzHeader frame object mapping. AMQP content header frames are mapped
on top of this class for creating or accessing their data and attributes.
c C s t | tj| || _|| _dS )a
Create a new instance of a AMQP ContentHeader object
:param int channel_number: The channel number for the frame
:param int body_size: The number of bytes for the body
:param pika.spec.BasicProperties props: Basic.Properties object
N)r r r FRAME_HEADER body_size
properties)r r r/ propsr
r
r r U s
zHeader.__init__c C s0 | j }|dtd| j j| j | |S )r$ r z>HxxQ)r0 r& r' r r r( r/ r r) r
r
r r a s
zHeader.marshalNr* r
r
r
r r- N s
r- c @ r, )BodyzBody frame object mapping class. AMQP content body frames are mapped on
to this base class for getting/setting of attributes/data.
c C r )z_
Parameters:
- channel_number: int
- fragment: unicode or str
N)r r r
FRAME_BODYfragment)r r r4 r
r
r r t r# z
Body.__init__c C s | | jgS r$ )r r4 r r
r
r r ~ s z Body.marshalNr* r
r
r
r r2 m r+ r2 c @ r, ) HeartbeatzHeartbeat frame object mapping class. AMQP Heartbeat frames are mapped
on to this class for a common access structure to the attributes/data
values.
c C s t | tjd dS )z,Create a new instance of the Heartbeat framer N)r r r FRAME_HEARTBEATr r
r
r r s zHeartbeat.__init__c C s | t S r5 )r listr r
r
r r s zHeartbeat.marshalNr* r
r
r
r r6 s
r6 c @ s& e Zd ZdZd ZdddZdd ZdS )ProtocolHeaderzqAMQP Protocol header frame class which provides a pythonic interface
for creating AMQP Protocol headers
Nc C s: d| _ |p tjd | _|ptjd | _|ptjd | _dS )zConstruct a Protocol Header frame object for the specified AMQP
version
:param int major: Major version number
:param int minor: Minor version number
:param int revision: Revision
r N)r r PROTOCOL_VERSIONmajorminorrevision)r r> r? r@ r
r
r r s zProtocolHeader.__init__c C s dt dd| j| j| j S )zReturn the full AMQP wire protocol frame data representation of the
ProtocolHeader frame
:rtype: str
AMQPBBBBr )r r r> r? r@ r r
r
r r s zProtocolHeader.marshal)NNNr* r
r
r
r r9 s
r9 c C s z | dd dkr t d| d\}}}dt|||fW S W n
tt jfy* Y dS w zt d| dd \}}}W n t jyE Y dS w tj| tj }|t | krVdS | |d
| t
tj krht
d | tj|d
}|tjkrt d |d } tj| }
|
|d |t||
fS |tjkrt d
|\} } }
tj| }||dd }|t||
|fS |tjkr|t||fS |tjkr|t fS t
d| )a Receives raw socket data and attempts to turn it into a frame.
Returns bytes used to make the frame and the frame
:param str data_in: The raw data stream
:rtype: tuple(bytes consumed, frame)
:raises: pika.exceptions.InvalidFrameError
r rA BBB )r Nz>BHL r; zInvalid FRAME_END markerr% z>HHQ NzUnknown frame type: %i)r unpack_fromr9
IndexErrorerrorunpackr FRAME_HEADER_SIZEFRAME_END_SIZEr r r r InvalidFrameErrorr! methodsdecoder r. r1 r- r3 r2 r7 r6 )data_inr> r? r@ r r
frame_size frame_end
frame_data method_idr" class_idweightr/ r0 outr
r
r decode_frame sH
rZ )r loggingr pikar r r pika.compatr getLoggerr LOGGER
AMQPObjectr r r- r2 r6 r9 rZ r
r
r
r s
%