o ͿSi0 @sddlZddlmZmZmZddlmZmZmZmZzddl Z dZ Wn e y/dZ dZ Ynwddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZd dlmZmZeeZ dedeefddZ!dedee"fddZ#deee$efde$deee$effddZ%defddZ&dedee$effddZ'de$dee(fdd Z)  d%dede$d!ee$d"e*dee$eff d#d$Z+dS)&N)datedatetime timedelta)AnyDictListOptionalTF) HTTPException) ZerodhaClient)IndicatorCalculator)StrategyFeatureCalculator)GLOBAL_INTRADAY_STORE)get_stock_history)refresh_stocks_master)MAX_CANDLES_PER_TIMEFRAMETEGPT_VERBOSE_LOGSvreturnc Cs|durdSt|trt|ts|St|tr|St|tr_|}|s(dSz t|ddWSty^t|dkr[zt|ddWYStyZYYdSwYdSwdS)NZz+00:00 ) isinstancerrstrstrip fromisoformatreplace Exceptionlen)rsrD/var/www/html/Trade-python/app/v1/services/tegpt/zerodha_services.py_as_dates.     r!cCs8z|dus t|tr WdSt|WStyYdSwN)rboolfloatr)rrrr _num/s  r%dailymodec Cs&|sgSg}|D]S}t|tsqt|d}t|d}t|d}t|d}t|d}|dusG|dusG|dusG|dusG|durHqt|d} ||||||| dq|s`gS|jd d d i} g} |D]D}|d}|d kr|} t t | j t | j d } | }ntt |j t |jd } | }|| vrg| |<| || ||qng}| D]Y}| |pg}|sq|dd}|dd}tdd|D}tdd|D}d}d}|D]}|ddurqd}|t|d7}q|||||d}|r ||d<||q|S)zAggregate daily candles into weekly/monthly candles. mode: - "week" => ISO week buckets (week start = Monday) - "month" => calendar month buckets (month start = day 1) ropenhighlowcloseNvolume)rr(r)r*r+r,cSs|dS)Nrr)xrrr Tsz"_aggregate_ohlcv..)keyweekrrcs|]}|dVqdS)r)Nr.0r-rrr nz#_aggregate_ohlcv..csr2)r*Nrr3rrr r5or6gFT)rr(r)r*r+)rdictr!getr%appendsort isocalendarrfromisocalendarintyearr0 isoformatmonthmaxminr$)r&r'rowsrdohlcrbucketsorderisostartr/outbv_sumhas_volr-candlerrr _aggregate_ohlcv8sj (     rScCs`t|d}|dd|i}|stddd|ds#tdd dt|d |d |dd S) z$Get Zerodha client for current user._idzerodha_settingsuser_idizZerodha settings not found status_codedetail access_tokenizZerodha not authenticatedapi_key api_secret)r[r\rZ)rr8find_oner r )db current_userrVsettingsrrr get_zerodha_client_services   razerodha_clientc Csz(t||dgdd}t|ddt|ddt|dd|dd WStyD}ztd td d t|d d}~ww)zwRefresh stocks master list from Zerodha instruments. IMPORTANT: This is the only allowed stock creation path. NSET)r^rb exchangesallow_cache_writeinsertedrupdatedkept timestamp)countrizFailed to refresh stocks masterizStocks master refresh failed: rWN)rr=r8rlogger exceptionr r)r^rbreserrr refresh_instruments_services> rosymbolcCspz-|pd}|d|ddddi}t|tr |dnd}|dur+t|WSdWSty7YdSw)z:Get instrument token for symbol from `stocks` master list.stocksrcrpexchangeinstrument_tokenrN)rupperr]rr7r8r=r)r^rpsymstocktokrrr get_instrument_tokens rz timeframes include_quotec s>|tiiddztd||pd}|s%dd<WSd}|durLz|d|dd p>|dd |i}Wn tyKd}Ynwt |t s\d |d d<WS| d }| d}|d <|d<|dur|d|d<WS|rz| d|g}| d|id<Wnty} zt d|| id<WYd} ~ nd} ~ wwg} |durdd|pgD} d| vsd| vsd| vrzt|t|ddd} Wn tyg} Ynw|D]} zct| pd} | dvr2|durgd| <nF| dkr| ddd| <n6| dkr't| ddd| <n&t| ddd| <n| dvrGtj|t|| ddd| <ngd| <Wqtyo} zt d|| | gd| <WYd} ~ qd} ~ wwi}trdD]9\} }|sq{zt|}t|}|r||| <Wq{ty} zt d|| | WYd} ~ q{d} ~ ww|r|d <trz-d dpg}|rt|}t|}d|id!<t|}|rd|id"<Wnty} z t d#|| WYd} ~ nd} ~ wwztj di d"ipi dd$}|r$|d%<Wnty?} z t d&|| WYd} ~ nd} ~ wwztt}Wn tyRd'}Ynw|r|d'krz* dpciD]\} }t |t rt!||kr|| dd| <qfWn tyYnwdrt"d(d)d#Dnd'}zt dpi$}fd*d+|D}Wn tyi}Ynwt%d,| drd-nd.|t|t&rt%d/|ds|d'krd0}t d1|||d<WSty} zt'd2|t| d<WYd} ~ Sd} ~ ww)3z-Fetch comprehensive market data for analysis.N)rpriquotecandleserrorz$[teGPT] Market data build start | %srqzsymbol is requiredrrrrcrsrpz'Stock not found in stocks master list: z. Refresh stocks master first.rustock_idz3instrument_token missing in stocks master list for zNSE:r}u!❌ Quote fetch failed for %s: %scSsh|] }t|qSr)rrlower)r4tfrrr sz$fetch_market_data..dayr0r@)rr0r@r~)r')5minute15minute30minuted)rbru timeframe max_candlesu%❌ Candle fetch failed for %s %s: %su*❌ Indicator summary failed for %s %s: %s indicatorspivotsfibu1❌ Pivot/Fibonacci calculation failed for %s: %s)candles_by_timeframer strategiesu2❌ Strategy feature calculation failed for %s: %srcss|]}t|VqdSr")r)r4rIrrr r5@r6z$fetch_market_data..cs*i|]}|tdp i|pgqS)r~)rr8)r4kdatarr Cs*z%fetch_market_data..z:[teGPT] Zerodha OK | %s | quote=%s | candles=%s | total=%dyesnou!🎉 ZERODHA DATA COMPLETE for %sz?No market data available from Zerodha (quote and candles empty)u'❌ %s for %s - likely auth/token issuezMarket data fetch failed for %s)(rutcnowr?rkdebugrrvr]rrr7r8 get_quoterrrrrSr get_intraday_candlesr= _PANDAS_OKitemspd DataFramer summarize_dataframecalculate_pivot_pointscalculate_fibonacci_levelsr summarize_strategiesrlistrsumvalueskeysinforrl)rbrpr{r^r|rxtokenr quote_datarnday_candles_dbwantrrindicators_summary tf_candlesdf_tfsummary day_candlesdf_dayr fib_levelsrmax_keepseries total_candlesrrhmsgrrr fetch_market_datas<  *                     (   r)NT),loggingrrrtypingrrrrpandasrrrfastapir app.v1.services.zerodha.clientr "app.v1.services.zerodha.indicatorsr )app.v1.services.zerodha.strategy_featuresr app.v1.services.intraday_storer app.v1.services.stock_historyrapp.v1.services.stocks_masterrconfigrr getLogger__name__rkr!r$r%rrSraror=rzr#rrrrr sL         . I