Initial commit — personal Telegram meme bot with inline search

db.py

@@ -0,0 +1,77 @@
+import aiosqlite
+import os
+
+DB_PATH = os.getenv("DB_PATH", "/data/media.db")
+
+
+async def init_db():
+    async with aiosqlite.connect(DB_PATH) as db:
+        await db.execute("""
+            CREATE TABLE IF NOT EXISTS media (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                owner_id INTEGER NOT NULL,
+                file_id TEXT NOT NULL,
+                file_unique_id TEXT NOT NULL UNIQUE,
+                media_type TEXT NOT NULL,
+                keywords TEXT NOT NULL,
+                added_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+            )
+        """)
+        await db.execute(
+            "CREATE INDEX IF NOT EXISTS idx_owner ON media(owner_id)"
+        )
+        await db.commit()
+
+
+async def add_media(owner_id: int, file_id: str, file_unique_id: str, media_type: str, keywords: list[str]):
+    kw = ",".join(k.strip().lower() for k in keywords if k.strip())
+    async with aiosqlite.connect(DB_PATH) as db:
+        await db.execute(
+            "INSERT OR REPLACE INTO media (owner_id, file_id, file_unique_id, media_type, keywords) VALUES (?,?,?,?,?)",
+            (owner_id, file_id, file_unique_id, media_type, kw),
+        )
+        await db.commit()
+
+
+async def search_media(user_id: int, query: str) -> list[dict]:
+    query = query.strip().lower()
+    async with aiosqlite.connect(DB_PATH) as db:
+        db.row_factory = aiosqlite.Row
+        if query:
+            cursor = await db.execute(
+                """
+                SELECT id, file_id, file_unique_id, media_type, keywords
+                FROM media
+                WHERE owner_id = ? AND (',' || keywords || ',') LIKE ?
+                ORDER BY added_at DESC
+                LIMIT 50
+                """,
+                (user_id, f"%,{query}%"),
+            )
+        else:
+            cursor = await db.execute(
+                """
+                SELECT id, file_id, file_unique_id, media_type, keywords
+                FROM media
+                WHERE owner_id = ?
+                ORDER BY added_at DESC
+                LIMIT 50
+                """,
+                (user_id,),
+            )
+        rows = await cursor.fetchall()
+        return [dict(r) for r in rows]
+
+
+async def delete_media(user_id: int, file_unique_id: str) -> bool:
+    async with aiosqlite.connect(DB_PATH) as db:
+        cur = await db.execute(
+            "DELETE FROM media WHERE owner_id = ? AND file_unique_id = ?",
+            (user_id, file_unique_id),
+        )
+        await db.commit()
+        return cur.rowcount > 0
+
+
+async def list_media(user_id: int) -> list[dict]:
+    return await search_media(user_id, "")