Deepvest/ProfilingAI
0
1o
2 �C~g� � @ s* d dl Z d dlZd dlZG dd� d�ZdS )� Nc @ s e Zd Zdd� Zdd� ZdS )�FinnhubDataProviderc C s t jdd�| _d S )NZ(csrq269r01qj3u0os3r0csrq269r01qj3u0os3rg)�api_key)�finnhub�Client�client)�self� r �9/Users/sandrine/Deepvest_system/src/core/data_provider.py�__init__ s z FinnhubDataProvider.__init__c 3 C s� t tj�|d��� �}t tj�|d��� �}z@| j�|d||�}|d dkrMt�tj|d dd�|d |d |d |d4 |d d ��}|j d
dd� |W S t5d|� �� W d S t yp } zt6d|� �� W Y d }~d S d }~ww )Nz%Y-%m-%d�D�s�ok�t)�unit�c�h�l�o�v)�date�close�high�low�open�volumer T)�inplacezError fetching data for zError: ) �int�datetime�strptime� timestampr �
stock_candles�pd� DataFrame� to_datetime� set_index�print� Exception) r �symbol�7start_date�end_dateZstart_timestampZ
end_timestamp�data�df�er r r �get_stock_data s, ���z"FinnhubDataProvider.get_stock_dataN)�__name__�8__module__� __qualname__r9 r- r r r r r s r )r r �pandasr! r r r r r �<module> s