diff options
| -rw-r--r-- | db.py | 19 | ||||
| -rw-r--r-- | games.py | 14 |
2 files changed, 31 insertions, 2 deletions
@@ -119,6 +119,25 @@ def remove_item(category: str, name: str) -> str | None: return actual_name +def count_items(category: str) -> int: + with _get_conn() as conn: + row = conn.execute( + "SELECT COUNT(*) AS cnt FROM items WHERE category = ?", (category,) + ).fetchone() + return row["cnt"] + + +def get_item_by_index(category: str, index: int) -> sqlite3.Row | None: + with _get_conn() as conn: + rows = conn.execute( + "SELECT name, note, added_by FROM items WHERE category = ? ORDER BY id ASC", + (category,), + ).fetchall() + if not rows or index < 1 or index > len(rows): + return None + return rows[index - 1] + + def find_item(category: str, keyword: str = "") -> sqlite3.Row | None: """Return a random item from the category, optionally filtered by keyword in note.""" keyword = keyword.strip().lower() @@ -27,8 +27,18 @@ async def handle_game_command(bot_api, room_id: str, sender: str, body: str, sel # Passive trigger: any message containing "what to play" if "what to play" in lower and not lower.startswith("!"): - pick = db.find_item(CATEGORY) - reply = f"Fresh from the list of considerations:\n\n{_format(pick)}" if pick else "No games in the list yet. Use !addgame to add one!" + after = lower.split("what to play", 1)[1].strip() + if after.isdigit(): + index = int(after) + total = db.count_items(CATEGORY) + pick = db.get_item_by_index(CATEGORY, index) + if pick is None: + reply = f"❌ Index {index} is out of range. Valid range: 1–{total}." + else: + reply = f"#{index} from the list of considerations:\n\n{_format(pick)}" + else: + pick = db.find_item(CATEGORY) + reply = f"Fresh from the list of considerations:\n\n{_format(pick)}" if pick else "No games in the list yet. Use !addgame to add one!" await bot_api.send_text_message(room_id, reply) return True |