o ̿SiD@s"UddlmZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z mZmZmZmZddlmZddlmZeeZed Zgd Zd ed <ed dp]dZeeddZ eeddZ!eeddZ"eeddZ#eeddZ$e%edddZ&e%eddZ'e%eddZ(e%ed d!Z)e%ed"d#dZ*ed$d%Gd&d'd'Z+drd,d-Z,dsd/d0Z-dtd4d5Z.dud:d;Z/dvd=d>Z0dwd?d@Z1dxdAdBZ2dydDdEZ3dzdIdJZ4d{dNdOZ5d|dWdXZ6d}d\d]Z7d~d`daZ8ddgdhZ9ddidjddkddpdqZ:dS)) annotationsN) dataclass)datetime timedelta)median)AnyDictIterableListOptionalTuple)ZoneInfo) ZerodhaClientz Asia/Kolkata)NIFTY 50z NIFTY BANKzNIFTY PSU BANKzNIFTY FIN SERVICEzNIFTY ITz NIFTY AUTOz NIFTY PHARMAzNIFTY HEALTHCARE INDEXz NIFTY METALz NIFTY ENERGYzNIFTY OIL & GASz NIFTY REALTYz NIFTY INFRAz NIFTY FMCGzNIFTY CONSUMER DURABLESzNIFTY CAPITAL GOODSz List[str]DEFAULT_INDEX_QUERIESMI_PRIMARY_INDEXr$MARKET_INTELLIGENCE_INTERVAL_SECONDS900 MI_EMA_PERIOD20 MI_ATR_PERIOD14MI_STRUCTURE_LOOKBACK6MI_TREND_LOOKBACK12MI_MIN_TREND_PCTz0.35Y@MI_SLOPE_ATR_FRACTIONz0.08MI_VOL_LOW_RATIOz0.85MI_VOL_HIGH_RATIOz1.25MI_SECTOR_MIN_MOVE_PCTz0.30T)frozenc@s>eZdZUded<ded<ded<ded<ded<ded<d S) CandlertsfloatohlcvN)__name__ __module__ __qualname____annotations__r/r/A/var/www/html/Trade-python/app/v1/services/market_intelligence.pyr#=s  r#r*rreturnOptional[float]cCsVz |dus t|tr WdSt|}t|st|rWdS|WSty*YdSwN) isinstanceboolr%mathisnanisinf Exception)r*fr/r/r0 _safe_floatGs r;Optional[datetime]cCsNt|tr|St|tr%|r%z t|ddWSty$YdSwdS)NZz+00:00)r4rstrstrip fromisoformatreplacer9)r*r/r/r0 _to_datetimeSs  rBrawIterable[Dict[str, Any]] List[Candle]c Csg}|pgD]W}t|tsqt|d}|sqt|d}t|d}t|d}t|d}t|d}d||||fvrDq|t|t|t|t|t|t|pXddq|jd d d |S) Ndateopenhighlowclosevolume)r$r&r'r(r)r*cSs|jSr3)r$)xr/r/r0osz$_normalize_candles..)key) r4dictrBgetr;appendr#r%sort) rCoutrowr$r&r'r(r)r*r/r/r0_normalize_candles_s"  4rVvalues List[float]periodintcCsl|sgStdt|}d|d}g}|d}|||ddD]}|||d|}||q"|S)Ng@?r)maxrZrR)rWrYkemaprevrMr/r/r0_emass   racandlescCspt|dkrgSg}|dj}|ddD] }t|j|jt|j|t|j|}|t||j}q|S)Nrr[)lenr)r]r'r(absrRr%)rbrT prev_closer)trr/r/r0 _true_rangess  (rhcCs\t|}|sdStdt|}t||krtt|t|S|| d}tt||S)Nr[)rhr]rZrdr%sum)rbrYtrswindowr/r/r0_atrs rlcCsXd}d}|D]}|j|j|jd}|||j7}||j7}q|dkr&dSt||S)NrLg@r)r'r(r)r*r%)rbpvvvr)tpr/r/r0_vwaps  rplookbackcCstdt|}t||kr|| dn|}t|dkrdSd}tdt|D]"}||j||djkr9|d7}||j||djkrI|d7}q'|S)z>Higher-high + higher-low score in the last `lookback` candles.Nrr[)r]rZrdranger'r()rbrqrkscoreir/r/r0_structure_scores rvar%bcCs0|dus|dur dS|dkrdSt|||SNr)r%)rwrxr/r/r0 _pct_changes rzatr_pctatr_pct_historyTuple[str, Dict[str, Any]]cCs|dur dddifSdd|D}|rt|n|}|r!t||nd}|tkr/d|||dfS|tkr;d |||dfSd|||dfS) NModeratereason missing_atrcSs g|] }|dur|dkr|qSryr/).0rMr/r/r0 s z%_volatility_state..r\High)r{baselineratioLow)rr%VOL_HIGH_RATIO VOL_LOW_RATIO)r{r|histbaserr/r/r0_volatility_states rrJr_ema_prevvwapatrstructure_score trend_retc Cs|pd}t||}|dkrt|t|knt|dk}|dup#||k} |dup+||k} |duo3|tk} |duo<|t k} ||koN| oN|dkoN|oN|dkoN| } ||ko`| o`|dko`|o`|dko`| }| rodd||||||dfS|r}dd ||||||dfSd d ||||||dfS) NrLrrcTrendingup) directionrJr_ ema_sloperrrBearishdownSidewaysflat)r%reSLOPE_ATR_FRACTION MIN_TREND_PCT)rJr_rrrrrslopeslope_okabovebelow trend_ok_up trend_ok_dnbullbearr/r/r0 _market_biassL $$$  r market_biasr>volatility_statecCsD|pd}|p d}|dkrdS|dkr |dvr dSdS)NrHrrI>bearishtrendingrMedium)r?lower)rrvbmbr/r/r0 _overall_risksrquery Optional[int]c Cs|pd}|s dSz'|dd|ddddd d d i}|r0|d dur0t|d WSWn ty:Ynwz'|dd|ddddd d d i}|ra|d durat|d WSWn tykYnw|d d|dddddd d i}|r|d rzt|d WStyYdSw|d d|ddddd d d i}|r|d rzt|d WStyYdSw|d ddidd i}t|tr|dnd}t|trM|rM|}|D]]}t|tsqt |dpd}t |dp d} t |dpd} | r)| dkr)q||ks3| |krL|d } zt| WStyKYdSwqdS)zResolve an instrument token from cached Zerodha instruments. We attempt exact match on tradingsymbol (case-insensitive), and fall back to name. rNstocks^$ru)z$regexz$optionsNSE)symbolexchangeinstrument_tokenr[)namerzerodha_instruments) tradingsymbolrtype nse_equity instrumentsrrr) r?find_onerQrZr9r4rPlistupperr>) dbrqdoccachedarrq_upperinstr$nmextokr/r/r0_resolve_instrument_token s       rzerodha_clientrrinterval days_backcCs@t}|tttd|d}|jj||||ddd}t|S)z@Fetch candles from Zerodha (raw list), normalize to Candle list.r[)daysF)r from_dateto_dater continuousoi)rutcnowrrZr]kitehistorical_datarV)rrrrendstartrCr/r/r0_fetch_candlesZs r15minute) captured_atrr index_queriesrrOptional[List[str]]Dict[str, Any]c: sD|pt}|p t}i}g}|D]!}zt||} Wn ty$d} Ynw| r,| ||<q||qt|vr>tdtd|t} t|| ||d} t | t t t t dkrbtdtdt | dd | D} t| t } | d }t | d kr|| d n|}| d }t| t }|r|rt||d nd}g}tt d tt | dD]$}| d|d}t|t }|r|d jr|t||d jd qt||dd\}}|tfdd | D}t|p| dd}t| t}t | t kr| t djn| dj}t||}t|t|t|||||d\}}t||} g}!g}"g}#i}$|D]O\}} |tkr9q-z t|| ||d}%t |%t t t t dkrSWq-dd |%D}&t|&t }'|&d }(|'d })t |'d krr|'d n|)}*t|%t }+|+r|(rt|+|(d nd},g}-tt d tt |%dD]'}|%d|d}t|t }|r|d jr|-t||d jd qt|,|-dd\}.}/t |%t kr|%t djn|%dj}0t|(|0}1t|)|*}2d}3|.dkrd}3|#|n<|(|)ko |1duo |1tko |2dk}4|(|)ko|1duo|1t ko|2dk}5|4r*d}3|!|n |5r4d}3|"||3|1|(t|)|2d|.i|/d|$|<Wq-ty}}6z)t !ddpXd"#}7|7dv}8|8rlt$%d ||6nt$&d!|WYd}6~6q-d}6~6ww|t|||| t't(|!t't(|"t't(|#d"||d#|$|d$d%}9|9S)&zpCompute deterministic Market Intelligence Summary. No LLMs, no probabilistic logic. Purely rule-based. NzPrimary index 'z"' not found in zerodha_instruments)rrrrzNot enough candles for z: cSg|]}|jqSr/r)rr)r/r/r0rz7compute_market_intelligence_summary..rcrr[ics$g|]}|jtkr|qSr/)r$ astimezoneISTrFr today_istr/r0rs$ir)rJr_rrrrrcSrr/rrr/r/r0rrNEUTRALrVOLATILEBULLISHBEARISHstate)classretrJr_r volatility QUIET_LOGStrue)0falsenooffz#[MI] Failed computing sector %s: %sz[MI] Failed computing sector %s)bullishrvolatile)biasr)marketsectorsmissing_indices)r primary_indexrrr overall_risksector_strength diagnostics))rrrrr9rRPRIMARY_MARKET_INDEX RuntimeErrorrrdr] EMA_PERIOD ATR_PERIODTREND_LOOKBACKrarlr%rsminr)rrrrFrprvSTRUCTURE_LOOKBACKrzrritemsSECTOR_MIN_MOVE_PCTosgetenvr?rloggerwarning exceptionsortedset):rrrrrr token_by_namemissingrr nifty_tok nifty_candlescloses ema_seriesema_lastr close_lastrr{ atr_pct_histrusubrw vol_statevol_diagsessionr struct_scorerefrr bias_diagrrrr sectors_debugrbcloses_sema_sclose_s ema_last_s ema_prev_satr_s atr_pct_satr_pct_hist_ssector_vol_statesector_vol_diagref_sret_sslope_sclsrdnerCquietsummaryr/rr0#compute_market_intelligence_summaryss        (       (    &(        r7)r*rr1r2)r*rr1r<)rCrDr1rE)rWrXrYrZr1rX)rbrEr1rX)rbrErYrZr1r2)rbrEr1r2)rbrErqrZr1rZ)rwr%rxr%r1r2)r{r2r|rXr1r})rJr%r_r%rr%rr2rr2rrZrr2r1r})rr>rr>r1r>)rr>r1r) rrrrZrr>rrZr1rE) rrrr<rr>rrZrrr1r); __future__rloggingr6r  dataclassesrrr statisticsrtypingrrr r r r zoneinfor app.v1.services.zerodha.clientr getLoggerr+rrrr.rr?rrZMI_INTERVAL_SECONDSrrr rr%rrrrr r#r;rBrVrarhrlrprvrzrrrrrr7r/r/r/r0sZ            7 M