6e84429ad1
Implements a Matrix bot that posts PC gaming deals and free game alerts: - CheapShark API integration (Steam, GOG, Humble, GMG) polled every 2 hours - Epic Games Store free games detection polled daily - IsThereAnyDeal historical low price flagging (optional) - SQLite deduplication with 30-day pruning - First-run population (records existing deals without posting) - Matrix-flavored HTML + plain text fallback messages - Dockerfile for containerized deployment https://claude.ai/code/session_01LPpSZFfyh6vdV5HGFWjoQX
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
import logging
|
|
from datetime import datetime, timedelta, timezone
|
|
|
|
import aiosqlite
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
SCHEMA = """
|
|
CREATE TABLE IF NOT EXISTS posted_deals (
|
|
id TEXT PRIMARY KEY,
|
|
source TEXT NOT NULL,
|
|
title TEXT NOT NULL,
|
|
posted_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS config (
|
|
key TEXT PRIMARY KEY,
|
|
value TEXT
|
|
);
|
|
"""
|
|
|
|
|
|
class Database:
|
|
def __init__(self, path: str):
|
|
self.path = path
|
|
self._db: aiosqlite.Connection | None = None
|
|
|
|
async def connect(self):
|
|
self._db = await aiosqlite.connect(self.path)
|
|
await self._db.executescript(SCHEMA)
|
|
await self._db.commit()
|
|
logger.info("Database initialized at %s", self.path)
|
|
|
|
async def close(self):
|
|
if self._db:
|
|
await self._db.close()
|
|
|
|
async def has_been_posted(self, deal_id: str) -> bool:
|
|
assert self._db is not None
|
|
cursor = await self._db.execute(
|
|
"SELECT 1 FROM posted_deals WHERE id = ?", (deal_id,)
|
|
)
|
|
row = await cursor.fetchone()
|
|
return row is not None
|
|
|
|
async def mark_posted(self, deal_id: str, source: str, title: str):
|
|
assert self._db is not None
|
|
await self._db.execute(
|
|
"INSERT OR IGNORE INTO posted_deals (id, source, title) VALUES (?, ?, ?)",
|
|
(deal_id, source, title),
|
|
)
|
|
await self._db.commit()
|
|
|
|
async def prune_old(self, days: int = 30):
|
|
assert self._db is not None
|
|
cutoff = datetime.now(timezone.utc) - timedelta(days=days)
|
|
result = await self._db.execute(
|
|
"DELETE FROM posted_deals WHERE posted_at < ?",
|
|
(cutoff.isoformat(),),
|
|
)
|
|
await self._db.commit()
|
|
if result.rowcount:
|
|
logger.info("Pruned %d old deal records", result.rowcount)
|
|
|
|
async def get_config(self, key: str) -> str | None:
|
|
assert self._db is not None
|
|
cursor = await self._db.execute(
|
|
"SELECT value FROM config WHERE key = ?", (key,)
|
|
)
|
|
row = await cursor.fetchone()
|
|
return row[0] if row else None
|
|
|
|
async def set_config(self, key: str, value: str):
|
|
assert self._db is not None
|
|
await self._db.execute(
|
|
"INSERT OR REPLACE INTO config (key, value) VALUES (?, ?)",
|
|
(key, value),
|
|
)
|
|
await self._db.commit()
|