From b9c35d3310601f038a18b93fee7f63e68bacf6a8 Mon Sep 17 00:00:00 2001
From: moshferatu <moshferatu@moshferatu.dev>
Date: Wed, 27 Nov 2024 12:40:53 -0800
Subject: [PATCH] Add script for calculating strategy signals based on imported
 strategies from Strategies module

---
 signals.py | 86 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 signals.py

diff --git a/signals.py b/signals.py
new file mode 100644
index 0000000..02d5981
--- /dev/null
+++ b/signals.py
@@ -0,0 +1,86 @@
+from datetime import datetime
+from pandas import DataFrame
+
+from strategies import advances
+from strategies import cumulative_rsi
+from strategies import declines
+from strategies import double_5s
+from strategies import down_days_in_a_row
+from strategies import end_of_month
+from strategies import high_volume_days
+from strategies import hilo_index_lows
+from strategies import ibs_rsi
+from strategies import internal_bar_strength
+from strategies import internal_bar_strength_band
+from strategies import large_moves_down
+from strategies import lower_lows
+from strategies import put_call_ratio_highs
+from strategies import rsi_power_zones
+from strategies import short_term_lows
+from strategies import tps
+from strategies import trin
+from strategies import trin_thrusts
+from strategies import turnaround
+from strategies import two_period_rsi
+from strategies import vix_above_moving_average
+from strategies import vix_reversal_1
+from strategies import vix_reversal_2
+from strategies import vix_reversal_3
+from strategies import vix_rsi
+
+from daily_data import get_daily_data
+
+def calculate_signals(data: DataFrame, days: int = 12) -> DataFrame:
+    start_date = data['Date'].min()
+    end_date = data['Date'].max()
+
+    advances_data = get_daily_data(symbol = 'IINA.Z', start_date = start_date, end_date = end_date)
+    declines_data = get_daily_data(symbol = 'IIND.Z', start_date = start_date, end_date = end_date)
+    new_highs_data = get_daily_data(symbol = 'FINH.Z', start_date = start_date, end_date = end_date)
+    new_lows_data = get_daily_data(symbol = 'FINL.Z', start_date = start_date, end_date = end_date)
+    put_volume_data = get_daily_data(symbol = 'VPOT.Z', start_date = start_date, end_date = end_date)
+    call_volume_data = get_daily_data(symbol = 'VCOT.Z', start_date = start_date, end_date = end_date)
+    trin_data = get_daily_data(symbol = 'RINT.Z', start_date = start_date, end_date = end_date)
+    vix_data = get_daily_data(symbol = 'VIX.XO', start_date = start_date, end_date = end_date)
+
+    strategy_signals = [
+        {'strategy': '2-Period RSI', 'signals': two_period_rsi(data)},
+        {'strategy': 'Advances', 'signals': advances(data, advances_data, declines_data)},
+        {'strategy': 'Cumulative RSI', 'signals': cumulative_rsi(data)},
+        {'strategy': 'Declines', 'signals': declines(data, declines_data, advances_data)},
+        {'strategy': 'Double 5\'s', 'signals': double_5s(data)},
+        {'strategy': 'Down Days in a Row', 'signals': down_days_in_a_row(data)},
+        {'strategy': 'End of Month', 'signals': end_of_month(data)},
+        {'strategy': 'High Volume Days', 'signals': high_volume_days(data)},
+        {'strategy': 'HILO Index Lows', 'signals': hilo_index_lows(data, new_highs_data, new_lows_data)},
+        {'strategy': 'IBS + RSI', 'signals': ibs_rsi(data)},
+        {'strategy': 'Internal Bar Strength', 'signals': internal_bar_strength(data)},
+        {'strategy': 'Internal Bar Strength Band', 'signals': internal_bar_strength_band(data)},
+        {'strategy': 'Large Moves Down', 'signals': large_moves_down(data)},
+        {'strategy': 'Lower Lows', 'signals': lower_lows(data)},
+        {'strategy': 'Put / Call Ratio Highs', 'signals': put_call_ratio_highs(data, put_volume_data, call_volume_data)},
+        {'strategy': 'RSI PowerZones', 'signals': rsi_power_zones(data)},
+        {'strategy': 'Short-Term Lows', 'signals': short_term_lows(data)},
+        {'strategy': 'TPS', 'signals': tps(data)},
+        {'strategy': 'TRIN', 'signals': trin(data, trin_data)},
+        {'strategy': 'TRIN Thrusts', 'signals': trin_thrusts(data, trin_data)},
+        {'strategy': 'Turnaround', 'signals': turnaround(data)},
+        {'strategy': 'VIX Above Moving Average', 'signals': vix_above_moving_average(data, vix_data)},
+        {'strategy': 'VIX Reversal 1', 'signals': vix_reversal_1(data, vix_data)},
+        {'strategy': 'VIX Reversal 2', 'signals': vix_reversal_2(data, vix_data)},
+        {'strategy': 'VIX Reversal 3', 'signals': vix_reversal_3(data, vix_data)},
+        {'strategy': 'VIX RSI', 'signals': vix_rsi(data, vix_data)}
+    ]
+
+    signal_data = []
+    for signal_info in strategy_signals:
+        signal_dict = {'Strategy': signal_info['strategy']}
+
+        signals = signal_info['signals'].tail(days)
+        dates = [datetime.strptime(str(date), '%Y-%m-%d').strftime('%m/%d') for date in data.tail(days)['Date']]
+        for date, signal in zip(dates, signals):
+            signal_dict[date] = signal
+
+        signal_data.append(signal_dict)
+    
+    return DataFrame(signal_data)
\ No newline at end of file