Check for free games

database/crud/free_games.py

@@ -0,0 +1,20 @@
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from database.schemas import FreeGame
+
+__all__ = ["add_free_games", "filter_present_games"]
+
+
+async def add_free_games(session: AsyncSession, game_ids: list[int]):
+    """Bulk-add a list of IDs into the database"""
+    games = [FreeGame(free_game_id=game_id) for game_id in game_ids]
+    session.add_all(games)
+    await session.commit()
+
+
+async def filter_present_games(session: AsyncSession, game_ids: list[int]) -> list[int]:
+    """Filter a list of game IDs down to the ones that aren't in the database yet"""
+    query = select(FreeGame.free_game_id).where(FreeGame.free_game_id.in_(game_ids))
+    matches: list[int] = (await session.execute(query)).scalars().all()
+    return list(set(game_ids).difference(matches))

database/schemas.py

@@ -33,6 +33,7 @@ __all__ = [
     "DadJoke",
     "Deadline",
     "EasterEgg",
+    "FreeGame",
     "GitHubLink",
     "Link",
     "MemeTemplate",
@@ -174,6 +175,14 @@ class EasterEgg(Base):
     startswith: bool = Column(Boolean, nullable=False, server_default="1")
 
 
+class FreeGame(Base):
+    """A temporarily free game"""
+
+    __tablename__ = "free_games"
+
+    free_game_id: int = Column(Integer, primary_key=True)
+
+
 class GitHubLink(Base):
     """A user's GitHub link"""