mirror of
https://github.com/wassname/pandas-ta.git
synced 2026-07-24 13:10:26 +08:00
122 lines
4.0 KiB
Python
122 lines
4.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
from importlib.util import find_spec
|
|
from pathlib import Path
|
|
from pkg_resources import get_distribution, DistributionNotFound
|
|
|
|
from pandas_ta._typing import Dict, IntFloat, ListStr
|
|
|
|
|
|
_dist = get_distribution("pandas_ta")
|
|
try:
|
|
# Normalize case for Windows systems
|
|
_here = Path(_dist.location) / __file__
|
|
if not _here.exists():
|
|
# not installed, but there is another version that *is*
|
|
raise DistributionNotFound
|
|
except DistributionNotFound:
|
|
__version__ = "Please install this project with setup.py"
|
|
|
|
version = __version__ = _dist.version
|
|
|
|
Imports: Dict[str, bool] = {
|
|
"alphaVantage-api": find_spec("alphaVantageAPI") is not None,
|
|
"dotenv": find_spec("dotenv") is not None,
|
|
"matplotlib": find_spec("matplotlib") is not None,
|
|
"mplfinance": find_spec("mplfinance") is not None,
|
|
"numba": find_spec("numba") is not None,
|
|
"yaml": find_spec("yaml") is not None,
|
|
"scipy": find_spec("scipy") is not None,
|
|
"sklearn": find_spec("sklearn") is not None,
|
|
"statsmodels": find_spec("statsmodels") is not None,
|
|
"stochastic": find_spec("stochastic") is not None,
|
|
"talib": find_spec("talib") is not None,
|
|
"tqdm": find_spec("tqdm") is not None,
|
|
"vectorbt": find_spec("vectorbt") is not None,
|
|
"yfinance": find_spec("yfinance") is not None,
|
|
"polygon": find_spec("polygon") is not None,
|
|
}
|
|
|
|
# Not ideal and not dynamic but it works.
|
|
# Will find a dynamic solution later.
|
|
Category: Dict[str, ListStr] = {
|
|
# Candles
|
|
"candles": [
|
|
"cdl_pattern", "cdl_z", "ha"
|
|
],
|
|
# Cycles
|
|
"cycles": ["ebsw", "reflex"],
|
|
# Momentum
|
|
"momentum": [
|
|
"ao", "apo", "bias", "bop", "brar", "cci", "cfo", "cg", "cmo",
|
|
"coppock", "cti", "er", "eri", "fisher", "inertia", "kdj", "kst",
|
|
"macd", "mom", "pgo", "ppo", "psl", "qqe", "roc", "rsi", "rsx",
|
|
"rvgi", "slope", "smi", "squeeze", "squeeze_pro", "stc", "stoch",
|
|
"stochf", "stochrsi", "td_seq", "trix", "tsi", "uo", "willr"
|
|
],
|
|
# Overlap
|
|
"overlap": [
|
|
"alligator", "alma", "dema", "ema", "fwma", "hilo", "hl2", "hlc3",
|
|
"hma", "hwma", "ichimoku", "jma", "kama", "linreg", "mama",
|
|
"mcgd", "midpoint", "midprice", "ohlc4", "pwma", "rma", "sinwma",
|
|
"sma", "smma", "ssf", "ssf3", "supertrend", "swma", "t3", "tema",
|
|
"trima", "vidya", "wcp", "wma", "zlma"
|
|
],
|
|
# Performance
|
|
"performance": ["log_return", "percent_return"],
|
|
# Statistics
|
|
"statistics": [
|
|
"entropy", "kurtosis", "mad", "median", "quantile", "skew", "stdev",
|
|
"tos_stdevall", "variance", "zscore"
|
|
],
|
|
# Transform
|
|
"transform": ["cube", "ifisher", "remap"],
|
|
# Trend
|
|
"trend": [
|
|
"adx", "amat", "aroon", "chop", "cksp", "decay", "decreasing", "dpo",
|
|
"increasing", "long_run", "psar", "qstick", "short_run", "trendflex",
|
|
"tsignals", "ttm_trend", "vhf", "vortex", "xsignals"
|
|
],
|
|
# Volatility
|
|
"volatility": [
|
|
"aberration", "accbands", "atr", "atrts", "bbands", "donchian",
|
|
"hwc", "kc", "massi", "natr", "pdist", "rvi", "thermo",
|
|
"true_range", "ui"
|
|
],
|
|
|
|
# Volume.
|
|
# Note: "vp" or "Volume Profile" is excluded since it does not return a Time Series
|
|
"volume": [
|
|
"ad", "adosc", "aobv", "cmf", "efi", "eom", "kvo", "mfi", "nvi",
|
|
"obv", "pvi", "pvo", "pvol", "pvr", "pvt", "vwap", "vwma", "wb_tsv"
|
|
],
|
|
}
|
|
|
|
CANDLE_AGG: Dict[str, str] = {
|
|
"open": "first",
|
|
"high": "max",
|
|
"low": "min",
|
|
"close": "last",
|
|
"volume": "sum"
|
|
}
|
|
|
|
# https://www.worldtimezone.com/markets24.php
|
|
EXCHANGE_TZ: Dict[str, IntFloat] = {
|
|
"NZSX": 12, "ASX": 11,
|
|
"TSE": 9, "HKE": 8, "SSE": 8, "SGX": 8,
|
|
"NSE": 5.5, "DIFX": 4, "RTS": 3,
|
|
"JSE": 2, "FWB": 1, "LSE": 1,
|
|
"BMF": -2, "NYSE": -4, "TSX": -4,
|
|
"GENR": 0 # Generated Data
|
|
}
|
|
|
|
RATE: Dict[str, IntFloat] = {
|
|
"DAYS_PER_MONTH": 21,
|
|
"MINUTES_PER_HOUR": 60,
|
|
"MONTHS_PER_YEAR": 12,
|
|
"QUARTERS_PER_YEAR": 4,
|
|
"TRADING_DAYS_PER_YEAR": 252, # Keep even
|
|
"TRADING_HOURS_PER_DAY": 6.5,
|
|
"WEEKS_PER_YEAR": 52,
|
|
"YEARLY": 1,
|
|
}
|