o ͿSi&@srddlmZddlmZmZmZddlZz ddlZddl Z dZ Wne y/dZdZ dZ YnwGdddZ dS)) annotations)DictTupleListNTFc@seZdZed8ddZed9d d Zed:d9d d Ze   d;dddZ ed8dd Z ed?d@d#d$Z edAdBd)d*Z edCd.d/Z edDdEd2d3Zed?dFd6d7Zd!S)GIndicatorCalculatordf pd.DataFramereturn pd.SeriescCsrtsdSd|vs|ddkrtjtj|jdS|d|d|dd}||d}|d}||S)Nvolumerindexhighlowclose) _PANDAS_OKsumpdSeriesnpnanr cumsum)r typical_pricecumulative_tp_volumecumulative_volumer@/var/www/html/Trade-python/app/v1/services/zerodha/indicators.pycalculate_vwaps z"IndicatorCalculator.calculate_vwapperiodintcCstsdS|dj|ddS)NrFspanadjustrewmmean)rrrrr calculate_emasz!IndicatorCalculator.calculate_emac Cs~tsdS|d}||dkd}||dkd }|jd|dd}|jd|dd}||}ddd|}|S)NrrF)alphar#d)rdiffwherer%r&) rrdeltagainlossavg_gainavg_lossrsrsirrr calculate_rsi!s z!IndicatorCalculator.calculate_rsi  fastslowsignal&Tuple[pd.Series, pd.Series, pd.Series]c Cs`tsdS|dj|dd}|dj|dd}||}|j|dd}||}|||fS)N)NNNrFr!r$) rr9r:r;ema_fastema_slow macd_line signal_line histogramrrrcalculate_macd1s z"IndicatorCalculator.calculate_macdDict[str, float]c CstsiSt|dkr iS|jd}|d|d|dd}d||d}d||d}||d|d}||d|d}|d|d|d}|d|d|d}|||||||dS) z;Calculate Standard Pivot Points (R3, R2, R1, P, S1, S2, S3)r)rrrr)R1R2R3PS1S2S3)rleniloc) rprevpr1s1r2s2r3s3rrrcalculate_pivot_pointsAs  z*IndicatorCalculator.calculate_pivot_pointsperiodslistDict[str, pd.Series]cCs0tsiSi}|D] }t|||d|<q|S)Nema_)rrr')rrXemasrrrrcalculate_ema_stackSs z'IndicatorCalculator.calculate_ema_stackcCstsdS|dS)Nr )rr,)rrrrcalculate_volume_delta\s z*IndicatorCalculator.calculate_volume_deltaN ema_periodscCsts|S|dur gd}t||d<t||d<t|\|d<|d<|d<t||}|D]\}}|||<q2t||d<|S) z)Add all technical indicators to DataFrameNr87r+vwapr4macd macd_signal macd_hist volume_delta)rrrr5rBr]itemsr^)rr_ ema_stacknameseriesrrradd_all_indicatorsbs  z&IndicatorCalculator.add_all_indicators @ multiplierfloatc Cs:tsdS|d|dd}|d|ddd|}|||}|||}tj|jd}tjd|jd}tdt |D]U} |d j | d|j | dkrwt |j | |j | d|j | <|d j | |j | krvd |j | <qCt |j | |j | d|j | <|d j | |j | krd|j | <qC||fS) z Calculate Supertrend indicator Returns: SuperTrend values and direction (1 for uptrend, -1 for downtrend) )NNrrrEcSs||SNr)xyrrrsz:IndicatorCalculator.calculate_supertrend..r r)rrD) rcombineabsrollingr&rrr rangerMrNminmax) rrrshl2atr upper_band lower_band supertrend directionirrrcalculate_supertrend}s&&       z(IndicatorCalculator.calculate_supertrendtimeframe_dataDict[str, pd.DataFrame]Dict[str, Dict[str, float]]cCsXtsiSi}|D]\}}|dus|jrq |dd}|dvr)t|||<q |S)N )30min30minutedayweekmonth)rrlemptylowerreplacerrW)rpivotstfrtf_normrrr calculate_multi_timeframe_pivotssz4IndicatorCalculator.calculate_multi_timeframe_pivotsr+lookbackcCstr |dus |jr iS||}|jriSt|d}t|d}||kr+iS||}||d||d||d||d||d||d }|d ||d|d }||||d S) a Calculate simple Fibonacci retracement and extension levels. Uses the highest high and lowest low over the last `lookback` bars (or all available if fewer) to define a swing, then computes standard retracement and extension levels. Nrrgh|?5?g r?g?g-?gx&?)0z23.6z38.250z61.8z78.6100g rh?)z127.2z161.8) swing_high swing_low retracements extensions)rrtailrtr~r})rrwindowrrr,rrrrrcalculate_fibonacci_levelss4        z.IndicatorCalculator.calculate_fibonacci_levels List[int]Dict[str, any]c Cstr |dus |jr iS|durgd}tj||d}|jd}dd}i}|D]}d|}||jvr@||||t|<q)||d||d ||d ||d ||d ||d d|jvrq||dnd||d|d }|S)aReturn a compact summary of key indicators for the latest bar. Expects a DataFrame with at least columns: high, low, close and ideally volume. This is intended to be used by higher-level services (like teGPT) to build structured numeric features for GPT prompts or internal scoring, without bloating the payload with full indicator series. Nr`)r_rDcSsLz|dus t|tr WdSt|}t|r|WSdWSty%YdSwru) isinstanceboolrtmathisfinite Exception)vfrrr _finite_floats z>IndicatorCalculator.summarize_dataframe.._finite_floatr[rrgr4rhrirjr rk) rrgr4rhrirjr rkema) rrrrpcopyrNcolumnsgetstr) rr_df_indlastr ema_valuesrcolsummaryrrrsummarize_dataframes0             z'IndicatorCalculator.summarize_dataframe)rrr r )rrrr r r )r()r6r7r8) rrr9r r:r r;r r r<)rrr rC)rrrXrYr rZru)rrr_rYr r)rqrr)rrrr rsrtr r )rrr r)r+)rrrr r r)rrr_rr r)__name__ __module__ __qualname__ staticmethodrr'r5rBrWr]r^rprrrrrrrrrs8      /r) __future__rtypingrrrrpandasrnumpyrrrrrrrrs