peperunas/paperone
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

order order

Browse Source
This commit is contained in:
Giulio De Pasquale 2025-10-11 12:07:10 +01:00
parent 52363eb4f4
commit de0b9086ba
1 changed files with 29 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
paperone.py
View File

@ -5,7 +5,7 @@ import math
from sys import exit from sys import exit
from datetime import datetime from datetime import datetime
from dotenv import load_dotenv from dotenv import load_dotenv
from typing import NoReturn from typing import NoReturn, List
from os import environ from os import environ
from enum import Enum from enum import Enum
from dataclasses import dataclass from dataclasses import dataclass
@ -18,6 +18,12 @@ class Indicator:
endpoint: str endpoint: str
@dataclass
class QueryResult:
datetime: datetime
value: float
class IndicatorEnum(Enum): class IndicatorEnum(Enum):
RSI = Indicator("rsi") RSI = Indicator("rsi")
@ -41,7 +47,7 @@ class TaapiClient:
self._base_url: str = "https://api.taapi.io" self._base_url: str = "https://api.taapi.io"
self._session: requests.Session = requests.Session() self._session: requests.Session = requests.Session()
def __build_indicator_url(self, indicator: Indicator) -> str: def __build_indicator_url__(self, indicator: Indicator) -> str:
return f"{self._base_url}/{indicator.endpoint}" return f"{self._base_url}/{indicator.endpoint}"
def query_indicator( def query_indicator(
@ -51,9 +57,10 @@ class TaapiClient:
target_date: datetime, target_date: datetime,
interval: str = "1d", interval: str = "1d",
results: int = 14, results: int = 14,
) -> requests.Response: ) -> List[QueryResult] | None:
backtrack_candles: int = self._candles_to_target_date(target_date, interval) ret : List[QueryResult] = []
target_url: str = self.__build_indicator_url(indicator) backtrack_candles: int = self.__candles_to_target_date__(target_date, interval)
target_url: str = self.__build_indicator_url__(indicator)
params: dict[str, str | int | bool] = { params: dict[str, str | int | bool] = {
"secret": self._api_key, "secret": self._api_key,
@ -68,10 +75,19 @@ class TaapiClient:
response: requests.Response = self._session.get(target_url, params=params) response: requests.Response = self._session.get(target_url, params=params)
return response if response.status_code != 200:
return None
data: dict[str, list[float] | list[int]] = response.json()
for val, ts in zip(data["value"], data["timestamp"]):
dt: datetime = datetime.fromtimestamp(ts)
ret.append(QueryResult(dt, val))
return ret
@staticmethod @staticmethod
def _candles_to_target_date( def __candles_to_target_date__(
target_date: datetime, target_date: datetime,
interval: str = "1h", interval: str = "1h",
current_time: datetime | None = None, current_time: datetime | None = None,
@ -139,23 +155,16 @@ def main() -> NoReturn:
ticker = "AAPL" ticker = "AAPL"
with TaapiClient(api_key) as client: with TaapiClient(api_key) as client:
response = client.query_indicator(ticker, indicator, date) results = client.query_indicator(ticker, indicator, date)
if response.status_code == 200: if not results:
data: dict[str, list[float] | list[int]] = response.json() print("Could not retrieve stuff")
print(f"{indicator} values for {ticker} on {format_date_readable(date)}:") exit(1)
for rsi_val, ts in zip(data["value"], data["timestamp"]): for r in [x for x in results if is_trading_day(x.datetime)]:
dt: datetime = datetime.fromtimestamp(ts) print(f"{format_date_readable(r.datetime)} : {r.value}")
if not is_trading_day(dt):
continue
print(f"{dt.strftime('%Y-%m-%d %H:%M:%S')} | RSI: {rsi_val:.2f}")
else:
print(f"Error: {response.status_code}")
print(response.text)
exit(0) exit(0)