feat: func output struct

Author: Yvictor

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "polars_talib"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

python/polars_talib/__init__.py

@@ -190,6 +190,27 @@ def get_function_groups():
     return __function_groups__.copy()
 
 
+def get_functions_output_struct():
+    """
+    Returns a dict with keys of function names and keys of the output struct
+    """
+    return {
+        "ht_phasor": ["inphase", "quadrature"],
+        "ht_sine": ["sine", "leadsine"],
+        "minmax": ["min", "max"],
+        "minmaxindex": ["minidx", "maxidx"],
+        "aroon": ["aroondown", "aroonup"],
+        "macd": ["macd", "macdsignal", "macdhist"],
+        "macdext": ["macd", "macdsignal", "macdhist"],
+        "macdfix": ["macd", "macdsignal", "macdhist"],
+        "stoch": ["slowk", "slowd"],
+        "stochf": ["fastk", "fastd"],
+        "stochrsi": ["fastk", "fastd"],
+        "bbands": ["upperband", "middleband", "lowerband"],
+        "mama": ["mama", "fama"],
+    }
+
+
 @pl.api.register_expr_namespace("ta")
 class TAExpr:
     def __init__(self, expr: pl.Expr):