Deepvest/ProfilingAI
0
1o
2 |ӊg�$ � @ sp d dl Zd dlZd dlmZmZmZ d dlmZm Z d dl3m Z d dl m
Z
e G dd� d��ZG dd � d �ZdS )4� N)�Dict�List�Optional)�datetime� timedelta)� dataclass)�StrategySignalc @ sz e Zd ZU dZejed< ejed< ejed< ejed< ee e5f ed< ee e6f ed< ee e7f ed< ee e8f ed < d9S ) �EnhancedBacktestResultu" Résultats détaillés du backtest�returns� positions�trades�signals�performance_metrics� risk_metrics�alternative_contribution�signal_qualityN) �__name__�10__module__� __qualname__�__doc__�pd�Series�__annotations__� DataFramer �str�float� r r �A/Users/sandrine/Deepvest_system/src/analysis/enhanced_backtest.pyr s 11 12131415r c @ s� e Zd ZdZ d defdd�Z d dee ded ed16e d e17f18d d
�Z de j
dee d efdd�Zdee ded efdd�Zdee deeef dede jd ee f19dd�Zdede jd e20fdd �Zd S ) �EnhancedBacktestEngineu9 Moteur de backtest avancé pour la stratégie améliorée�@B �initial_capitalc C s || _ || _|| _|| _d S )N)�strategy� risk_manager� data_fetcherr )�self�enhanced_strategyr" r# r r r r �__init__ s 21z EnhancedBacktestEngine.__init__T�symbols�22start_date�end_date�use_alternative_data�returnc � s< �z�| j �|||�I dH }| �|j|�}| j}dd� |D �}|jD ]a} zC|r0| �|| �I dH }23ni }24| jj||jd| � |25d�I dH } | � | |||j| �} | �26||| |j| �\}}| � || ||| | � W q! t y� }
zt
d| � d|
� �� W Y d}
~
q!d}
~
ww | �||�W S t y� }
zt
d|
� �� � d}
~
ww )u7 Exécute un backtest complet avec données alternativesNc S s i | ]}|d �qS )r r )�.0�symbolr r r �27<dictcomp>1 s z@EnhancedBacktestEngine.run_enhanced_backtest.<locals>.<dictcomp>)� portfolio� market_data�alternative_datau& Erreur pendant le backtest à la date �: zErreur dans le backtest: )r# �fetch_market_data�_initialize_results�indexr � _get_historical_alternative_datar! �generate_trade_signals�loc�_execute_signalsZ_update_portfolioZ_record_results� Exception�print�_calculate_final_metrics)r$ r' r( r) r* r0 �results�portfolio_value�current_positions�dater1 r
r �er r r �run_enhanced_backtest$ sb �28 � ���� ����z,EnhancedBacktestEngine.run_enhanced_backtest�datesc C sJ t j|td�t j||td�t jg d�d�t jg d�d�t jg d�d�d�S )u9 Initialise les structures de données pour les résultats)r5 �dtype)r5 �columnsrD �r- �type�size�price�cost)rE )r- � direction�29confidencerH )r- Z signal_type�value)r> r r r
�alternative_signals)r r r r )r$ rC r' r r r r4 j s �z*EnhancedBacktestEngine._initialize_resultsr@ c 30 � s� �z%| j �||�I dH }| j �||�I dH }| j �||�I dH }|||d�W S tyA } ztd|� �� i W Y d}~S d}~ww )u0 Récupère les données alternatives historiquesN)� satellite� social_media� web_trafficu- Erreur récupération données alternatives: )r# Z fetch_historical_satellite_dataZ fetch_historical_social_dataZ fetch_historical_traffic_datar: r; )r$ r' r@ �satellite_data� social_data� traffic_datarA r r r r6 t s( �31�32�33����z7EnhancedBacktestEngine._get_historical_alternative_datar
r? r> r0 c C s� g }|D ]m}zN|j dkr.| �|||�}|dkr-|�|jd|||j |||j d d�� n%|j dkrS| �||�}|dkrS|�|jd|||j |||j d d�� W q tyq } ztd|j� d|� �� W Y d}~qd}~ww |S ) u Exécute les signaux de trading�buyr g����MbP?rF �sellu Erreur exécution signal r2 N)rK Z_calculate_buy_size�appendr- Z_calculate_sell_sizer: r; ) r$ r
r? r> r0 r �signalrH rA r r r r9 � sF 34���35��� ��z'EnhancedBacktestEngine._execute_signalsr= c 36 C s� zM|d � � �� }|d jd | j d | �|�| �|�| �|�d�}| j�|�}| � |d |d �}| �37|d |�}t ||d |d |d ||||d �W S t ya } zt
d38|� �� � d }~ww ) u* Calcule les métriques finales du backtestr> ������ )� total_return�
annual_return� sharpe_ratio� max_drawdownr
rN r r )r39 r r r
r r r r u" Erreur calcul métriques finales: N)�40pct_change�dropna�ilocr Z_calculate_annual_return�_calculate_sharpe_ratio�_calculate_max_drawdownr" Zcalculate_risk_metricsZ#_calculate_alternative_contributionZ_evaluate_signal_qualityr r: r; ) r$ r= r0 �portfolio_returnsr r r r rA r r r r< � s: � ��� ��z/EnhancedBacktestEngine._calculate_final_metricsN)r )T)r r r r r r& r r r �boolr rB r �
DatetimeIndexr r4 r6 r r r9 r r< r r r r r sL �41������42� F43��44� �45���46� *r )�pandasr �numpy�np�typingr r r r r � dataclassesr Zsrc.models.signalsr r r r r r r �<module> s