o sg@stdZddlZddlmZmZddlmZddlmZdgZGdddZ e d kr8e dd Z e e j d d d dSdS)z&Get stock data from the OpenBB libraryN)datetime timedelta)obb) OpenBBErroryfinancec@s>eZdZdZd ddZddefd d Zeddefd d ZdS)OpenBBStockDataz&Get Stock data from the OpenBB library1dNrcCsXt|_||jvrtd|dd|j||_||_|s'|jdd|_dS||_dS)au Initialize the class with the interval and start date interval: str: The interval of the stock data (default: 1d, options: 1m, 1h, 1d, 1W, 1M) minute, hour, day, week, month start_date: str: The start date of the stock data (default: 5 years from today) provider: str: The provider of the stock data (default: yfinance) zInvalid provider: z , options: z, )durationN) PROVIDERS provider_list ValueErrorjoinproviderinterval_get_start_date start_date)selfrrrrA/var/www/html/XCapMarket/services/stock_info/openbb_stock_data.py__init__s  zOpenBBStockData.__init__r r c Csz7|std|||_|s|j}tjjj||j|j|jd }dd|j D}|d }||d}|WSty`}zt dtd||jtd |d |jd ||d }~wty}zt dtd||jtd |d |jd ||d }~ww) a Get stock data from the OpenBB library ticker: str: The ticker of the stock start_date: str: The start date of the stock data (default: 5 years from today) duration: int: The duration of the stock data (default: 5 years) z&Ticker is required to fetch stock data)symbolrrrcSsg|]}|dqS)%Y-%m-%d)strftime).0daterrr >sz2OpenBBStockData.get_stock_data..close)r close_pricez8%s: Error fetching stock prices for %s from %s. Error: z%Y/%m/%d %H:%M:%SzError fetching z stock prices from z: N)r rrrequityprice historicalrrto_dfindextolistloggingerrorrnowrr) rtickerrr stock_datadates close_prices stock_dicterrrget_stock_data'sZ     zOpenBBStockData.get_stock_datacCsRt}|td|dd}|dvr"|d}|t|d8}|d}|S)z Get the start date of the stock data duration: int: The duration of the stock data in years (default: 5 years) im)days)r r)rr'rweekdayr)r todayrstart_date_offsetrrrr[s   zOpenBBStockData._get_start_date)rNr)NNr )r ) __name__ __module__ __qualname____doc__rintr. staticmethodrrrrrr s  4r__main__)rr/U)r r()r9r%rropenbbr$openbb_core.app.model.abstract.errorrr rr6 openbb_stockprintr.rrrrs  a