o ͿSi*@spddlmZmZddlmZmZddlZddlZddl Z ddl Z ddl Z e dZ e ejGdddZdS)) KiteConnect KiteTicker)OptionalUnionN ZerodhaClientc@seZdZdMdededeefddZdedefd d Zd e ddfd d Z dNdede fddZ defddZ dede fddZdede fddZ        dOded e d!eded"ed#ed$ed%eed&eed'ee d(eedefd)d*Zde fd+d,Zdefd-d.Zde fd/d0Zd1edefd2d3Zdefd4d5Zdefd6d7ZdMd8eede fd9d:Zd;edefdede fd?d@Z A AdPdBeee fdCedDedEedFedGedejfdHdIZdAdAdJdBeee fdCedFedGedef dKdLZ dS)QrNapi_key api_secret access_tokencCsz||_||_||_t|jd|_|jr|j|ji|_i|_tt dd|_ t t dd|_ t t dd|_dS)N)rZERODHA_MAX_RETRIES3ZERODHA_RETRY_BASE_SLEEP_SECz0.6ZERODHA_THROTTLE_SLEEP_SECz0.05)rrr rkiteset_access_token_instruments_cache_instruments_cache_tsintosgetenv _max_retriesfloat_retry_base_sleep_throttle_sleep)selfrrr rrr zZerodha session establishedz"Failed to generate Zerodha sessionN) rgenerate_sessionrr rloggerinfor6 exception)rr> session_datarrrrr?Hs   zZerodhaClient.generate_session refresh_tokenc Cs\z|j|}|d|_|j|jtd|WSty-}ztd|d}~ww)z+(Optional) If using refresh token mechanismr zAccess token refreshedzFailed to refresh access tokenN)rrDr rr@rAr6rB)rrDrCrrrrrDWs    zZerodhaClient.refresh_tokenMARKETCNCDAYsymbolquantitytransaction_type order_typeproductvalidityprice trigger_pricedisclosed_quantitytagc  Csz.|jj||||||||| | | d } dd| D} |jjdi| } td| | WStyD}z td||d}~ww)zPlace a market order) variety tradingsymbolr.rJrIrKrLrMrNrOrPrQcSsi|] \}}|dur||qSNr).0kvrrr sz-ZerodhaClient.place_order..zOrder placed successfully: zOrder placement failed for Nr)rVARIETY_REGULARitems place_orderr@rAr6rB)rrHrIrJr.rKrLrMrNrOrPrQpayloadorder_idrrrrr[cs.zZerodhaClient.place_orderc C4z|jWSty}ztd|d}~ww)NzFailed to fetch profile)rprofiler6r@rBrrrrr get_profile  zZerodhaClient.get_profilec Cr^)NzError fetching holdings)rholdingsr6r@rBr`rrr get_holdingsrbzZerodhaClient.get_holdingsc Cr^)NzError fetching positions)r positionsr6r@rBr`rrr get_positionsrbzZerodhaClient.get_positionsr]c Cs8z|jj|dWSty}ztd|d}~ww)N)r]zError fetching order history)r order_historyr6r@rB)rr]rrrrget_order_historys zZerodhaClient.get_order_historyc Cr^)NzError fetching orders)rordersr6r@rBr`rrr get_ordersrbzZerodhaClient.get_ordersc Cs4z|jWSty}ztd|d}~ww)zReturn recent executed trades (Kite tradebook). Note: KiteConnect returns the user's trades for the day/session. zError fetching tradesN)rtradesr6r@rBr`rrr get_tradess  zZerodhaClient.get_tradessegmentc CsZz|rt|r|jt|WS|jWSty,}ztd|d}~ww)zReturn available margins (account-level). KiteConnect: - margins() => all segments - margins(segment) => specific segment, e.g. "equity" zError fetching marginsN)r r1rmarginsr6r@rB)rrmrrrr get_marginss  zZerodhaClient.get_marginsric Cs6z|j|WSty}ztd|d}~ww)zReturn per-order margin requirements. Input: list of order dicts for KiteConnect `order_margins`. This is read-only and does not place orders. zError fetching order marginsN)r order_marginsr6r@rB)rrirrrrget_order_marginss zZerodhaClient.get_order_marginssymbolsc Csd}t|jdD]?}z|jdkrt|j|j|WStyH}z|}||jkr>||r>| |WYd}~q t dd}~wwt d|)z%symbols = ['NSE:RELIANCE', 'NSE:TCS']Nr0rzError fetching quotes) r4rrr(r)rquoter6r#r+r@rB)rrrr:r$rrrr get_quotes"      zZerodhaClient.get_quoteFinstrument_tokeninterval from_dateto_date continuousoic Csd}t|jdD]b}z4|jdkrt|j|jj||||||d} t| } | j s;t | d| d<| j ddd| WSt yk} z | }||jkr^| | r^||WYd} ~ q td| d} ~ wwtd||) zFetch historical OHLC dataNr0rrurwrxrvryrzdateT)inplacez Error fetching historical data: )r4rrr(r)rhistorical_datapd DataFrameempty to_datetime set_indexr6r#r+r@error) rrurvrwrxryrzr:r$datadfrrrrget_historical_datas:       z!ZerodhaClient.get_historical_data)ryrzc Csd}t|jdD]L}z|jdkrt|j|jj||||||d} | p&gWStyU} z | }||jkrH|| rH| |WYd} ~ q t d| d} ~ wwt d||)zFetch historical data as raw Kite records with retry/throttle. This is used for intraday-only snapshots where the caller wants the raw list (and does its own trimming/caching). Nr0rr{z#Error fetching historical records: ) r4rrr(r)rr~r6r#r+r@r) rrurvrwrxryrzr:r$rrrrrget_historical_data_records s2    z)ZerodhaClient.get_historical_data_recordsrT)r,r-)r,rErFrGNNNN)FF)!__name__ __module__ __qualname__r rrr6boolr#rr+r;r=dictr?rDrr[ralistrdrfrhrjrlrorqrtrrrrrrrrrr s     (    .  ) kiteconnectrrtypingrrpandasrloggingrr(r' getLoggerr@setLevelINFOrrrrrs