wowtoken-app/wow-token-app-bot
2
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Once and for all reformat.

Browse Source 
Will be using black code formatter
This commit is contained in:
Emily Doherty
2024-12-08 17:07:26 -08:00
parent 3a06464c29
commit 34badf17eb
25 changed files with 359 additions and 295 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
token_bot/history_manager/history_manager.py
View File

@@ -9,23 +9,31 @@ from token_bot.token_database.region import Region
class HistoryManager:
HIGH_FIDELITY_PERIOD = '72h'
HIGH_FIDELITY_PERIOD = "72h"
def __init__(self, token_db: tdb.Database):
self._history : Dict[Flavor, Dict[Region, History]] = {}
self._tdb : tdb.Database = token_db
self._history: Dict[Flavor, Dict[Region, History]] = {}
self._tdb: tdb.Database = token_db
for flavor in Flavor:
self._history[flavor] = {}
for region in Region:
self._history[flavor][Region(region)] = History(flavor, Region(region))
async def _retrieve_data(self, flavor: Flavor, region: Region) -> List[Tuple[datetime.datetime, int]]:
high_fidelity_time = datetime.datetime.now(tz=datetime.UTC) - datetime.timedelta(hours=72)
async def _retrieve_data(
self, flavor: Flavor, region: Region
) -> List[Tuple[datetime.datetime, int]]:
high_fidelity_time = datetime.datetime.now(
tz=datetime.UTC
) - datetime.timedelta(hours=72)
all_history = await self._tdb.history(flavor, region)
high_fidelity_history = await self._tdb.history(flavor, region, self.HIGH_FIDELITY_PERIOD)
high_fidelity_history = await self._tdb.history(
flavor, region, self.HIGH_FIDELITY_PERIOD
)
final_response = []
def _convert_to_datetime(data: Tuple[str, int]) -> Tuple[datetime.datetime, int]:
def _convert_to_datetime(
data: Tuple[str, int]
) -> Tuple[datetime.datetime, int]:
return datetime.datetime.fromisoformat(data[0]), data[1]
for data_point in all_history:
@@ -33,11 +41,12 @@ class HistoryManager:
if datetime_tuple[0] < high_fidelity_time:
final_response.append(datetime_tuple)
final_response.extend(_convert_to_datetime(data_point) for data_point in high_fidelity_history)
final_response.extend(
_convert_to_datetime(data_point) for data_point in high_fidelity_history
)
return final_response
async def load_data(self):
for flavor in Flavor:
for r in Region:
@@ -48,19 +57,17 @@ class HistoryManager:
await history.add_price(item)
self._history[flavor][region] = history
async def update_data(self, flavor: Flavor, region: Region) -> List[Alert]:
history = self._history[flavor][region]
current_price_data = await self._tdb.current(flavor)
current_region_data = current_price_data[region.value.lower()]
datum = (
datetime.datetime.fromisoformat(current_region_data[0]),
current_region_data[1]
current_region_data[1],
)
if datum != history.last_price_datum:
return await history.add_price(datum)
return []
def get_history(self, flavor, region) -> History:
return self._history[flavor][region]