Deepvest/ProfilingAI
0
1o
2 �C~g � @ s� d dl Zd dlZd dlmZmZmZ d dlm Z d dl3m4Z5m Z d dl m
Z
G dd� de
�Ze G dd � d ��ZG d6d � d �ZdS ) � N)�Dict�List�Optional)� dataclass)�datetime� timedelta)�Enumc @ s4 e Zd ZdZdZdZdZdZdZdZ dZ7d Z d8Z d S ) �MarketScenariou Types de scénarios de marché� bull_market� bear_market�high_volatility�low_volatility�crisis�recoveryZsidewaysZhigh_inflation� stagflationN)
�__name__�9__module__� __qualname__�__doc__� BULL_MARKET� BEAR_MARKET�HIGH_VOLATILITY�LOW_VOLATILITY�CRISIS�RECOVERYZSIDEWAYSZHIGH_INFLATION� STAGFLATION� r r �@/Users/sandrine/Deepvest_system/src/analysis/scenario_manager.pyr 10 s r c @ sJ e Zd ZU dZeed< eed< eed< ee ef ed< ee ef ed< dS )�ScenarioConfigu Configuration d'un scénario�
scenario_type�duration�severity�11parameters�macro_conditionsN)12r r r r r �__annotations__�int�floatr �strr r r r r s 13 r c @ s� e Zd ZdZdd� Zdejdedejfdd�Zd e e d14e15e e f de16fd d �Z
de e fd
d�Zdejdedejfdd�Zdejdedejfdd�Zdejde17e e f dejfdd�Zdejd18e19e e f dede20fdd�Zde21dede22fdd �Zd S ) �ScenarioManageru% Gestionnaire de scénarios de marchéc C s || _ || _|| _d S )N)� data_fetcher�market_analyzer� risk_manager)�selfr) r* r+ r r r �__init__" s 23zScenarioManager.__init__� base_data�config�returnc 24 C s� z4|� � }|jtjkr| �||�}n|jtjkr | �||�}n |jtjkr+| �||�}| � ||j25�}|W S t yO } zt d|� �� |W Y d}~S d}~ww )u, Génère un scénario de marché spécifiqueu* Erreur dans la génération du scénario: N)
�copyr r r �_generate_bull_marketr Z_generate_bear_marketr �_generate_crisis�_add_macro_conditionsr# � Exception�print)r, r. r/ �
scenario_data�er r r �generate_scenario' s ��z!ScenarioManager.generate_scenario� scenarios�base_portfolioc � s� �i }|D ]?}z | j | �� I dH |d�}| j||||d�I dH }|||jj< W q tyD } ztd|j� d|� �� W Y d}~qd}~ww |S )u/ Teste la stratégie sous différents scénariosN)r. r/ )�strategyr7 r; r/ u Erreur test scénario z: )r9 Z_get_base_data�_backtest_scenarior �valuer5 r6 ) r, r< r: r; �resultsZscenario_configr7 �scenario_resultsr8 r r r � test_strategy_under_scenarios? s( � � �"��z-ScenarioManager.test_strategy_under_scenariosc C s& t tjdddddd�ddd d26�d �gS ) u+ Génère une série de scénarios de stress�? 皙�����?g @g�������?g333333�?)�volatility_increaseZcorrelation_increaseZliquidity_decreaseg� ��Q���g� ��Q��?g{�G�z�?)�27gdp_growth�inflation_rate�
interest_rate)r r r! r" r# )r r r )r, r r r �generate_stress_scenarios] s ����z)ScenarioManager.generate_stress_scenarios�datac 28 C sv z |� � }d|j }|| }d}|| }d| �� |jd W S ty: } ztd|� �� |W Y d}~S d}~ww )u) Génère un scénario de marché haussierg-C ��6*?rC � r u! Erreur génération bull market: N)�29pct_changer! �cumprod�ilocr5 r6 )r, rI r/ �returnsZbull_adjustmentZadjusted_returnsZvol_adjustmentr8 r r r r2 r s 30��z%ScenarioManager._generate_bull_marketc 31 32 C s� zKd|j }|�� }t|jd �}t�d||�}|jd|� d| �� 9 < |j|d� }|�� d|j d }d| �� |jd |j|d�< |W S t33yf } zt d| � �� |W Y d} ~ S d} ~ ww )u Génère un scénario de criseg333333ÿg333333�?r NrJ rD u Erreur génération crise: ) r! r1 r% r �np�linspacerM rL rK r" r5 r6 )34r, rI r/ Z initial_dropZ data_crisisZ drop_periodZ drop_returnsZ35vol_periodZ
increased_volr8 r r r r3 � s 36 ��z ScenarioManager._generate_crisisr# c C s, |� � }|�� D ] \}}||d|� �< q|S )u5 Ajoute les conditions macro-économiques aux donnéesZmacro_)r1 �items)r, rI r# Zdata_with_macro� conditionr> r r r r4 � s z%ScenarioManager._add_macro_conditionsr7 c 37 � sx �z | j �|� |j||| j d�I dH }| �||�}|||d�W S ty; } ztd|� �� i W Y d}~S d}~ww )u1 Exécute un backtest sur un scénario spécifique)rI Zinitial_portfolior+ N)r/ r? �analysisu Erreur backtest scénario: )r+ Zadjust_for_scenario�backtest�_analyze_scenario_resultsr5 r6 )r, r< r7 r; r/ r? rS r8 r r r r= � s$ � � ���z"ScenarioManager._backtest_scenarior? c C s< | � |d �| �|d �| �|d �| �||�| �|�d�S )u% Analyse les résultats d'un scénariorN )� max_drawdown�
recovery_timeZrisk_adjusted_returnZstrategy_adaptationZ risk_management_effectiveness)�_calculate_max_drawdown�_calculate_recovery_timeZ _calculate_risk_adjusted_returnZ _evaluate_strategy_adaptationZ_evaluate_risk_management)r, r? r/ r r r rU � s 38�z)ScenarioManager._analyze_scenario_resultsN)r r r r r- �pd� DataFramer r9 r r r' r& rA rH r2 r3 r4 r= rU r r r r r( s2 �39�40� " �41��42� r( )�numpyrO �pandasrZ �typingr r r � dataclassesr r r �enumr r r r( r r r r �<module> s