Kr08 commited on
Commit
bc87714
·
verified ·
1 Parent(s): 7057428

Update data_fetcher.py

Browse files
Files changed (1) hide show
  1. data_fetcher.py +3 -3
data_fetcher.py CHANGED
@@ -1,7 +1,7 @@
1
  import yfinance as yf
2
  import pandas as pd
3
  from pytickersymbols import PyTickerSymbols
4
- from config import ticker_dict, START_DATE, END_DATE
5
 
6
  def get_stocks_from_index(idx):
7
  stock_data = PyTickerSymbols()
@@ -10,8 +10,8 @@ def get_stocks_from_index(idx):
10
  stock_names = [f"{stock['name']}:{stock['symbol']}" for stock in stocks]
11
  return stock_names
12
 
13
- def get_stock_data(ticker_name, interval):
14
- series = yf.download(tickers=ticker_name, start=START_DATE, end=END_DATE, interval=interval)
15
  return series.reset_index()
16
 
17
  def get_company_info(ticker):
 
1
  import yfinance as yf
2
  import pandas as pd
3
  from pytickersymbols import PyTickerSymbols
4
+ from config import ticker_dict
5
 
6
  def get_stocks_from_index(idx):
7
  stock_data = PyTickerSymbols()
 
10
  stock_names = [f"{stock['name']}:{stock['symbol']}" for stock in stocks]
11
  return stock_names
12
 
13
+ def get_stock_data(ticker_name, interval, start_date, end_date):
14
+ series = yf.download(tickers=ticker_name, start=start_date, end=end_date, interval=interval)
15
  return series.reset_index()
16
 
17
  def get_company_info(ticker):