Path: blob/main/SmartApi/__pycache__/smartConnect.cpython-36.pyc
413 views
3
�|�_E4 � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
j Z d dl
Z
d dl
mZ d dlZd dlZd dlZd dlmZmZ eje�ZG dd� de�ZdS )� )�urljoinN)�get)� __version__� __title__c @ s� e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d d d!d!d"d#d$d%d&d'd(d)d*d+d,d-�Z!dhd0d1�Z"d2d3� Z#d4d5� Z$d6d7� Z%d8d9� Z&d:d;� Z'd<d=� Z(did>d?�Z)djd@dA�Z*dkdBdC�Z+dldDdE�Z,dmdFdG�Z-dHdI� Z.dJdK� Z/dLdM� Z0dNdO� Z1dPdQ� Z2dRdS� Z3dTdU� Z4dVdW� Z5dXdY� Z6dZd[� Z7d\d]� Z8d^d_� Z9d`da� Z:dbdc� Z;ddde� Z<dfdg� Z=d.S )n� SmartConnectz#https://apiconnect.angelbroking.comz1https://smartapi.angelbroking.com/publisher-login� ZMISZCNCZNRMLZCOZBOZMARKET�LIMITzSL-MZSLZregularZbo�coZamo�BUYZSELL�DAYZIOC�NSEZBSEZNFOZCDSZBFOZMCXZNCDEXZCOMPLETEZREJECTEDZ CANCELLEDz//rest/auth/angelbroking/user/v1/loginByPasswordz(/rest/secure/angelbroking/user/v1/logoutz-/rest/auth/angelbroking/jwt/v1/generateTokensz,/rest/secure/angelbroking/user/v1/getProfilez-/rest/secure/angelbroking/order/v1/placeOrderz./rest/secure/angelbroking/order/v1/modifyOrderz./rest/secure/angelbroking/order/v1/cancelOrderz//rest/secure/angelbroking/order/v1/getOrderBookz-/rest/secure/angelbroking/order/v1/getLtpDataz//rest/secure/angelbroking/order/v1/getTradeBookz(/rest/secure/angelbroking/user/v1/getRMSz1/rest/secure/angelbroking/portfolio/v1/getHoldingz./rest/secure/angelbroking/order/v1/getPositionz2/rest/secure/angelbroking/order/v1/convertPosition)z api.loginz
api.logoutz api.tokenz api.refreshzapi.user.profilezapi.order.placezapi.order.modifyzapi.order.cancelzapi.order.bookz api.ltp.datazapi.trade.bookz
api.rms.limitz api.holdingz api.positionzapi.convert.positionNFc C s� || _ || _d | _|
| _|| _|| _|| _|r2|ni | _|p@| j| _ |pL| j
| _ | r�t j
� | _t jjf | �} | jjd| � td� nt | _t jjj� d S )Nzhttps://zin pool)�debug�api_key�session_expiry_hook� disable_ssl� access_token�
refresh_token�userId�proxies�_rootUrl�root�_default_timeout�timeout�requestsZSessionZ
reqsessionZadaptersZ HTTPAdapterZmount�printZpackagesZurllib3Zdisable_warnings) �selfr r r r r r
r r Zpoolr Z
reqadapter� r �+C:\smartapi-python\SmartApi\smartConnect.py�__init__X s"
zSmartConnect.__init__c C s t |�std��|| _d S )Nz0Invalid input type. Only functions are accepted.)�callable� TypeErrorr )r �methodr r r �setSessionExpiryHooko s z!SmartConnect.setSessionExpiryHookc C s t S )N)r r r r r � getUserIdt s zSmartConnect.getUserIdc C s
|| _ d S )N)r )r �idr r r � setUserIdw s zSmartConnect.setUserIdc C s
|| _ d S )N)r )r r r r r �setAccessTokenz s zSmartConnect.setAccessTokenc C s
|| _ d S )N)r )r r r r r �setRefreshToken~ s z SmartConnect.setRefreshTokenc C s d| j | jf S )zYGet the remote login url to which a user should be redirected to initiate the login flow.z
%s?api_key=%s)�
_login_urlr )r r r r � login_url� s zSmartConnect.login_urlc C sX |r |j � ni }| j| jf |�}t|� t| j|�}t|� tj� }tj|�}t d�j
} dj t j
ddtj� ��}
| j} d} d}
d}| || |
| | |
|d�}| jr�| j}d j|�|d
< | jr�tjd j||||d �� yZtj|||d$kr�tj|�nd|d%k�rtj|�nd|| j d| j| jd� }td|j� W n( t k
�rV } z
|�W Y dd}~X nX | j�rxtjdj|j |jd�� d|d k�r&ytj |jj d��}W n* t k
�r� t!j"dj|jd���Y nX |j d ��r"| j#�r�|j d k�r�|d d k�r�| j#� t$t!|d t!j%�}||d |j d ��|S d!|d k�r:|jS t!j"d"j|d |jd#���dS )&zMake an HTTP request.zhttps://api.ipify.org�:z..z%012xzapplication/json�USERZWEB)z Content-typezX-ClientLocalIPzX-ClientPublicIPz X-MACAddressZAcceptz X-PrivateKeyz
X-UserTypez
X-SourceIDz Bearer {}Z
Authorizationz*Request: {method} {url} {params} {headers})r! �url�params�headers�POST�PUTN�GET�DELETET)�datar- r. ZverifyZallow_redirectsr r zThe Response ContentzResponse: {code} {content})�code�content�jsonz Content-type�utf8zDCouldn't parse the JSON response received from the server: {content})r5 Z
error_typei� ZTokenException�message)r4 �csvz@Unknown Content-type ({content_type}) with response: ({content}))Z content_typer5 )r/ r0 )r1 r2 )&�copy�_routes�formatr r r �socketZ gethostnameZ
gethostbynamer �text�join�reZfindall�uuidZgetnoder r r
�logr Zrequestr6 �dumpsr r r r5 � ExceptionZ status_code�loads�decode�
ValueError�exZ
DataExceptionr �getattrZGeneralException)r �router! Z
parametersr- Zurir, ZhostnameZ
clientLocalIPZclientPublicIPZ
macAddressZ
privateKeyZacceptZuserTypeZsourceIDr. Z auth_header�r�er3 Zexpr r r �_request� st
"zSmartConnect._requestc C s | j |d|�S )z#Alias for sending a DELETE request.r2 )rM )r rJ r- r r r �_deleteRequest� s zSmartConnect._deleteRequestc C s | j |d|�S )z Alias for sending a PUT request.r0 )rM )r rJ r- r r r � _putRequest� s zSmartConnect._putRequestc C s | j |d|�S )z!Alias for sending a POST request.r/ )rM )r rJ r- r r r � _postRequest� s zSmartConnect._postRequestc C s | j |d|�S )z Alias for sending a GET request.r1 )rM )r rJ r- r r r � _getRequest� s zSmartConnect._getRequestc C s� ||d�}| j d|�}|d d }| j|� |d d }| j|� | j|�}|d d }t|� | j|� d| |d d< ||d d< td|� |S ) N)�
clientcode�passwordz api.loginr3 �jwtToken� refreshTokenrR zBearer r+ )rP r& r' �
getProfiler r% ) r �
clientCoderS r- ZloginResultObjectrT rU �userr$ r r r �generateSession� s
z SmartConnect.generateSessionc C s | j dd|i�}|S )Nz
api.logoutrR )rP )r rW ZlogoutResponseObjectr r r �terminateSession� s z SmartConnect.terminateSessionc C s* | j dd|i�}|d d }| j|� |S )Nz api.tokenrU r3 rT )rP r& )r r �responserT r r r �
generateToken s
zSmartConnect.generateTokenc C sP | j d| j| jd��}i }d|kr2|d d |d< | j|d< |d d |d< |S )Nz api.refresh)rT rU rT r3 rR rU )rP r r r )r r[ ZtokenSetr r r �renewAccessToken s
z SmartConnect.renewAccessTokenc C s | j dd|i�}td|� |S )Nzapi.user.profilerU z USER PROFILE)rQ r )r rU rX r r r rV s
zSmartConnect.getProfilec C sD |}x&t |j� �D ]}|| d kr||= qW | jd|�d d }|S )Nzapi.order.placer3 �orderid)�list�keysrP )r � orderparamsr- �k�
orderResponser r r �
placeOrder s
zSmartConnect.placeOrderc C s< |}x&t |j� �D ]}|| d kr||= qW | jd|�}|S )Nzapi.order.modify)r_ r` rP )r ra r- rb rc r r r � modifyOrder+ s
zSmartConnect.modifyOrderc C s | j d||d��}|S )Nzapi.order.cancel)�varietyr^ )rP )r Zorder_idrf rc r r r � cancelOrder7 s zSmartConnect.cancelOrderc C s |||d�}| j d|�}|S )N)�exchange�
tradingsymbol� symboltokenz api.ltp.data)rP )r rh ri rj r- ZltpDataResponser r r �ltpData; s
zSmartConnect.ltpDatac C s | j d�}|S )Nzapi.order.book)rQ )r ZorderBookResponser r r � orderBookD s
zSmartConnect.orderBookc C s | j d�}|S )Nzapi.trade.book)rQ )r ZtradeBookResponser r r � tradeBookI s
zSmartConnect.tradeBookc C s | j d�}|S )Nz
api.rms.limit)rQ )r ZrmsLimitResponser r r �rmsLimitM s
zSmartConnect.rmsLimitc C s | j d�}|S )Nz api.position)rQ )r ZpositionResponser r r �positionQ s
zSmartConnect.positionc C s | j d�}|S )Nz api.holding)rQ )r ZholdingResponser r r �holdingU s
zSmartConnect.holdingc C s< |}x&t |j� �D ]}|| d kr||= qW | jd|�}|S )Nzapi.convert.position)r_ r` rP )r ZpositionParamsr- rb ZconvertPositionResponser r r �convertPositionY s
z SmartConnect.convertPositionc C s t d j� t S )Nz-python/)r �
capitalizer )r r r r � _user_agentb s zSmartConnect._user_agent)
NNNNNFNNNF)N)N)N)N)N)>�__name__�
__module__� __qualname__r r( r Z PRODUCT_MISZ PRODUCT_CNCZ PRODUCT_NRMLZ
PRODUCT_COZ
PRODUCT_BOZORDER_TYPE_MARKETZORDER_TYPE_LIMITZORDER_TYPE_SLMZ
ORDER_TYPE_SLZVARIETY_REGULARZ
VARIETY_BOZ
VARIETY_COZ VARIETY_AMOZTRANSACTION_TYPE_BUYZTRANSACTION_TYPE_SELLZ VALIDITY_DAYZ VALIDITY_IOCZ EXCHANGE_NSEZ EXCHANGE_BSEZ EXCHANGE_NFOZ EXCHANGE_CDSZ EXCHANGE_BFOZ EXCHANGE_MCXZEXCHANGE_NCDEXZSTATUS_COMPLETEZSTATUS_REJECTEDZSTATUS_CANCELLEDr; r r" r# r% r& r' r) rM rN rO rP rQ rY rZ r\ r] rV rd re rg rk rl rm rn ro rp rq rs r r r r r s�
U
r )Zsix.moves.urllib.parser �sysr9 r6 Zdateutil.parserZdateutilZhashlibZloggingZdatetimeZsmartapi.smartExceptionsZsmartExceptionsrH r r r@ rA r= Zsmartapi.versionr r Z getLoggerrt rB �objectr r r r r �<module> s