feat: add redis cache

This commit is contained in:
2024-08-21 22:53:50 +03:00
parent 0638fb8d50
commit d3ef03a6a0
19 changed files with 194 additions and 24 deletions

View File

@@ -1,9 +1,10 @@
import datetime
from ..api import Api
from .model import Schedule
class ScheduleApi:
class ScheduleApi(Api):
async def get_channels(self) -> list[str]:
raise NotImplementedError

View File

@@ -0,0 +1,32 @@
import datetime
from aiocache import cached
from gallery.sketch.cached import CachedApi
from .api import ScheduleApi
from .model import Schedule
class CachedScheduleApi(ScheduleApi, CachedApi[ScheduleApi]):
CACHE_KEY = "schedule"
@cached(
key_builder=lambda fun, self: f"api.{self.CACHE_KEY}.{self.provider}.channels",
alias=CachedApi.CACHE_ALIAS,
ttl=CachedApi.CACHE_TTL,
)
async def get_channels(self) -> list[str]:
return await self._api.get_channels()
@cached(
key_builder=lambda fun, self, channel_id, date: (
f"api.{self.CACHE_KEY}.{self.provider}.channel.{channel_id}.{date}"
),
alias=CachedApi.CACHE_ALIAS,
ttl=CachedApi.CACHE_TTL,
)
async def get_channel_schedule(
self, channel_id: str, date: datetime.date
) -> Schedule:
return await self._api.get_channel_schedule(channel_id, date)