Files
Place-Order-Trading-Bot/patch_drawdown_protection_v2.py
T

148 lines
5.0 KiB
Python

#!/usr/bin/env python3
"""
🔧 Notebook Patch V2: Drawdown Protection Integration (CORRECTED)
Fügt Drawdown Protection korrekt nach Session Filter ein
"""
import json
import sys
from pathlib import Path
NOTEBOOK_PATH = "TradingBot_V1.6_Adaptive_Complete_CORRECTED.ipynb"
def patch_notebook():
"""Patcht das Notebook mit Drawdown Protection (V2 - CORRECTED)"""
print("=" * 70)
print("🔧 NOTEBOOK PATCH V2: Drawdown Protection (CORRECTED)")
print("=" * 70)
# Load notebook
print(f"\n1️⃣ Loading notebook: {NOTEBOOK_PATH}")
try:
with open(NOTEBOOK_PATH, 'r', encoding='utf-8') as f:
notebook = json.load(f)
except FileNotFoundError:
print(f"❌ Error: Notebook not found: {NOTEBOOK_PATH}")
print(" Make sure you're in the correct directory!")
return False
print(f"✅ Loaded {len(notebook['cells'])} cells")
# Find Session Filter cell (Cell 27)
print("\n2️⃣ Finding Session Filter cell...")
session_filter_cell_index = None
for i, cell in enumerate(notebook['cells']):
if cell['cell_type'] == 'code':
source = ''.join(cell['source'])
if 'create_session_filtered_check' in source and 'adaptive_trading_check =' in source:
session_filter_cell_index = i
print(f"✅ Found Session Filter at cell {i}")
break
if session_filter_cell_index is None:
print("❌ Error: Could not find Session Filter cell!")
print(" Looking for cell with 'create_session_filtered_check' and 'adaptive_trading_check ='")
return False
# Create new cell with CORRECT integration
print("\n3️⃣ Creating enhanced Session Filter + Drawdown Protection cell...")
new_cell_source = """# ==========================================
# TRADING CHECK: SESSION FILTER + DRAWDOWN PROTECTION
# ==========================================
from session_filter_patch import (
create_session_filtered_check,
SESSION_WHITELIST_CONFIG,
is_session_allowed
)
from drawdown_protection import create_protected_trading_check
print("🔧 Setting up Trading Check...")
# Step 1: Create base session-filtered trading check
base_trading_check = create_session_filtered_check(
rhythm_manager=rhythm_manager,
execute_func=execute_trade_v2_adaptive,
symbol=symbol,
strategy_name=strategy_name,
max_positions=max_positions,
logger=logger,
datetime=datetime
)
print("✅ Session Filter aktiviert!")
print(" Deaktivierte Sessions:")
for session, enabled in SESSION_WHITELIST_CONFIG['enabled_sessions'].items():
status = "✅ AKTIV" if enabled else "❌ DEAKTIVIERT"
print(f" • {session.upper():8s}: {status}")
# Step 2: Wrap with Drawdown Protection
adaptive_trading_check = create_protected_trading_check(infra, base_trading_check)
drawdown_protection = adaptive_trading_check.protection
print("\\n🛡️ Drawdown Protection aktiviert!")
print(f" • Daily Loss Limit: ${drawdown_protection.max_daily_loss}")
print(f" • Weekly Loss Limit: ${drawdown_protection.max_weekly_loss}")
print(f" • Monthly Loss Limit: ${drawdown_protection.max_monthly_loss}")
print(f" • Max Consecutive Losses: {drawdown_protection.max_consecutive_losses}")
print(f" • Cooldown: {drawdown_protection.cooldown_hours}h")
print("\\n✅ Trading Check ist jetzt vollständig geschützt!")
print(" 📊 Session Filter: Aktiv")
print(" 🛡️ Drawdown Protection: Aktiv")
"""
new_cell = {
"cell_type": "code",
"execution_count": None,
"metadata": {},
"outputs": [],
"source": new_cell_source.split('\n')
}
# Replace the Session Filter cell
print(f"✅ Replacing cell {session_filter_cell_index} with enhanced version")
notebook['cells'][session_filter_cell_index] = new_cell
# Backup original
backup_path = NOTEBOOK_PATH.replace('.ipynb', '_backup_before_drawdown_v2.ipynb')
print(f"\n4️⃣ Creating backup: {backup_path}")
with open(backup_path, 'w', encoding='utf-8') as f:
json.dump(notebook, f, indent=1, ensure_ascii=False)
print("✅ Backup created")
# Save patched notebook
print(f"\n5️⃣ Saving patched notebook: {NOTEBOOK_PATH}")
with open(NOTEBOOK_PATH, 'w', encoding='utf-8') as f:
json.dump(notebook, f, indent=1, ensure_ascii=False)
print("✅ Notebook patched successfully!")
print("\n" + "=" * 70)
print("✅ PATCH V2 COMPLETE!")
print("=" * 70)
print("\nNext steps:")
print("1. Open Jupyter Notebook")
print("2. Kernel → Restart & Run All")
print("3. Verify both Session Filter and Drawdown Protection are active")
print("\nBackup saved to:", backup_path)
print("\n💡 Expected output after restart:")
print(" ✅ Session Filter aktiviert!")
print(" 🛡️ Drawdown Protection aktiviert!")
print(" ✅ Trading Check ist jetzt vollständig geschützt!")
return True
if __name__ == "__main__":
success = patch_notebook()
sys.exit(0 if success else 1)