o )mhZB@srdZddlZddlZddlmZmZmZm Z ddl m Z m Z ddl Z ddl mZddlZddlZGdddZdS)uL Gestionnaire de corrélation BTC pour pondération des entrées de trading N)DictListOptionalTuple)datetime timedelta)dequec@seZdZdZd,defddZddZd d Zd d Zd-d e dede e fddZ d-d e de e de fddZ d-de e fddZde de fddZde de fddZd e de de fddZd e d edeee ffd!d"Zd#edefd$d%Zdefd&d'Zd(e d)e ede efd*d+ZdS).BTCCorrelationManageruy Calcule et gère les corrélations entre BTC et autres cryptos pour optimiser la distribution des positions Ndlookback_periodscCs||_||_td|_i|_t|d|_i|_d|_ d|_ dddd|_ d d d d d d |_ ddddd|_ t|_d|_d|_dS)u Args: hyperliquid_client: Client pour récupérer les données de marché lookback_periods: Nombre de périodes pour calculer la corrélation r )maxlenNi,皙?333333?333333?)highmediumlow?ffffff??g? very_highrrrvery_low )rrrrF)clientr logging getLoggerlogger price_cacher btc_pricescorrelation_cachecache_timestampcache_durationcorrelation_thresholdsweight_factorsmax_correlated_exposure threadingLocklock update_threadrunning)selfhyperliquid_clientr r0T/var/www/html/HyperOracle/crypto_trader/simple_trading_system/correlation_manager.py__init__s4     zBTCCorrelationManager.__init__cCsB|jsd|_tj|jd|_d|j_|j|jddSdS)u6Démarre la mise à jour périodique des corrélationsT)targetz&Background correlation updates startedN) r-r)Thread_background_updaterr,daemonstartr infor.r0r0r1start_background_updatesCs z.BTCCorrelationManager.start_background_updatescCs"d|_|jr|jjdddSdS)u*Arrête les mises à jour en arrière-planFr)timeoutN)r-r,joinr9r0r0r1stop_background_updatesLsz-BTCCorrelationManager.stop_background_updatesc Csn|jr5z |t|jWn ty/}z|jd|tdWYd}~nd}~ww|jsdSdS)u4Thread de mise à jour périodique des corrélationszError in background updater: <N)r-update_all_correlationstimesleepr% Exceptionr error)r.er0r0r1r5Rsz)BTCCorrelationManager._background_updatersymbolperiodsreturnc Cs|jsgS|p |j}z&|jj|d|d}|r.dd|D}t||kr+|| dWS|WSWgStyO}z|jd|d|WYd}~gSd}~ww)u-Récupère l'historique des prix d'un symbole1h)lookback_hourscSs&g|]}|drt|ddqS)closer)getfloat.0cr0r0r1 hs&z;BTCCorrelationManager.get_price_history..Nz!Error fetching price history for : )rr get_candles_datalenrBr rC)r.rErFcandlespricesrDr0r0r1get_price_history\s ""z'BTCCorrelationManager.get_price_historyr"c CszW|dur |d}|sWdS||}|rt|t|kr!WdStt|}tt|}t|dkrRt|dkrUt||d}t|rMWdSt|WSWdSWdStyv}z|j d|d|WYd}~dSd}~ww)u Calcule la corrélation entre un symbole et BTC Returns: Coefficient de corrélation entre -1 et 1 NBTC?r)rz"Error calculating correlation for rQ) rVrSnpdifflogcorrcoefisnanrLrBr rC)r.rEr" symbol_prices btc_returnssymbol_returns correlationrDr0r0r1calculate_correlationos0      "z+BTCCorrelationManager.calculate_correlationsymbolsc CsR|jzo|d}|s|jdWWddS|s?|jr?z|j}|r5d|vr5dd|dD}Wngd}Yi}|D]}|dkr]|||}|||||d||<qC||_ t |_ |j d t|d Wnty}z|jd |WYd}~n d}~wwWddSWddS1swYdS) u=Met à jour le cache des corrélations pour tous les symbolesrWz1Could not fetch BTC prices for correlation updateNuniversecSsg|]}|dqS)namer0)rNassetr0r0r1rPszABTCCorrelationManager.update_all_correlations..) ETHSOLARBMATICAVAXBNBLINKUNIAAVECRVALGOGRT)rb weight_factorcategoryzUpdated correlations for z symbolszError updating correlations: )r+rVr warningrget_metarcget_weight_factorget_correlation_categoryr#rnowr$r8rSrBrC)r.rdr"meta new_cacherErbrDr0r0r1r?sF         %"z-BTCCorrelationManager.update_all_correlationsrbcCs<t|}|dkr dS|dkrdS|dkrdS|dkrdSd S) u%Catégorise le niveau de corrélationrrr rrrrrr)abs)r.rbabs_corrr0r0r1rysz.BTCCorrelationManager.get_correlation_categorycCs||}|j|dS)u Retourne le facteur de pondération basé sur la corrélation Returns: Facteur entre 0.4 et 1.2 r)ryr'rK)r.rbrur0r0r1rxs z'BTCCorrelationManager.get_weight_factor base_sizecCsL|r|||jvr|j|d}||S||}||}||S)u Calcule la taille de position pondérée selon la corrélation Args: symbol: Le symbole à trader base_size: Taille de position de base Returns: Taille de position ajustée rt)needs_cache_updater?r#rcrx)r.rErrtrbr0r0r1get_position_weights    z)BTCCorrelationManager.get_position_weightcurrent_positionsc s|r|||jvr||}||}n|j|d}dddddd|D]}||jvrC|j|d}|dd|<q+|dvrfdd}|jdd}||krfd d |d |d fStfd dD} | dkr~d d| ddfSdS)u Détermine si une nouvelle position peut être ouverte selon les corrélations Args: symbol: Symbole à trader current_positions: Positions actuelles Returns: (peut_ouvrir, raison) rurrrYrrrrrFu4Limite de positions fortement corrélées atteinte (/)c3s@|]}||dkr dn |dkrdn|dkrdndVqdS)rg@rg?rrrXNr0)rNcatcategory_countsr0r1 s  z=BTCCorrelationManager.should_open_position..ru,Exposition totale pondérée trop élevée (.1fz/10))TOK) rr?r#rcrykeysrKr(sum) r.rErrbru pos_symbol pos_categorycurrent_high_corr max_allowedtotal_weighted_exposurer0rr1should_open_positions.       z*BTCCorrelationManager.should_open_position positionsc Cs|s dddddSg}|D] }||jvr#|t|j|dq||}|t|q|s9dddddSt|}t|}d|}tdd|D}|rX|t |nd} t |t |t |t | t ||dS)u Calcule les métriques de corrélation du portefeuille Returns: Dictionnaire avec les métriques du portefeuille rr)avg_correlationmax_correlationdiversification_scorerisk_concentrationrbcss|] }|dkrdVqdS)r rYNr0rMr0r0r1rMsz>BTCCorrelationManager.get_portfolio_metrics..)rrrrtotal_positionshigh_corr_positions) rr#appendr}rcrZmeanmaxrrSrL) r.r correlationsrEcorravg_corrmax_corrdiversificationhigh_corr_countrr0r0r1get_portfolio_metrics&s<     z+BTCCorrelationManager.get_portfolio_metricscCs&|jsdSt|j}||jkS)u+Vérifie si le cache doit être mis à jourT)r$rrz total_secondsr%)r.elapsedr0r0r1rYs z(BTCCorrelationManager.needs_cache_updateavailable_balancesignalsc Cs|sgS|r |g}|D]K}|dd}||jvr+|j|}|d}|d}n ||}||}|dd} | |} |} || d<|| d<| | d<||| d<|| q|j d d d d g} d } d d d d d d}|D]\}|d}|dvr|d|d}||j dkrd|d<qs|d}||d}| ||dkrd|d<qs||d<||d|d<| || |7} ||d d||<t | dkrnqs|j dt | d| dd| |dd d!| S)"u/ Recommande une distribution optimale des positions selon les corrélations Args: available_balance: Balance disponible pour trading signals: Liste des signaux de trading Returns: Liste des signaux avec tailles de position ajustées rErbrtscorerXadjusted_scorecorrelation_categorycSs|dS)Nrr0)xr0r0r1szMBTCCorrelationManager.get_recommended_position_distribution..T)keyreverserrrrru"Limite corrélation haute atteinte skip_reasong?rzBudget insuffisant position_sizer position_percentrYrzPosition distribution: z positions, Total allocated: $z.2fz (rz%))rr?rKr#rcrxcopyryrsortr(rSr r8)r.rrenhanced_signalssignalrE corr_datarbrtoriginal_scorerenhanced_signalallocated_signalstotal_allocatedrru total_highbase_position_size adjusted_sizer0r0r1%get_recommended_position_distributionasf              z;BTCCorrelationManager.get_recommended_position_distribution)Nr )N)__name__ __module__ __qualname____doc__intr2r:r=r5strrrLrVrcr?ryrxrrrboolrrrrr0r0r0r1r s(/  () 43r )rnumpyrZpandaspdtypingrrrrrrr collectionsrr)r@r r0r0r0r1s