work on functionality to change parameters of charts and refactor some files out

1 files changed, 12 insertions, 5 deletions

diff --git a/main.py b/main.py
index 2a4ddff..5ddd8af 100644
--- a/main.py
+++ b/main.py
@@ -2,19 +2,17 @@ import requests
 import json
 import datetime
 
-print("hello")
-
 """
 First pull data from yahoo api
 """
 params = { #'period1' : '1753487940', 
            #'period2' : '1753725600', 
-           'interval' : '1d', 
-           'range' : '1y',
+           'interval' : '1wk', # 1m, 2m, 5m, 15m, 30m, 60m, 90m, 1h, 4h, 1d, 5d, 1wk, 1mo, 3mo
+           'range' : '5y', # "1d","5d","1mo","3mo","6mo","1y","2y","5y","10y","ytd","max"
            'events' : 'div|split|earn', 
            'includePrePost' : 'false' }
 headers = {'User-agent' : 'fin-backtesting-proj'}
-r = requests.get("https://query2.finance.yahoo.com/v8/finance/chart/AAPL", headers=headers, params=params)
+r = requests.get("https://query2.finance.yahoo.com/v8/finance/chart/SPY", headers=headers, params=params)
 
 print(r.url)
 print("status_code:\t", r.status_code)
@@ -28,6 +26,15 @@ timestamps = data_obj['chart']['result'][0]['timestamp']
 close_prices = data_obj['chart']['result'][0]['indicators']['quote'][0]['close']
 print('close_price len: ', len(close_prices), 'timestamps len: ', len(timestamps))
 
+# clean out null's and 0s
+for i in range(len(timestamps)):
+    if close_prices[i] == None or close_prices[i] == 0:
+        del close_prices[i]
+        del timestamps[i]
+        i -= 1
+
+print('close_price len: ', len(close_prices), 'timestamps len: ', len(timestamps))
+
 # save timestamps and close prices into separate files
 timestamps_encoded = json.dumps(timestamps)
 close_prices_encoded = json.dumps(close_prices)