import pprint from app.db.userdata import ScrobbleTable, UserTable from app.lib.home.recentlyadded import get_recently_added_items from app.lib.home.get_recently_played import get_recently_played from app.lib.recipes import HomepageRoutine from app.store.homepage import HomepageStore class RecentlyPlayed(HomepageRoutine): ITEM_LIMIT = 15 store_key = "recently_played" def __init__(self, userid: int | None = None) -> None: """ The userid is provided when we are running this routine outside a cron job. ie. when a user records a new scrobble. """ self.userids = [userid] if userid else [user.id for user in UserTable.get_all()] # NOTE: When the userid is provided # we need to update the store for that userid only # using the last scrobble entry. self.update_only = userid is not None super().__init__() @property def is_valid(self): return True def run(self): if self.update_only: last_entry = ScrobbleTable.get_last_entry(self.userids[0]) if last_entry: items = get_recently_played( limit=self.ITEM_LIMIT, userid=self.userids[0], _entries=[last_entry] ) try: item = items[0] store_entry = HomepageStore.entries[self.store_key].items[ self.userids[0] ][0] except IndexError: store_entry = None item = None if ( store_entry and item and store_entry.get("type", "") + store_entry.get("hash", "") == item.get("type", "") + item.get("hash", "") ): # If the item is the same as the one in the store # only update the timestamp HomepageStore.entries[self.store_key].items[self.userids[0]][0][ "timestamp" ] = item["timestamp"] else: # Otherwise, insert the new item # and remove the oldest item if there are more than 15 items HomepageStore.entries[self.store_key].items[self.userids[0]].insert( 0, item ) if ( len( HomepageStore.entries[self.store_key].items[self.userids[0]] ) > self.ITEM_LIMIT ): HomepageStore.entries[self.store_key].items[ self.userids[0] ].pop() for userid in self.userids: items = get_recently_played(limit=self.ITEM_LIMIT, userid=userid) HomepageStore.entries[self.store_key].items[userid] = items class RecentlyAdded(HomepageRoutine): ITEM_LIMIT = 15 store_key = "recently_added" @property def is_valid(self): return True def __init__(self): super().__init__() def run(self): items = get_recently_added_items(limit=self.ITEM_LIMIT) # NOTE: Recently added is a global entry # So we don't need a userid HomepageStore.entries[self.store_key].items[0] = items