Path: blob/main/SmartApi/__pycache__/smartSocket.cpython-36.pyc
413 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\smartSocket.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 | j jr| j j| ||� d S )N)r �
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 1 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' r5 r r r r N 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� | j | j� d }| jjr&| r&tj� }t| j|dd� i }tjs~|rrd|d< t j
tj |d�| _ d| j _
| j j� n tj f |� d S )N� )ZcontextFactoryZtimeoutFZinstallSignalHandlers)�targetr T)r@ r4 r ZisSecurer ZClientContextFactoryr r Zrunning� threadingZThreadZrunZwebsocket_threadZdaemon�start)r ZthreadedZdisable_ssl_verification�proxyZcontext_factoryZoptsr r r �connectp s zSmartSocket.connectc C s" | j r| j j| j jkrdS dS d S )NTF)r �stateZ
STATE_OPEN)r r r r � is_connected� s zSmartSocket.is_connectedc C s | j r| j j||� d S )N)r Z sendClose)r r r r r r �_close� s zSmartSocket._closec C s | j ||� dS )z Close the WebSocket connection.N)rI )r r r r r r �close� s zSmartSocket.closec C s t j� 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
N)r �stop)r r r r rK � s zSmartSocket.stopc C s� |}y`dd| j | j| jd�}| jjtjtj|��� d|| j | j| jd�}| jjtjtj|��� dS tk
r� } z | j dj
t |��d� � W Y d d }~X nX d S )NZcn� )ZtaskZchannel�token�userZacctidTz Error while request sending: {})r ) r6 r7 r Z sendMessage�six�b�json�dumps� ExceptionrI �format�str)r rM 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)�printrT rU 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)rX rT rU 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 r8 �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� | j |�}g }�x�|D �]�}| j|dd�}|d@ }|| jd krDdnd}|| jd krZdnd }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�ttd
t |�d��D ]h\} }
| | dk�r@dnd j
| j||
|
d �| j||
d |
d �| | j||
d |
d d d �d �� �q(W | td < qW |j
t� |S ) Nr r, �� r2 g �cAg Y@r3 FT�, �0 �< �@ �last_trade_time�4 Zoi�8 Z oi_day_highZ
oi_day_low� timestamp)�buy�sell� r- rf re r0 �
�H)� byte_format)ZquantityZpriceZorders�depth) �_split_packets� _unpack_int� EXCHANGE_MAPr Z
fromtimestamprS �d� enumerate�rangerY �append)r �bin�packets�dataZpacketZinstrument_tokenZsegmentZdivisorZtradablera rd rk �i�pr r r rZ � s<
&
z6SmartSocket._parse_text_message.<locals>._parse_binary) rO ZPY2�type�bytes�decoderX �base64Z b64decoderQ �loads�
ValueError)r r ru rZ 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 rs rD �endrj r r r rm 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 ri )rj )rY rm rq rr )r rs Znumber_of_packetsrt �jrv Z
packet_lengthr r r rl s zSmartSocket._split_packets)FN)FFN)NN)NN)r~ )r" r# r$ rn r4 r6 r7 r r@ rF rH rI rJ rK rW r? r= r< r> r; r[ rm rl r r r r r( ? s8
?
r( )rO �sysZtimerQ r� ZloggingrC r{ r Ztwisted.internetr r Ztwisted.pythonr Z twisted_logZautobahn.twisted.websocketr r r r r& �objectr( r r r r �<module> s