Q-jWddlmZddlmZmZddlmZmZmZddl m Z m Z m Z ddl mZmZddlmZmZmZmZmZmZmZddlmZmZdd lmZ dd ed ed ededededededededededededefdZy ))nan) DataFrameSeries)DictLikeIntIntFloat)emalinregsma) decreasing increasing)simplify_columnsunsigned_differencesv_boolv_mamodev_offset v_pos_defaultv_series)bbandskc)momNhighlowclose bb_lengthbb_std kc_length kc_scalar mom_length mom_smoothuse_trmamodeprenanoffsetkwargsreturnc t|d}t|d}t|d}t|d}t||||dz}t||}t||}t||}|||yt|d}t|d}t| d} t | d } t | } | j d d } | j d d }| j d d }| j dd }t|||| }t|||||| | }t||_ t||_ |rg|j|j}|j|j}dd||zz|jzz}t||z |}n;t||}| j!dk(rt#||}n t%||}|j&|j&kD|j(|j(kz}|j&|j&k|j(|j(kDz}||z}| dk7rD|j+| }|j+| }|j+| }|j+| }d| vrX|j-| dd |j-| dd |j-| dd |j-| dd | rdnd}|d|d|d|d|z }||rdndz }d||_|r?|j1t2}|j1t2}|j1t2}| r*t||dz }t4|d|t4|d|t4|d||j.|d|d|d|i}t7||j8 }|j.|_d!x|_|_|r||dk\} ||dk}!t=| d "\}"}#t=|!d "\}$}%|"|z}"|#|z}#|%|z}%|$|z}$|"j?dt4d |#j?dt4d |%j?dt4d |$j?dt4d |tA|z}&|tC|z}'|&j?dt4d |'j?dt4d d| vr|&j-| dd |'j-| dd |"j-| dd |#j-| dd |%j-| dd |$j-| dd |&|d#<|'|d$<|"|d%<|#|d&<|%|d'<|$|d(<|S))aSqueeze This indicator, based on John Carter's "TTM Squeeze" indicator, attempts identify momentum using volatility. Sources: * "Mastering the Trade" (chapter 11), John Carter * [thinkorswim](https://tlc.thinkorswim.com/center/reference/Tech-Indicators/studies-library/T-U/TTM-Squeeze) * [tradestation](https://tradestation.tradingappstore.com/products/TTMSqueeze) * [tradingview](https://www.tradingview.com/scripts/lazybear/) Parameters: high (Series): ```high``` Series low (Series): ```low``` Series close (Series): ```close``` Series bb_length (int): BB period. Default: ```20``` bb_std (float): BB Std. Dev. Default: ```2``` kc_length (int): KC period. Default: ```20``` kc_scalar (float): KC scalar. Default: ```1.5``` mom_length (int): Momentum Period. Default: ```12``` mom_smooth (int): Momentum Smoothing period. Default: ```6``` mamode (str): One of: "ema" or "sma". Default: ```"sma"``` prenan (bool): Apply prenans. Default: ```False``` offset (int): Post shift. Default: ```0``` Other Parameters: tr (value): Use True Range for Keltner Channels. Default: ```True``` asint (bool): Returns as ```Int```. Default: ```True``` lazybear (value): LazyBear's TradingView. Default: ```False``` detailed (value): Extra detailed. Default: ```False``` fillna (value): ```pd.DataFrame.fillna(value)``` Returns: (DataFrame): * Default: 4 columns * Detailed: 10 columns Note: Volatility * Increasing: ```kc``` and ```bbands``` difference increases * Decreasing: ```kc``` and ```bbands``` difference decreases  rNg@g?r FtrTasintdetailedlazybear)lengthstdr#)r0scalarr#r,g?)r0r rfillna)inplacehlr__LBSQZSQZ_ONSQZ_OFFSQZ_NO)indexmomentum)r-SQZ_INCSQZ_DECSQZ_PINCSQZ_PDECSQZ_NDECSQZ_NINC)"rmaxrrrrpoprrrcolumnsrollingminbr rlowerr r lushiftr3nameastypeintrrr>categoryrreplacer r )(rrrrrrrr r!r"r#r$r%r&_lengthr-r.r/bbdkch highest_high lowest_lowavg_squeezemomo squeeze_on squeeze_off no_squeeze_props nanlengthdatadf pos_squeeze neg_squeezepos_incpos_decneg_incneg_decsqz_incsqz_decs( h/Users/jameslopez/projects/TradingBot25/.venv/lib/python3.12/site-packages/pandas_ta/momentum/squeeze.pyr[r[sfi,Ii,Iz2.Jz1-J)Y J?!CG D' "D 3 C UG $E |s{em 63 'Fi-I fe $F FE "F f F ZZd #F JJw %Ezz*e,Hzz*e,H yfV DC  c59& C #3'CK"3'CK||I.224 [[+//1 c\J67#%%?@i85, <<>U "$z2G$z2G%%#%%-CEECEEM2J55355=SUUSUU]3K |+J{--'%%f- !''/ %%f- 6vh'6&*D96(+T:&*D9RuF )AfXQyk9+>>F xeR'F>GL &&s+ !((- &&s+   9-1 !$ :I"% JY!$ :I  g : K :  D 4u{{ +BllBG%//BK'"gl+ gk* / 4H/ 4H77773-3-3-3-Jw//Jw//3-3- v  NN6(+TN : NN6(+TN : NN6(+TN : NN6(+TN : NN6(+TN : NN6(+TN : W W!X!X!X!X I) NNNNNNNNNN)numpyrpandasrrpandas_ta._typingrrrpandas_ta.overlapr r r pandas_ta.trendr r pandas_ta.utilsrrrrrrrpandas_ta.volatilityrrrboolstrr[rmrlrxs$55..2, /315.2'+x xx&,xx#+xx'/x x), x  x "% x  x x#+xxrm