| import sys, re |
|
|
| with open('main.py', 'r', encoding='utf-8') as f: |
| text = f.read() |
|
|
| pattern = r'''# Force close at configured time.*?elif is_trading_window\(\):''' |
| new_code = '''# Force close at configured time |
| if is_close_time(): |
| # Custom logic: End of Day Sweep |
| # If in profit -> SELL. If in loss -> HOLD OVERNIGHT. |
| to_close = [] |
| portfolio.refresh() |
| for sym, pos in portfolio.positions().items(): |
| current = pos.get('current_price', 0) |
| entry = pos.get('entry_price', 0) |
| if current > entry: |
| to_close.append(sym) |
| |
| if to_close: |
| logger.info("End of Day Sweep: Selling profitable positions: %s", to_close) |
| closed_symbols = [] |
| for sym in to_close: |
| try: |
| from execution.broker import close_position |
| close_position(sym) |
| closed_symbols.append(sym) |
| except Exception as e: |
| logger.error("Failed to close %s: %s", sym, e) |
| if closed_symbols: |
| alert(f"?? End of Day Sweep: Locked in profit for: {', '.join(closed_symbols)}") |
| |
| elif is_trading_window():''' |
|
|
| text = re.sub(pattern, new_code, text, flags=re.DOTALL) |
| with open('main.py', 'w', encoding='utf-8') as f: |
| f.write(text)
|
|
|
