U
9%xh�W � @ s� d Z ddlZddlZddlZddlZddlmZmZ ddlmZm Z m
Z
m Z m Z ddl
ZddlZddlmZmZ ddlmZ ddlmZmZ ddlZejejd� e� e �Z G d d
� d
e�Z G d d � d e�Z G d
d� de�Z!G dd� de�Z"G dd� de�Z#G dd� de�Z$G dd� de�Z%G dd� de�Z&G dd� de�Z'G dd � d e�Z(ed d d d �Z)e)j*ed!gd"d!gd!gd#� e+e+d$�d%d&�Z,e e- e+d'�d(d)�Z.e)�/d*�d+d,� �Z0e)�/d-�d.d/� �Z1e)j/d0e d1�e+d2�d3d4��Z2e)j/d5e!d1�dWe+e+e+d8�d9d:��Z3e)j/d;e#d1�dXe+e+e+d>�d?d@��Z4e)j/dAe
e" d1�dYe+e5dC�dDdE��Z6e)j/dFe(d1�e+d2�dGdH��Z7e)j/dIe%d1�dZe+e e+ dJ�dKdL��Z8e)�9dM�e e+ef dN�dOdP��Z:e)j/dQe'd1�dRdS� �Z;dTdU� Z<e dVk�r�e<� dS )[z�Yahoo Finance Server
A financial data analysis server inspired by the Model Context Protocol,
providing comprehensive Yahoo Finance data through a clean API.
� N)�datetime� timedelta)�Any�Dict�List�Optional�Union)�FastAPI�
HTTPException)�CORSMiddleware)� BaseModel�Field)�levelc @ s� e Zd ZU dZeed< dZeed< dZeed< dZ e
e ed< dZ e
e ed < dZ
e
e ed
< dZe
e ed < dZe
e ed < dZe
e ed
< dZe
e ed< dZe
e ed< dS )� StockInfoz Stock information data model.�symbol� � company_name� �
current_priceN�
market_cap�pe_ratio�dividend_yield�fifty_two_week_high�fifty_two_week_low�volume�sector�industry)�__name__�
__module__� __qualname__�__doc__�str�__annotations__r r �floatr r �intr r r r r r r � r% r% �9/Users/yyt/Desktop/yf-mcp/src/yahoo_finance_mcp/server.pyr s
r c @ sB e Zd ZU dZeed< eed< eed< eed< eed< eed< dS ) �HistoricalPricez Historical price data point.�date�open�high�low�closer N�r r r r r! r" r# r$ r% r% r% r& r' ) s
r' c @ sN e Zd ZU dZeed< eed< eed< eed< eed< eed< ee ed< d S )
�HistoricalDataz!Historical price data collection.r �period�interval� data_points�
start_date�end_date�pricesN) r r r r r! r" r$ r r' r% r% r% r& r. 3 s
r. c @ s> e Zd ZU dZeed< eed< eed< eed< dZeed< dS ) � NewsArticlezNews article data model.�title� publisher�link� published_atr �summaryN)r r r r r! r" r: r% r% r% r& r5 > s
r5 c @ sB e Zd ZU dZeed< eed< eed< eeeeef f ed< dS )�FinancialStatementzFinancial statement data.r r/ �statement_type�dataN)r r r r r! r" r r r% r% r% r& r; G s
r; c @ sJ e Zd ZU dZeed< eed< eed< eed< eed< eed< eed< d S )
�OptionsContractzOptions contract data.�strike�
last_price�bid�askr �
open_interest�implied_volatilityN)r r r r r# r" r$ r% r% r% r& r> O s
r> c @ sF e Zd ZU dZeed< eed< ee ed< ee ed< ee ed< dS )� OptionsDatazOptions data collection.r �
expiration�available_expirations�calls�putsN)r r r r r! r" r r> r% r% r% r& rE Z s
rE c @ s: e Zd ZU dZeed< eed< eed< eed< eed< dS )� MarketIndexzMarket index data.�namer �change�change_percentr Nr- r% r% r% r& rJ c s
rJ c @ s* e Zd ZU dZeed< eeef ed< dS )�
MarketSummaryzMarket summary data.� summary_date�indicesN)r r r r r! r" r rJ r% r% r% r& rN l s
rN c @ s> e Zd ZU dZeed< eeef ed< eeeef ed< dS )�AnalystRecommendationz Analyst recommendation data.r �current_recommendations�recent_changesN) r r r r r! r" r r r r% r% r% r& rQ r s
rQ �Yahoo Finance Serverz;Financial data analysis server providing Yahoo Finance data�0.1.0)r6 � description�version�*T)�
allow_origins�allow_credentials�
allow_methods�
allow_headers)r �returnc C s$ | r | � � stddd��| � � �� S )z Validate and clean stock symbol.� z Stock symbol cannot be empty�� status_code�detail)�stripr
�upper�r r% r% r& �validate_symbol� s re )�amountr] c C sv | dkr dS t | �dkr*d| d d�d�S t | �dkrHd| d d�d�S t | �d krfd| d d�d
�S d| d��S dS ) zFormat currency amount.N�N/Ag ��mB�$z.2f�Tg e��A�Bg ��.A�M)�abs)rf r% r% r& �format_currency� s rm �/c � s ddddddddd d
d gd �S )
z&Root endpoint with server information.rT rU z Financial data analysis server�/stock/{symbol}�/historical/{symbol}�/financial/{symbol}�/news/{symbol}�/analyst/{symbol}�/options/{symbol}�/compare�/market-summary)rK rW rV �available_endpointsr% r% r% r% r& �root� s ��rx z/toolsc
� s� ddddddid�dd d
dd d d
�d�ddddddd�d�dddddd�d�dddddid�d d d dd d �d�d!d"d#d$d%d&�d�d'd(d)i d�giS )*z List available tools/endpoints.�tools�get_stock_infoz#Get comprehensive stock informationro r zStock ticker symbol)rK rV �endpoint�
parameters�get_historical_priceszGet historical price datarp z>Time period (1d, 5d, 1mo, 3mo, 6mo, 1y, 2y, 5y, 10y, ytd, max)zIData interval (1m, 2m, 5m, 15m, 30m, 60m, 90m, 1h, 1d, 5d, 1wk, 1mo, 3mo)�r r/ r0 �get_financial_statementszGet financial statementsrq z>Type of statement (income_statement, balance_sheet, cash_flow)z Period type (annual, quarterly)�r r< r/ �get_stock_newszGet latest news articlesrr zMaximum number of articles�r � max_articles�get_analyst_recommendationszGet analyst recommendationsrs �get_options_datazGet options datart z#Specific expiration date (optional)�r �expiration_date�compare_stockszCompare multiple stocksru z List of stock symbols to comparez&Specific metrics to compare (optional))�symbols�metrics�get_market_summaryzGet market summaryrv r% r% r% r% r& �
list_tools� sn ��� �� ��
���
��
���r� ro )�response_modelrd c � s z�t | �} t�| �}|j}|r*t|�dkr<tdd| � �d��t| |�dd�|�dd�|�d �|�d
�|�d �|�d �|�d
�|�d�|�d�|�d�d� W S tk
r� � Y nP tk
� r� } z0t �
d| � d|� �� tddt |�� �d��W 5 d}~X Y nX dS )z,Get comprehensive information about a stock.� � zNo data found for symbol: r_ �longNamer � currentPricer � marketCap�
trailingPE�
dividendYield�fiftyTwoWeekHigh�fiftyTwoWeekLowr r r ) r r r r r r r r r r r z Error getting stock info for �: �� z&Failed to retrieve stock information: N) re �yf�Ticker�info�lenr
r �get� Exception�logger�errorr! )r �stockr� �er% r% r& rz � s0
�
rz rp �1mo�1dr~ c
� s` z�t | �} t�| �}|j||d�}|jr:tdd| � �d��g }|�� D ]z\}}|�t|� d�t
t |d �d�t
t |d �d�t
t |d �d�t
t |d
�d�t �
|d �r�t|d �nd d
�� qFt| ||t|�|jd � d�|jd � d�|d�W S tk
�r � Y nP tk
�rZ } z0t�d| � d|� �� tddt|�� �d��W 5 d}~X Y nX dS )z&Get historical price data for a stock.)r/ r0 r� z No historical data found for r_ �%Y-%m-%d�Open� �High�Low�Close�Volumer )r( r) r* r+ r, r �����)r r/ r0 r1 r2 r3 r4 z$Error getting historical prices for r� r� z$Failed to retrieve historical data: N)re r� r� �history�emptyr
�iterrows�appendr' �strftime�roundr# �pd�notnar$ r. r� �indexr� r� r� r! ) r r/ r0 r� �histr4 r( �rowr� r% r% r& r} s<
�
� r} rq �income_statement�annualr� c
� s� �z8t | �} t�| �}|dkr6|dkr.|j}q�|j}nR|dkrV|dkrN|j}q�|j}n2|dkrv|dkrn|j}q�|j}nt dd|� �d��|j
r�t dd |� d
| � �d��i }|j D ]x}t |d �r�|�
d �nt|�}i ||< |jD ]H}|j||f } t�| �r�t| ttf��rt| �nt| �|| t|�< q�q�t| |||d
�W S t k
�rR � Y nP tk
�r� }
z0t�d| � d|
� �� t ddt|
�� �d��W 5 d}
~
X Y nX dS )z'Get financial statements for a company.r� r� �
balance_sheet� cash_flowr^ zInvalid statement type: r_ r� zNo z data found for r� z%Y)r r/ r< r= z'Error getting financial statements for r� r� z#Failed to retrieve financial data: N)re r� r� �
financials�quarterly_financialsr� �quarterly_balance_sheet�cashflow�quarterly_cashflowr
r� �columns�hasattrr� r! r� �locr� r� �
isinstancer$ r# r; r� r� r� ) r r< r/ r� �stmtr= �col�year�idx�valuer� r% r% r& r I sJ
0�r rr �
r� c � s8 z�t | �} t�| �}|j}|s$g W S g }|d|� D ]�}zTt�|�dd��}|�t|�dd�|�dd�|�dd�|� d�|�d d�d
�� W q4 t
k
r� } z t � d |� �� W Y �q4W 5 d}~X Y q4X q4|W S t
k
r� � Y nP t
k
�r2 } z0t �d | � d
|� �� t
ddt|�� �d��W 5 d}~X Y nX dS )z%Get latest news articles for a stock.N�providerPublishTimer r6 r r7 r8 �%Y-%m-%d %H:%M:%Sr: )r6 r7 r8 r9 r: zError processing article: zError getting news for r� r� zFailed to retrieve news: r_ )re r� r� �newsr �
fromtimestampr� r� r5 r� r� r� �warningr
r� r! ) r r� r� r� �articles�article�published_time�
article_errorr� r% r% r& r� � s4
� r� rs c
� s� �zBt | �} t�| �}|j}|j}i }g }|dk r�|js�|jd }t|jd�rZ|j� d�nt
|j�t |� dd��t |� dd��t |� dd��t |� d d��t |� d
d��d �}|dk �r6|j�s6|�
d ��� D ]f\}}|�t|d�r�|� d
�nt
|�t
|� dd��t
|� dd��t
|� dd��t
|� dd��d�� q�t| ||d�W S tk
�r\ � Y nP tk
�r� } z0t�d| � d| � �� tddt
| �� �d��W 5 d} ~ X Y nX dS )z(Get analyst recommendations for a stock.Nr� r� z%Y-%m� strongBuyr �buy�hold�sell�
strongSell)r/ �
strong_buyr� r� r� � strong_sellr� r� �Firmr �ToGrade� FromGrade�Action)r( �firm�to_grade�
from_grade�action)r rR rS z*Error getting analyst recommendations for r� r� z!Failed to retrieve analyst data: r_ )re r� r� �recommendations�upgrades_downgradesr� �ilocr� rK r� r! r$ r� �headr� r� rQ r
r� r� r� )
r r� r� r� rR rS �latestr( r� r� r% r% r& r� � sF
�
�
�r� rt r� c � s� �z^t | �} t�| �}|j}|s2tdd| � �d��|r`||krZtdd|� dt|�� �d��|}n|d }|�|�}g }g }|jj�s^|j� � D ]�\}} |�
t t | d �t
�| d �r�t | d �nd
t
�| d �r�t | d �nd
t
�| d �r�t | d �nd
t
�| d
��rt| d
�ndt
�| d ��r4t| d �ndt
�| d ��rRt | d �nd
d�� q�|jj�sJ|j� � D ]�\}} |�
t t | d �t
�| d ��r�t | d �nd
t
�| d ��r�t | d �nd
t
�| d ��r�t | d �nd
t
�| d
��r t| d
�ndt
�| d ��r t| d �ndt
�| d ��r<t | d �nd
d�� �qrt| |t|�||d�W S tk
�rx � Y nP tk
�r� }
z0t�d| � d|
� �� tddt|
�� �d��W 5 d}
~
X Y nX dS )z Get options data for a stock.r� z No options data available for r_ r^ zExpiration date z not available. Available: r r? � lastPricer rA rB r � openInterest�impliedVolatility)r? r@ rA rB r rC rD )r rF rG rH rI z Error getting options data for r� r� z!Failed to retrieve options data: N)re r� r� �optionsr
�list� option_chainrH r� r� r� r> r# r� r� r$ rI rE r� r� r� r! ) r r� r� � expirations�exp_date�
options_chainrH rI �_r� r� r% r% r& r� � sf
�
�
�
�r� ru )�requestc � s� �z`| � dg �}| � d�}|s*tddd��|dkrDddd d
d d d
g}g t�� �d�i d�}i }|D ]�}zBt|�}t�|�}|j}|r�t |�dkr�|||< |d �
|� W qb t k
r� } z"t �
d|� d|� �� W Y �qbW 5 d}~X Y qbX qb|s�tddd��|D ]\} i |d | < |d D ]@}|| � | �}
|
dk �rF|
|d | |< nd|d | |< �q�q |W S tk
�rz � Y nJ t k
�r� } z*t �d|� �� tddt|�� �d��W 5 d}~X Y nX dS )z/Compare multiple stocks across various metrics.r� r� r^ z$At least one symbol must be providedr_ Nr� r� r� r� r� r� r r� )r� �comparison_dater� r� �Failed to get data for r� r� z)No valid stock data found for any symbolsrg zError comparing stocks: r� zFailed to compare stocks: )r� r
r �nowr� re r� r� r� r� r� r� r� r� r� r! ) r� r� r� �
comparison�
stock_datar r� r� r� �metricr� r% r% r& r� sT
� �
r� rv c � s� �zJdddddd�} i }| � � D �]\}}z�t�|�}|jdd�}|js�|d jd
}t|�d krp|d jd n|}|| }|d
kr�|| d nd
} t|tt |�d�tt |�d�tt | �d�t
� |d jd
�r�t |d jd
�nd
d�||< W q t
k
�r2 }
z"t�d|� d|
� �� W Y �q W 5 d}
~
X Y q X q tt�� �d�|d�W S t
k
�r� }
z*t�d|
� �� tddt|
�� �d��W 5 d}
~
X Y nX dS )z&Get a summary of major market indices.zS&P 500z Dow Jones�NASDAQz Russell 2000�VIX)z^GSPCz^DJIz^IXICz^RUTz^VIX�2d)r/ r� r� r� �����r �d r� r� )rK r rL rM r r� r� Nr� )rO rP z Error getting market summary: r� z#Failed to retrieve market summary: r_ )�itemsr� r� r� r� r� r� rJ r� r# r� r� r$ r� r� r� rN r r� r� r� r
r! ) rP � summary_datar rK �tickerr� �current�previousrL �
change_pctr� r% r% r& r� P sD �
(� �r� c C sJ t t�dd��} t�dd�}t�d|� d| � �� tjd|| dd d
� d S ) z.Main entry point for the Yahoo Finance server.�YF_SERVER_PORT�8000�YF_SERVER_HOSTz 127.0.0.1z!Starting Yahoo Finance Server on �:z yahoo_finance_mcp.server:appFr� )�host�port�reload� log_levelN)r$ �os�getenvr� r� �uvicorn�run)r r r% r% r& �main} s �r �__main__)r� r� )r� r� )r� )N)=r �asyncio�json�loggingr r r �typingr r r r r �yfinancer� �pandasr� �fastapir r
�fastapi.middleware.corsr �pydanticr r
r � basicConfig�INFO� getLoggerr r� r r' r. r5 r; r> rE rJ rN rQ �app�add_middlewarer! re r# rm r� rx r� rz r} r r$ r� r� r� �postr� r� r r% r% r% r& �<module> s�
��
I ��, ��7! 0 D6
,
