waifu/kemoverse
waifu/kemoverse
3
2
Fork 3
Code Issues 11 Pull requests 2 Projects 1 Releases Packages Wiki Activity

get cards function

Browse source
This commit is contained in:
w 2025-06-25 00:21:37 -03:00
parent cce941471f
commit 8c5c860ef6
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
bot/db_utils.py
View file

@ -1,5 +1,5 @@
# Kemoverse - a gacha-style bot for the Fediverse.
# Copyright © 2025 Waifu VD15, Moon, and contributors.
# Copyright © 2025 Waifu, VD15, Moon, and contributors.
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
@ -75,6 +75,19 @@ def get_random_card() -> Card | None:
'image_url': chosen['file_id']
}
def get_cards(card_ids: list[int]) -> list[tuple]:
'''
Retrieves stats for a list of card IDs.
Returns a list of tuples: (id, name, rarity, file_id, power, charm, wit, ...)
'''
if not card_ids:
return []
placeholders = ','.join('?' for _ in card_ids)
query = f'SELECT * FROM cards WHERE id IN ({placeholders})'
CURSOR.execute(query, card_ids)
return CURSOR.fetchall()
def get_player(username: str) -> int:
'''Retrieve a player ID by username, or return None if not found.'''