Path: blob/main/SmartApi/__pycache__/socketTP.cpython-36.pyc
722 views
3
���_{* � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
m Z d dl m
Z d dlmZmZmZ G dd� de�ZG dd � d e�ZG d
d � d e�ZdS ) � N)�datetime)�reactor�ssl)�log)�WebSocketClientProtocol�WebSocketClientFactory� connectWSc s< e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Z� ZS ) �SmartSocketClientProtocolc s t t| �j||� d S )N)�superr �__init__)�self�args�kwargs)� __class__� �'C:\smartapi-python\SmartApi\socketTP.pyr s z"SmartSocketClientProtocol.__init__c C s" | | j _| j jr | j j| |� dS )z7Called when WebSocket server connection was establishedN)�factory�ws�
on_connect)r �responser r r � onConnect s z#SmartSocketClientProtocol.onConnectc C s | j jr| j j| � d S )N)r �on_open)r r r r �onOpen s z SmartSocketClientProtocol.onOpenc C s( t d||� | jjr$| jj| ||� d S )Nz/Called when text or binary message is received.)�printr �
on_message)r �payload� is_binaryr r r � onMessage s z#SmartSocketClientProtocol.onMessagec C s8 |s | j jr | j j| ||� | j jr4| j j| ||� dS )z!Called when connection is closed.N)r �on_error�on_close)r Z was_clean�code�reasonr r r �onClose$ s
z!SmartSocketClientProtocol.onClose) �__name__�
__module__� __qualname__r r r r r" �
__classcell__r r )r r r s
r c s e Zd ZeZ� fdd�Z� ZS )�SmartSocketClientFactoryc s@ d| _ d| _d| _d| _d| _d| _d| _tt| �j ||� dS )z/Initialize with default callback method values.FN)
�debugr r r r r r r
r' r )r r
r )r r r r 0 s z!SmartSocketClientFactory.__init__)r# r$ r% r Zprotocolr r& r r )r r r' - s r' c
@ s� e Zd Zddddddddd d
� Zd Zd Zd Zd/dd�Zdd� Zd0dd�Z dd� Z
d1dd�Z d2dd�Z dd� Z
d d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd3d+d,�Zd-d.� Zd S )4� SmartSocket� � � � � � � � � ) ZnseZnfo�cdsZbseZbfoZbsecdsZmcxZmcxsx�indicesz>wss://omnefeeds.angelbroking.com/NestHtml5Mobile/socket/streamNFc C sF |p| j | _|| _|| _|| _d | _d | _d | _d | _d | _ d | _
d S )N) �ROOT_URI�root�
feed_token� client_coder( �on_ticksr r r r r )r �
FEED_TOKEN� CLIENT_CODEr( r6 r r r r M s zSmartSocket.__init__c K sX t |f|�| _| jj| _| j| j_| j| j_| j| j_| j| j_ | j
| j_ | j | j_
dS )z%Create a WebSocket client connection.N)r' r r r( �_on_openr � _on_errorr � _on_closer � _on_messager � _on_connectr )r Zurlr r r r �_create_connection^ s
z SmartSocket._create_connectionc C s� t d� | j| j� d }t | jj|� | jjr<| r<tj� }t d|� t| j|dd� i }tj s�|r�t d� d|d< t
j tj |d�| _
d | j
_| j
j� n tj f |� d S )
NZConnect�context_factory� )ZcontextFactoryZtimeoutzinside threadedFZinstallSignalHandlers)�targetr T)r rA r5 r ZisSecurer ZClientContextFactoryr r Zrunning� threadingZThreadZrunZwebsocket_threadZdaemon�start)r ZthreadedZdisable_ssl_verification�proxyrB Zoptsr r r �connecto s"
zSmartSocket.connectc C s* t d� | jr"| jj| jjkr"dS dS d S )Nz-Check if WebSocket connection is established.TF)r r �stateZ
STATE_OPEN)r r r r � is_connected� s zSmartSocket.is_connectedc C s t d� | jr | jj||� d S )Nz Close the WebSocket connection.)r r Z sendClose)r r r! r r r �_close� s zSmartSocket._closec C s | j ||� dS )z Close the WebSocket connection.N)rK )r r r! r r r �close� s zSmartSocket.closec C s t d� tj� dS )z�Stop the event loop. Should be used if main thread has to be closed in `on_close` method.
Reconnection mechanism cannot happen past this method
�stopN)r r rM )r r r r rM � s zSmartSocket.stopc C s� t d� d}y�t d� dd| j| j| jd�}| jjtjtj|��� t |� d|| j| j| jd�}t |� | jjtjtj|��� t |� dS t k
r� } z | j
dj t |��d � � W Y d d }~X nX d S )
Nz Request Sendz$nse_cm|2885&nse_cm|1594&nse_cm|11536ZInsideZcn� )ZtaskZchannel�token�userZacctidTz Error while request sending: {})r! )
r r7 r8 r Z sendMessage�six�b�json�dumps� ExceptionrK �format�str)r rO ZstrwatchlistscripsZrequest�er r r � send_request� s" zSmartSocket.send_requestc C s || _ | jr| j| |� d S )N)r r )r r r r r r r@ � s zSmartSocket._on_connectc C s, t dj|t|��� | jr(| j| ||� dS )z3Call `on_close` callback when connection is closed.zConnection closed: {} - {}N)r rV rW r )r r r r! r r r r> � s zSmartSocket._on_closec C s, t dj|t|��� | jr(| j| ||� dS )z9Call `on_error` callback when connection throws an error.zConnection error: {} - {}N)r rV rW r )r r r r! r r r r= � s zSmartSocket._on_errorc C sN | j r| j | ||� | jr<|r<t|�dkr<| j| | j|�� |sJ| j|� dS )z9Call `on_message` callback when text message is received.r- N)r r9 �len�
_parse_binary�_parse_text_message)r r r r r r r r? � s zSmartSocket._on_messagec C s | j r| j | �S d S )N)r )r r r r r r< � s zSmartSocket._on_openc C sv t j r<t|�tkr<|jd�}td|� tj|�}td|� ytj |�}td|� W n t
k
rh dS X dd� }dS )zParse text message.zutf-8ZPAYLOADZDATANc S s� t d� | j|�}g }�x�|D �]�}| j|dd�}|d@ }|| jd krLdnd}|| jd krbd nd
}ytj| j|d d ��} W n tk
r� d } Y nX ytj| j|d
d��}
W n tk
r� d }
Y nX | td< | j|d d�td< | j|dd�td< | j|dd
�td< |
td< g g d�} x�tt dt
|�d��D ]h\} }
| | dk�rHdnd j | j||
|
d �| j||
d |
d �| | j||
d |
d d d �d �� �q0W | td < q W |j t� |S )!Nz1Parse binary data to a (list of) ticks structure.r r- �� r3 g �cAg Y@r4 FT�, �0 �< �@ �last_trade_time�4 Zoi�8 Z oi_day_highZ
oi_day_low� timestamp)�buy�sell� r. rg rf r1 �
�H)� byte_format)ZquantityZpriceZorders�depth) r �_split_packets� _unpack_int� EXCHANGE_MAPr Z
fromtimestamprU �d� enumerate�rangerZ �append)r �bin�packets�dataZpacketZinstrument_tokenZsegmentZdivisorZtradablerb re rl �i�pr r r r[ � s>
&
z6SmartSocket._parse_text_message.<locals>._parse_binary) rQ ZPY2�type�bytes�decoder �base64Z b64decoderS �loads�
ValueError)r r rv r[ r r r r\ � s
z SmartSocket._parse_text_message�Ic C s t jd| |||� �d S )z(Unpack binary data as unsgined interger.�>r )�structZunpack)r rt rF �endrk r r r rn s zSmartSocket._unpack_intc C s� t |�dk rg S | j|dddd�}g }d}xPt|�D ]D}| j|||d dd�}|j||d |d | � � |d | }q4W |S )z.Split the data to individual packets of ticks.r+ r rj )rk )rZ rn rr rs )r rt Znumber_of_packetsru �jrw Z
packet_lengthr r r rm s zSmartSocket._split_packets)FN)FFN)NN)NN)r )r# r$ r% ro r5 r7 r8 r rA rH rJ rK rL rM rY r@ r> r= r? r< r\ rn rm r r r r r) > s8
?
r) )rQ �sysZtimerS r� ZloggingrE r| r Ztwisted.internetr r Ztwisted.pythonr Z twisted_logZautobahn.twisted.websocketr r r r r' �objectr) r r r r �<module> s