Qubx 0.2.75__cp311-cp311-manylinux_2_35_x86_64.whl → 0.2.76__cp311-cp311-manylinux_2_35_x86_64.whl

qubx/__init__.py

@@ -70,6 +70,10 @@ if runtime_env() in ["notebook", "shell"]:
         # process data manager
         __manager = None
 
+        @line_magic
+        def qubx(self, line: str):
+            self.qubx_setup("dark" + " " + line)
+
         @line_magic
         def qubxd(self, line: str):
             self.qubx_setup("dark" + " " + line)

qubx/_nb_magic.py

@@ -1,5 +1,5 @@
 """"
-Here stuff we want to have in every Jupyter notebook after calling %qube magic
+Here stuff we want to have in every Jupyter notebook after calling %qubx magic
 """
 
 import qubx
@@ -29,7 +29,7 @@ def np_fmt_reset():
 if runtime_env() in ["notebook", "shell"]:
 
     # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
-    # -- all imports below will appear in notebook after calling %%alphalab magic ---
+    # -- all imports below will appear in notebook after calling %%qubx magic ---
     # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
 
     # - - - - Common stuff - - - -
@@ -40,16 +40,52 @@ if runtime_env() in ["notebook", "shell"]:
 
     # - - - - TA stuff and indicators - - - -
     import qubx.pandaz.ta as pta
+    import qubx.ta.indicators as ta
 
     # - - - - Portfolio analysis - - - -
+    from qubx.core.metrics import (
+        tearsheet,
+        chart_signals,
+        get_symbol_pnls,
+        get_equity,
+        portfolio_metrics,
+        pnl,
+        drop_symbols,
+        pick_symbols,
+    )
+
+    # - - - - Data reading - - - -
+    from qubx.data.readers import (
+        CsvStorageDataReader,
+        MultiQdbConnector,
+        QuestDBConnector,
+        AsOhlcvSeries,
+        AsPandasFrame,
+        AsQuotes,
+        AsTimestampedRecords,
+        RestoreTicksFromOHLC,
+    )
+
     # - - - - Simulator stuff - - - -
-    # - - - - Learn stuff - - - -
+    from qubx.backtester.simulator import simulate
+    from qubx.backtester.optimization import variate
+
     # - - - - Charting stuff - - - -
     from matplotlib import pyplot as plt
     from qubx.utils.charting.mpl_helpers import fig, subplot, sbp, plot_trends, ohlc_plot
+    from qubx.utils.charting.lookinglass import LookingGlass
 
     # - - - - Utils - - - -
-    from qubx.pandaz.utils import scols, srows, ohlc_resample, continuous_periods, generate_equal_date_ranges
+    from qubx.pandaz.utils import (
+        scols,
+        srows,
+        ohlc_resample,
+        continuous_periods,
+        generate_equal_date_ranges,
+        drop_duplicated_indexes,
+        retain_columns_and_join,
+        rolling_forward_test_split,
+    )
 
     # - setup short numpy output format
     np_fmt_short()

qubx/core/helpers.py

@@ -357,4 +357,4 @@ def set_parameters_to_object(strategy: Any, **kwargs):
             _log_info += f"\n\tset <green>{k}</green> <- <red>{v_str}</red>"
 
     if _log_info:
-        logger.info(f"<yellow>{strategy.__class__.__name__}</yellow> new parameters:" + _log_info)
+        logger.debug(f"<yellow>{strategy.__class__.__name__}</yellow> new parameters:" + _log_info)

qubx/core/lookups.py

@@ -152,8 +152,8 @@ class InstrumentsLookup:
                     v["base"],
                     v["quote"],
                     v["settle"],
-                    min_tick=float(info["tickSize"]),
-                    min_size_step=float(v["precision"]["price"]),
+                    min_tick=float(v["precision"]["price"]),
+                    min_size_step=float(v["precision"]["amount"]),
                     min_size=v["precision"]["amount"],
                     futures_info=FuturesInfo(
                         contract_type=info["type"],

qubx/core/metrics.py

@@ -33,6 +33,9 @@ MINUTELY = HOURLY * 60
 HOURLY_FX = DAILY * 24
 MINUTELY_FX = HOURLY_FX * 60
 
+_D1 = pd.Timedelta("1D")
+_W1 = pd.Timedelta("1W")
+
 
 def absmaxdd(data: List | Tuple | pd.Series | np.ndarray) -> Tuple[float, int, int, int, pd.Series]:
     """
@@ -113,7 +116,7 @@ def max_drawdown_pct(returns):
     return np.nanmin((cumrets - max_return) / max_return)
 
 
-def portfolio_returns(portfolio_log: pd.DataFrame, method="pct", init_cash=0) -> pd.Series:
+def portfolio_returns(portfolio_log: pd.DataFrame, method="pct", init_cash: float = 0.0) -> pd.Series:
     """
     Calculates returns based on specified method.
 
@@ -323,7 +326,7 @@ def omega_ratio(returns, risk_free=0.0, required_return=0.0, periods=DAILY):
     return (numer / denom) if denom > 0.0 else np.nan
 
 
-def aggregate_returns(returns, convert_to):
+def aggregate_returns(returns: pd.Series, convert_to: str) -> pd.DataFrame | pd.Series:
     """
     Aggregates returns by specified time period
     :param returns: pd.Series or np.ndarray periodic returns of the strategy, noncumulative
@@ -583,17 +586,21 @@ def monthly_returns(
     return pd.concat((100 * r_month, acc_balance), axis=1, keys=["Returns", "Balance"])
 
 
-def portfolio_symbols(df: pd.DataFrame) -> List[str]:
+def portfolio_symbols(src: pd.DataFrame | TradingSessionResult) -> List[str]:
     """
     Get list of symbols from portfolio log
     """
+    df = src.portfolio_log if isinstance(src, TradingSessionResult) else src
     return list(df.columns[::5].str.split("_").str.get(0).values)
 
 
-def pnl(x: pd.DataFrame, c=1, cum=False, total=False, resample=None):
+def pnl(
+    src: pd.DataFrame | TradingSessionResult, c=1, cum=False, total=False, resample=None
+) -> pd.Series | pd.DataFrame:
     """
     Extract PnL from portfolio log
     """
+    x = src.portfolio_log if isinstance(src, TradingSessionResult) else src
     pl = x.filter(regex=".*_PnL").rename(lambda x: x.split("_")[0], axis=1)
     comms = x.filter(regex=".*_Commissions").rename(lambda x: x.split("_")[0], axis=1)
     r = pl - c * comms
@@ -603,18 +610,21 @@ def pnl(x: pd.DataFrame, c=1, cum=False, total=False, resample=None):
     return r.sum(axis=1) if total else r
 
 
-def drop_symbols(df: pd.DataFrame, *args, quoted="USDT"):
+def drop_symbols(src: pd.DataFrame | TradingSessionResult, *args, quoted="USDT") -> pd.DataFrame:
     """
     Drop symbols (is quoted currency) from portfolio log
     """
     s = "|".join([f"{a}{quoted}" if not a.endswith(quoted) else a for a in args])
+    df = src.portfolio_log if isinstance(src, TradingSessionResult) else src
     return df.filter(filter(lambda si: not re.match(f"^{s}_.*", si), df.columns))
 
 
-def pick_symbols(df: pd.DataFrame, *args, quoted="USDT"):
+def pick_symbols(src: pd.DataFrame | TradingSessionResult, *args, quoted="USDT") -> pd.DataFrame:
     """
     Select symbols (is quoted currency) from portfolio log
     """
+    df = src.portfolio_log if isinstance(src, TradingSessionResult) else src
+
     # - pick up from execution report
     if "instrument" in df.columns and "quantity" in df.columns:
         rx = "|".join([f"{a}.*" for a in args])
@@ -663,11 +673,19 @@ def portfolio_metrics(
         returns = returns[:end]
         returns_on_init_bp = returns_on_init_bp[:end]
 
-    # aggregate them to daily (if we have intraday portfolio)
+    # - aggregate returns to higher timeframe
     try:
-        if infer_series_frequency(returns) < pd.Timedelta("1D").to_timedelta64():
-            returns_daily = aggregate_returns(returns, "daily")
-            returns_on_init_bp = aggregate_returns(returns_on_init_bp, "daily")
+        _conversion = "daily"
+        match (_s_freq := infer_series_frequency(returns)):
+            case _ if _s_freq <= _D1.to_timedelta64():
+                _conversion = "daily"
+            case _ if _s_freq > _D1.to_timedelta64() and _s_freq <= _W1.to_timedelta64():
+                _conversion = "weekly"
+            case _:
+                _conversion = "monthly"
+
+        returns_daily = aggregate_returns(returns, _conversion)
+        returns_on_init_bp = aggregate_returns(returns_on_init_bp, _conversion)
     except:
         returns_daily = returns
 
@@ -792,7 +810,7 @@ def get_equity(
     sessions: TradingSessionResult | list[TradingSessionResult],
     account_transactions: bool = True,
     timeframe: str | None = None,
-) -> pd.DataFrame:
+) -> pd.DataFrame | pd.Series:
     if timeframe is None:
         timeframe = _estimate_timeframe(sessions)
 

qubx/data/__init__.py

@@ -0,0 +1,11 @@
+from .readers import (
+    DataReader,
+    CsvStorageDataReader,
+    MultiQdbConnector,
+    QuestDBConnector,
+    AsOhlcvSeries,
+    AsPandasFrame,
+    AsQuotes,
+    AsTimestampedRecords,
+    RestoreTicksFromOHLC,
+)

qubx/data/readers.py

@@ -903,7 +903,7 @@ class QuestDBConnector(DataReader):
         where = f"where symbol in ({', '.join(quoted_symbols)}) and timestamp >= '{start}' and timestamp < '{stop}'"
         table_name = QuestDBSqlCandlesBuilder().get_table_name(f"{exchange}:{symbols[0]}")
 
-        _rsmpl = f"sample by {timeframe}"
+        _rsmpl = f"sample by {QuestDBSqlCandlesBuilder._convert_time_delta_to_qdb_resample_format(timeframe)}"
 
         query = f"""
         select timestamp, 
@@ -934,7 +934,7 @@ class QuestDBConnector(DataReader):
             select timestamp, symbol, sum(quote_volume) as qvolume 
             from "{table_name}"
             where timestamp >= '{start}' and timestamp < '{stop}'
-            SAMPLE BY {timeframe}
+            SAMPLE BY {QuestDBSqlCandlesBuilder._convert_time_delta_to_qdb_resample_format(timeframe)}
         )
         select symbol, avg(qvolume) as quote_volume from sampled
         group by symbol

qubx/pandaz/__init__.py

@@ -1,4 +1,15 @@
 from .utils import (
-    srows, scols, continuous_periods, ohlc_resample, retain_columns_and_join, dict_to_frame, 
-    generate_equal_date_ranges, rolling_forward_test_split
-)
+    srows,
+    scols,
+    continuous_periods,
+    ohlc_resample,
+    retain_columns_and_join,
+    select_column_and_join,
+    dict_to_frame,
+    drop_duplicated_indexes,
+    process_duplicated_indexes,
+    generate_equal_date_ranges,
+    rolling_forward_test_split,
+    shift_series,
+    check_frame_columns,
+)

qubx/pandaz/utils.py

@@ -90,11 +90,11 @@ def rolling_forward_test_split(
             yield (np.array(range(i - training_period, i)), np.array(range(i, i + test_period)))
 
 
-def generate_equal_date_ranges(start: str | pd.Timestamp, end: str | pd.Timestamp, freq: int, units: str) -> Iterable:
+def generate_equal_date_ranges(start: str, end: str, freq, units):
     """
     Generator for date ranges:
 
-    for s,e in generate_equal_date_ranges('2019-01-01', '2022-05-17', 1, 'Y'):
+    for s,e in generate_ranges('2019-01-01', '2022-05-17', 1, 'Y'):
         print(s, e)
     ------------------
     2019-01-01 2019-12-31
@@ -103,9 +103,8 @@ def generate_equal_date_ranges(start: str | pd.Timestamp, end: str | pd.Timestam
     2022-01-01 2022-05-17
     """
     _as_f = lambda x: pd.Timestamp(x).strftime("%Y-%m-%d")
-
-    # - for case when end - start < freq it won't got into the loop
-    b = [start, start] if _as_f(start) > _as_f(end) else [start, end]
+    if pd.Timestamp(end) - pd.Timestamp(start) < freq * pd.Timedelta(f"1{units}"):
+        b = [start, end]
 
     for a, b in rolling_forward_test_split(pd.Series(0, pd.date_range(start, end)), freq, freq, units=units):
         yield _as_f(a[0]), _as_f(a[-1])

qubx/utils/__init__.py

@@ -1,4 +1,4 @@
 from .time import convert_seconds_to_str, convert_tf_str_td64, floor_t64, time_to_str, infer_series_frequency
 from .misc import runtime_env, version, Struct, Stopwatch
-from .charting.mpl_helpers import set_mpl_theme, fig, sbp, vline, hline, ellips, set_mpl_theme, ohlc_plot
+from .charting.mpl_helpers import fig, sbp, vline, hline, ellips, set_mpl_theme, ohlc_plot, plot_trends
 from .charting.lookinglass import LookingGlass

{qubx-0.2.75.dist-info → qubx-0.2.76.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: Qubx
-Version: 0.2.75
+Version: 0.2.76
 Summary: Qubx - quantitative trading framework
 Home-page: https://github.com/dmarienko/Qubx
 Author: Dmitry Marienko

{qubx-0.2.75.dist-info → qubx-0.2.76.dist-info}/RECORD

@@ -1,5 +1,5 @@
-qubx/__init__.py,sha256=dIHcfXv3uDX-kJ6qRGrH5OUW7evnGFf6A2VsLQlr2jc,6047
-qubx/_nb_magic.py,sha256=vfV892I5rkA_Zbno5Dsl9osPTGiXTCV3afZRDrTlcOQ,1915
+qubx/__init__.py,sha256=4WwQDq3Se_Vi3exTyYFnEAz1QuB-0aQWIexMbg0STgY,6152
+qubx/_nb_magic.py,sha256=_acaAfFSAYZNm2I3iuSyrl8e7_gp26VnP2u6hx25HXg,2729
 qubx/backtester/__init__.py,sha256=g9K4pp4ieY4rRKE0eA_02LMo-c4vnk-QqtVn_ff9F5U,66
 qubx/backtester/ome.py,sha256=FmtRMDCeEo8ZZP-yLswLilX6zx6SjJeieMiaLS7zrn0,11145
 qubx/backtester/optimization.py,sha256=ATHoiEGwbDIk0-lbFIxvVDmH_f-vtS10YV41pfOTCBA,7297
@@ -10,20 +10,21 @@ qubx/core/account.py,sha256=uIU-EUMuH5owWhHTmU1pZ0WrkQkiv9YiFqSZW9dFDK8,9066
 qubx/core/basics.py,sha256=WA6LrmCG1si6pfyvezLRkfts575912AhGJMVbwBWSa0,21802
 qubx/core/context.py,sha256=WnHsItJ1OQ2jD0cHbYOMQ9ck4txLTwtf8Dp-zCLsUww,37759
 qubx/core/exceptions.py,sha256=W1gG_0fE3o2EMGUfOeOl3vVJDp8Z1iHLv4iZ0ThNkTs,306
-qubx/core/helpers.py,sha256=R-xi4lgmYvWpUARIYcbcrNrRlEap5wc-GQ5sE4riUxQ,12663
+qubx/core/helpers.py,sha256=V_3505UiQhQcNwKVFN3S7hQAgmfU0Lly3JX1kk3GW8E,12664
 qubx/core/loggers.py,sha256=zpehm2-RdKG1ISe6t3DiM3RrL_zzGni3YFcxfgDkV24,16575
-qubx/core/lookups.py,sha256=qGOSb2SIxmgVPGJFLJnPDI1P9hCzFMVipRDcVLUm8qk,14599
-qubx/core/metrics.py,sha256=pL1obxnk8I7bDF41bQcOZ7MDsq4Wmj6g5cRiV7cFZKQ,38657
-qubx/core/series.cpython-311-x86_64-linux-gnu.so,sha256=HwGNM0szT5qwZX6C2RS5W4kRqMtJhVpckokYeHnWMmM,779016
+qubx/core/lookups.py,sha256=2TH9kySRnUMTm_1uMVqk-FisyofWT22u7KGLicWzlhU,14607
+qubx/core/metrics.py,sha256=Y33YgwyGbbZ7KGAcmcQ1mNi8Rg_D76b3704Q8TrQL5s,39521
+qubx/core/series.cpython-311-x86_64-linux-gnu.so,sha256=Bion-CubUyl6u0Cbq28o2weEYilyB4PIUDIh8LguiAQ,779016
 qubx/core/series.pxd,sha256=kF7QeLFgCM-C2hDxVvJM97ZOmyw1v7JEI9WfPKtQ6xs,3002
 qubx/core/series.pyi,sha256=bRBWOaDUBe4Hm2SvraoVeT-n5NFLA92fi8ezrArx9nY,2831
 qubx/core/series.pyx,sha256=UhZsnT7R3-UED05Fnp3mGxr4RSdMpWAHB4lkcz46MFo,32598
 qubx/core/strategy.py,sha256=tTXhDbFTH63yj53mLgVEDBZxYOxC51prcsyXWkdTgLs,10991
-qubx/core/utils.cpython-311-x86_64-linux-gnu.so,sha256=BKuT2LUOVCVEen9gT8XNhwKxmFjr1SEJXZyDqb16nZg,82504
+qubx/core/utils.cpython-311-x86_64-linux-gnu.so,sha256=zbA9MBzGwbiGVsyI1WMk68YBIg_JviygAt6s6NGeH1E,82504
 qubx/core/utils.pyi,sha256=DAjyRVPJSxK4Em-9wui2F0yYHfP5tI5DjKavXNOnHa8,276
 qubx/core/utils.pyx,sha256=-8ek58CrbqWZq0-OY7WSREsCXBoBeWrD_wEYbIBS9rI,1696
+qubx/data/__init__.py,sha256=Ln1GB07OWaD5BcwlapaoHBxUvw2kUGXXS24vgfeRdhk,216
 qubx/data/helpers.py,sha256=A0NGzhpXYWD92-GeB8TghwMnR0NW8bjcNJOCXybQw3g,782
-qubx/data/readers.py,sha256=2VgaHKpHg67qej0Ty6D9q-zLRKEMamQY5pEIIJJcAZw,40062
+qubx/data/readers.py,sha256=t6bG1NSzYZzJc8IYIGnrP-W3RS1ioWkeEX1byaQSQsc,40200
 qubx/gathering/simplest.py,sha256=Ez3YFZMKH3o0jV0Qbg1SuZiuFNs_5No_C7wZ6vOeqfo,3814
 qubx/impl/ccxt_connector.py,sha256=ja_0WJDyZfkzqhNvoV-c5CCg15YnbRIThxw0TTNdwcc,13066
 qubx/impl/ccxt_customizations.py,sha256=WUhDT9x2SYuFrOyBIbk2D9Q_U_5QZhtLomLq88Egf_c,6230
@@ -31,11 +32,11 @@ qubx/impl/ccxt_trading.py,sha256=OW0s_kW7GZlbDz_pGBlzLCUT2iDK4dhaBtmIAvj7QDw,988
 qubx/impl/ccxt_utils.py,sha256=IS5st30L9wkK0akUy2gyPXju3ewk0Obnsdh0R-I_s98,4134
 qubx/math/__init__.py,sha256=QRat0pqEqmbJF7CIHBuqnB1xUek4CsDDhv7SpgssnFw,57
 qubx/math/stats.py,sha256=xgzBU3PLxQ42jBMpI57UVuQ2uGGRYxGKuEd4eiGvjnE,1695
-qubx/pandaz/__init__.py,sha256=Iw5uzicYGSC3FEKZ-W1O5-7cXq_P0kH11-EcXV0zZhs,175
+qubx/pandaz/__init__.py,sha256=xlZvm48eBur0U7N4F4xTKqh7-n5fN0wghj1Tb8lCCNk,332
 qubx/pandaz/ta.py,sha256=NthiiueUoqWGRcjovcKKThcCcdImZn3JRdWDA2vL28k,85075
-qubx/pandaz/utils.py,sha256=XB28Zwv3cXWbKFXbcV5QGj_d6w-i8Yo4LYkX8aPuCHo,19613
+qubx/pandaz/utils.py,sha256=O5wXkYKiTCdGR7hZR4tQtFTbTF-q_kEk-n1DyR-P3X0,19519
 qubx/ta/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qubx/ta/indicators.cpython-311-x86_64-linux-gnu.so,sha256=GNi4LN7GZgIfQOSKMEW19CmctnUW84lulZkdLbwq7TM,576168
+qubx/ta/indicators.cpython-311-x86_64-linux-gnu.so,sha256=My1BpH0ZeZqsobY-j2HT6sWxVsFyvIHIfv5QVrXyahc,576168
 qubx/ta/indicators.pxd,sha256=YzPDhbbNPmy3m4NafwIRF5sNFsmkq-MM1gbeAygtBwM,4007
 qubx/ta/indicators.pyi,sha256=mM_7ISmGQDgjhwxCoZXDw7Rm9fynQSYjevV7fRoLdNc,1683
 qubx/ta/indicators.pyx,sha256=GuxB63YApFnA9IWNJDbBt90J1sOoxTf3jDWYQ3uD9So,24306
@@ -44,7 +45,7 @@ qubx/trackers/composite.py,sha256=rYQCaxu4IKLf5sS3DCyUUJaSUQwbCy6UMoVjsS-481A,62
 qubx/trackers/rebalancers.py,sha256=dp-jemxczT8ndkjjzljsIgGnnA6lZQhsf5c4YdmHKCw,6048
 qubx/trackers/riskctrl.py,sha256=URRZmYgb1-jcW4RTb2QXUozTnBiGhFEWHtTEjaqow5Q,23491
 qubx/trackers/sizers.py,sha256=vh49p2I_5LxYfznOQxlM6f0K1wadgtz7y9NSxg3WdNQ,5953
-qubx/utils/__init__.py,sha256=AL2YibJ3tqBKsZZLUjM9N2J5yy-Kq__k_44oTODQ5sM,321
+qubx/utils/__init__.py,sha256=W9_1KPjh4E9p6E6HzG9aRcTr7hn0PVz42Agi_WfWfJc,319
 qubx/utils/_pyxreloader.py,sha256=FyqGzfSpZGYziB8JYS5AP3cLRAvJSIPAKgwQn0E4YQ0,12017
 qubx/utils/charting/lookinglass.py,sha256=VNSFVfEyAH6NbsvJYcbMvPyGRIRKtLTK1WQ-adH4qkA,39580
 qubx/utils/charting/mpl_helpers.py,sha256=z6vL0IllTZeD-Oe-alVW-kYOP_deHyyoFcukvOVwSz8,35354
@@ -53,6 +54,6 @@ qubx/utils/misc.py,sha256=Av0mhrPCy5NZRrRmjOAhTKusa8wVdL7vCQtEy9bVnz4,10450
 qubx/utils/ntp.py,sha256=LZo4FPVY3rqLUV9VWkLcZaPOpUDFC8Qleynmfggg9No,1758
 qubx/utils/runner.py,sha256=Czo01KUCc9Oj9TIcs03d6Qh7fOpQV5w8oH6UDZ6Yqn0,9539
 qubx/utils/time.py,sha256=fIVWYRmqRT1zkLIWy9jo_Fpfpc03S0sYyOcrHZcfF74,5198
-qubx-0.2.75.dist-info/METADATA,sha256=yq_PoJL-wZQFfQ_Ccmci1z7MNvatP0NL79SIomC421U,2573
-qubx-0.2.75.dist-info/WHEEL,sha256=MLOa6LysROdjgj4FVxsHitAnIh8Be2D_c9ZSBHKrz2M,110
-qubx-0.2.75.dist-info/RECORD,,
+qubx-0.2.76.dist-info/METADATA,sha256=1cu9k670WYRhCYjWj7EOfdyI_kAe6PYmufXJJ7TOa0A,2573
+qubx-0.2.76.dist-info/WHEEL,sha256=MLOa6LysROdjgj4FVxsHitAnIh8Be2D_c9ZSBHKrz2M,110
+qubx-0.2.76.dist-info/RECORD,,