diff --git a/src/utils/data_utils.py b/src/utils/data_utils.py index cdac9f7..645d8dc 100644 --- a/src/utils/data_utils.py +++ b/src/utils/data_utils.py @@ -4,7 +4,6 @@ from datetime import datetime, timedelta from trading.position_calculator import PositionCalculator from utils.common_utils import get_user_input, get_stock_data, get_qualified_stocks from typing import Optional -import requests def get_float_input(prompt: str) -> Optional[float]: return get_user_input(prompt, float) @@ -14,36 +13,27 @@ def get_current_prices(tickers: list) -> dict: if not tickers: return {} - polygon_api_key = os.environ.get("POLYGON_API_KEY") - if not polygon_api_key: - print("Error: POLYGON_API_KEY environment variable not set.") - return {} - prices = {} try: - for ticker in tickers: - url = f"https://api.polygon.io/v2/last/trade/{ticker}?apiKey={polygon_api_key}" - response = requests.get(url) + with create_client() as client: + for ticker in tickers: + query = f""" + SELECT close + FROM stock_db.stock_prices_daily + WHERE ticker = '{ticker}' + ORDER BY date DESC + LIMIT 1 + """ + result = client.query(query) - if response.status_code == 200: - data = response.json() try: - prices[ticker] = data['results']['price'] + prices[ticker] = result.result_rows[0][0] except KeyError: prices[ticker] = 0.0 - else: - print(f"Error fetching price for {ticker}: {response.status_code} - {response.text}") - prices[ticker] = 0.0 - - # Verify that we have prices for all tickers - for ticker in tickers: - if ticker not in prices or prices[ticker] == 0: - print(f"Could not fetch price for {ticker}") - prices[ticker] = 0.0 except Exception as e: print(f"Error fetching prices: {e}") - # If there's an error, fall back to setting price to 0.0 + # If there's an error, set price to 0.0 for ticker in tickers: if ticker not in prices: prices[ticker] = 0.0