Kr08 commited on
Commit
d825735
·
verified ·
1 Parent(s): 0864cc9

Create data_fetcher.py

Browse files
Files changed (1) hide show
  1. data_fetcher.py +34 -0
data_fetcher.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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()
8
+ index = ticker_dict[idx]
9
+ stocks = list(stock_data.get_stocks_by_index(index))
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):
18
+ stock = yf.Ticker(ticker)
19
+ info = stock.info
20
+
21
+ fundamentals = {
22
+ "Company Name": info.get("longName", "N/A"),
23
+ "Sector": info.get("sector", "N/A"),
24
+ "Industry": info.get("industry", "N/A"),
25
+ "Market Cap": f"${info.get('marketCap', 'N/A'):,}",
26
+ "P/E Ratio": round(info.get("trailingPE", 0), 2),
27
+ "EPS": round(info.get("trailingEps", 0), 2),
28
+ "52 Week High": f"${info.get('fiftyTwoWeekHigh', 'N/A'):,}",
29
+ "52 Week Low": f"${info.get('fiftyTwoWeekLow', 'N/A'):,}",
30
+ "Dividend Yield": f"{info.get('dividendYield', 0) * 100:.2f}%",
31
+ "Beta": round(info.get("beta", 0), 2),
32
+ }
33
+
34
+ return pd.DataFrame(list(fundamentals.items()), columns=['Metric', 'Value'])