Update polygon_loader.py
Browse files- polygon_loader.py +22 -17
polygon_loader.py
CHANGED
|
@@ -2,50 +2,55 @@ import requests
|
|
| 2 |
import pandas as pd
|
| 3 |
from datetime import datetime, timedelta
|
| 4 |
|
| 5 |
-
API_KEY = "UUrRnlZXDllgVVn9bekrSTiBYRUqNFwp" # Replace this with your
|
| 6 |
|
| 7 |
def fetch_ohlcv(symbol="AAPL"):
|
| 8 |
-
|
| 9 |
-
start_date = (datetime.utcnow() - timedelta(days=2)).strftime('%Y-%m-%d')
|
| 10 |
end_date = (datetime.utcnow() - timedelta(days=1)).strftime('%Y-%m-%d')
|
| 11 |
|
| 12 |
url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/range/1/minute/{start_date}/{end_date}"
|
| 13 |
-
|
| 14 |
params = {
|
| 15 |
"adjusted": "true",
|
| 16 |
"sort": "asc",
|
| 17 |
-
"limit":
|
| 18 |
"apiKey": API_KEY,
|
| 19 |
}
|
| 20 |
|
| 21 |
-
print(
|
| 22 |
-
print(
|
| 23 |
|
| 24 |
try:
|
| 25 |
-
|
| 26 |
-
print(
|
| 27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
-
|
| 30 |
-
|
| 31 |
|
| 32 |
if "results" not in data:
|
| 33 |
-
print("β 'results' key
|
| 34 |
return None
|
| 35 |
|
| 36 |
if not data["results"]:
|
| 37 |
-
print("β 'results'
|
| 38 |
return None
|
| 39 |
|
| 40 |
df = pd.DataFrame(data["results"])
|
| 41 |
df["t"] = pd.to_datetime(df["t"], unit="ms")
|
| 42 |
df.set_index("t", inplace=True)
|
| 43 |
-
df.rename(columns={
|
|
|
|
|
|
|
|
|
|
| 44 |
df = df[["Open", "High", "Low", "Close", "Volume"]]
|
| 45 |
|
| 46 |
-
print(f"β
|
| 47 |
return df
|
| 48 |
|
| 49 |
except Exception as e:
|
| 50 |
-
print(
|
| 51 |
return None
|
|
|
|
| 2 |
import pandas as pd
|
| 3 |
from datetime import datetime, timedelta
|
| 4 |
|
| 5 |
+
API_KEY = "UUrRnlZXDllgVVn9bekrSTiBYRUqNFwp" # π Replace this with your actual key
|
| 6 |
|
| 7 |
def fetch_ohlcv(symbol="AAPL"):
|
| 8 |
+
start_date = (datetime.utcnow() - timedelta(days=5)).strftime('%Y-%m-%d')
|
|
|
|
| 9 |
end_date = (datetime.utcnow() - timedelta(days=1)).strftime('%Y-%m-%d')
|
| 10 |
|
| 11 |
url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/range/1/minute/{start_date}/{end_date}"
|
|
|
|
| 12 |
params = {
|
| 13 |
"adjusted": "true",
|
| 14 |
"sort": "asc",
|
| 15 |
+
"limit": 5000,
|
| 16 |
"apiKey": API_KEY,
|
| 17 |
}
|
| 18 |
|
| 19 |
+
print("π‘ URL:", url)
|
| 20 |
+
print("π‘ Params:", params)
|
| 21 |
|
| 22 |
try:
|
| 23 |
+
response = requests.get(url, params=params)
|
| 24 |
+
print("π΅ Status Code:", response.status_code)
|
| 25 |
+
|
| 26 |
+
# Show full response if something goes wrong
|
| 27 |
+
if response.status_code != 200:
|
| 28 |
+
print("β Full response:", response.text)
|
| 29 |
+
return None
|
| 30 |
|
| 31 |
+
data = response.json()
|
| 32 |
+
print("π Response JSON:", data)
|
| 33 |
|
| 34 |
if "results" not in data:
|
| 35 |
+
print("β 'results' key not found in response.")
|
| 36 |
return None
|
| 37 |
|
| 38 |
if not data["results"]:
|
| 39 |
+
print("β Empty 'results' list.")
|
| 40 |
return None
|
| 41 |
|
| 42 |
df = pd.DataFrame(data["results"])
|
| 43 |
df["t"] = pd.to_datetime(df["t"], unit="ms")
|
| 44 |
df.set_index("t", inplace=True)
|
| 45 |
+
df.rename(columns={
|
| 46 |
+
"o": "Open", "h": "High", "l": "Low", "c": "Close", "v": "Volume"
|
| 47 |
+
}, inplace=True)
|
| 48 |
+
|
| 49 |
df = df[["Open", "High", "Low", "Close", "Volume"]]
|
| 50 |
|
| 51 |
+
print(f"β
Successfully loaded {len(df)} rows.")
|
| 52 |
return df
|
| 53 |
|
| 54 |
except Exception as e:
|
| 55 |
+
print("β Exception:", str(e))
|
| 56 |
return None
|