python source code of tsi

jesse-master
- .github
  - ISSUE_TEMPLATE.md
- jesse
  - services
    - statistics.py
    - tradingview.py
    - charts.py
    - db.py
    - cache.py
    - project_maker
      - project_template
        storage
        __init__.py
        config.py
        strategies
        ExampleStrategy
        __init__.py
        __init__.py
        routes.py
        __init__.py
        .gitignore
      - __init__.py
    - selectors.py
    - candle.py
    - report.py
    - table.py
    - strategy_maker
      - ExampleStrategy
        __init__.py
      - __init__.py
    - logger.py
    - api.py
    - notifier.py
    - broker.py
    - file.py
    - __init__.py
    - validators.py
    - required_candles.py
  - libs
    - dynamic_numpy_array
      - __init__.py
    - __init__.py
  - config.py
  - enums
    - __init__.py
  - strategies
    - Test46
      - __init__.py
    - Test19
      - __init__.py
    - Test48
      - __init__.py
    - Test11
      - __init__.py
    - Test44
      - __init__.py
    - Test27
      - __init__.py
    - Strategy.py
    - Test06
      - __init__.py
    - Test42
      - __init__.py
    - Test21
      - __init__.py
    - Test38
      - __init__.py
    - Test14
      - __init__.py
    - Test20
      - __init__.py
    - Test41
      - __init__.py
    - Test05
      - __init__.py
    - Test16
      - __init__.py
    - Test26
      - __init__.py
    - Test33
      - __init__.py
    - Test32
      - __init__.py
    - Test36
      - __init__.py
    - Test12
      - __init__.py
    - Test39
      - __init__.py
    - Test37
      - __init__.py
    - Test10
      - __init__.py
    - Test40
      - __init__.py
    - Test43
      - __init__.py
    - Test45
      - __init__.py
    - Test28
      - __init__.py
    - Test34
      - __init__.py
    - Test25
      - __init__.py
    - Test08
      - __init__.py
    - Test31
      - __init__.py
    - Test02
      - __init__.py
    - __init__.py
    - Test35
      - __init__.py
    - Test29
      - __init__.py
    - Test17
      - __init__.py
    - Test18
      - __init__.py
    - Test13
      - __init__.py
    - Test23
      - __init__.py
    - Test03
      - __init__.py
    - Test47
      - __init__.py
    - Test24
      - __init__.py
    - Test22
      - __init__.py
    - Test30
      - __init__.py
    - Test01
      - __init__.py
    - Test09
      - __init__.py
    - Test07
      - __init__.py
    - Test15
      - __init__.py
    - Test04
      - __init__.py
  - models
    - Candle.py
    - Trade.py
    - Position.py
    - Ticker.py
    - Orderbook.py
    - Order.py
    - __init__.py
    - utils.py
    - Exchange.py
    - CompletedTrade.py
    - Route.py
  - exchanges
    - exchange.py
    - __init__.py
    - sandbox
      - Sandbox.py
      - __init__.py
  - store
    - state_trades.py
    - state_positions.py
    - state_orders.py
    - state_orderbook.py
    - __init__.py
    - state_logs.py
    - state_tickers.py
    - state_completed_trades.py
    - state_exchanges.py
    - state_candles.py
    - state_app.py
  - indicators
    - sma.py
    - adosc.py
    - decycler.py
    - rocr.py
    - bollinger_bands_width.py
    - gauss.py
    - ht_dcperiod.py
    - beta.py
    - wclprice.py
    - wilders.py
    - pivot.py
    - stochastic.py
    - ichimoku_cloud.py
    - stochf.py
    - var.py
    - trange.py
    - ht_sine.py
    - rocp.py
    - srsi.py
    - wma.py
    - macd.py
    - tsi.py
    - dec_osc.py
    - obv.py
    - dm.py
    - bop.py
    - zscore.py
    - pattern_recognition.py
    - apo.py
    - dx.py
    - avgprice.py
    - ht_phasor.py
    - sarext.py
    - rocr100.py
    - keltner.py
    - stddev.py
    - aroonosc.py
    - macdext.py
    - natr.py
    - hma.py
    - t3.py
    - pvi.py
    - ppo.py
    - vwmacd.py
    - fisher.py
    - frama.py
    - di.py
    - linearreg_intercept.py
    - smma.py
    - mama.py
    - sar.py
    - roc.py
    - marketfi.py
    - emv.py
    - ema.py
    - fosc.py
    - emd.py
    - vwma.py
    - minmax.py
    - medprice.py
    - linearreg.py
    - acosc.py
    - adx.py
    - supersmoother.py
    - midprice.py
    - vosc.py
    - donchian.py
    - vidya.py
    - kama.py
    - qstick.py
    - linearreg_angle.py
    - cci.py
    - bollinger_bands.py
    - aroon.py
    - midpoint.py
    - ht_dcphase.py
    - lrsi.py
    - cmo.py
    - __init__.py
    - linearreg_slope.py
    - correl.py
    - rsi.py
    - dpo.py
    - trix.py
    - alligator.py
    - atr.py
    - adxr.py
    - ultosc.py
    - trima.py
    - zlema.py
    - nvi.py
    - mfi.py
    - tsf.py
    - kvo.py
    - typprice.py
    - gatorosc.py
    - dema.py
    - wad.py
    - ht_trendline.py
    - mass.py
    - willr.py
    - tema.py
    - msw.py
    - supertrend.py
    - mom.py
    - ao.py
    - ad.py
    - itrend.py
    - ht_trendmode.py
    - cvi.py
  - modes
    - routes_mode
      - __init__.py
    - optimize_mode
      - __init__.py
      - Genetics.py
    - __init__.py
    - backtest_mode
      - __init__.py
    - import_candles_mode
      - drivers
        testnet_binance_futures.py
        coinbase.py
        bitfinex.py
        binance_futures.py
        __init__.py
        binance.py
        interface.py
      - __init__.py
  - __init__.py
  - utils.py
  - exceptions
    - __init__.py
  - research
    - __init__.py
    - get_candles.py
  - factories
    - candle_factory.py
    - order_factory.py
    - __init__.py
  - routes
    - __init__.py
  - helpers.py
- LICENSE
- setup.py
- .travis.yml
- README.md
- tests
  - test_indicators.py
  - test_state_orderbook.py
  - test_state_candle.py
  - test_state_orders.py
  - test_state_exchanges.py
  - test_import_candles.py
  - test_state_trades.py
  - test_backtest.py
  - test_candle_service.py
  - test_position.py
  - test_strategy_utils.py
  - test_state_logs.py
  - test_conflicting_orders.py
  - test_helpers.py
  - test_exchange.py
  - test_state_ticker.py
  - test_broker.py
  - data
    - test_candles_0.py
    - __init__.py
    - test_candles_1.py
  - __init__.py
  - test_order.py
  - test_metrics.py
  - test_parent_strategy.py
  - test_router.py
  - test_trade.py
- requirements.txt
- .gitignore
- assets

from typing import Union

import numpy as np
import talib

from jesse.helpers import get_candle_source


def tsi(candles: np.ndarray, long_period=25, short_period=13, source_type="close", sequential=False) -> Union[
    float, np.ndarray]:
    """
     True strength index (TSI)

    :param candles: np.ndarray
    :param long_period: int - default: 25
    :param short_period: int - default: 13
    :param source_type: str - default: "close"
    :param sequential: bool - default=False

    :return: float | np.ndarray
    """
    if not sequential and len(candles) > 240:
        candles = candles[-240:]

    source = get_candle_source(candles, source_type=source_type)
    r = 100 * (talib.EMA((talib.EMA(talib.MOM(source, 1), long_period)), short_period)) / (
        talib.EMA((talib.EMA(np.absolute(talib.MOM(source, 1)), long_period)), short_period))

    return r if sequential else r[-1]