diff --git a/bot/db_utils.py b/bot/db_utils.py index e87def9..b8d00b1 100644 --- a/bot/db_utils.py +++ b/bot/db_utils.py @@ -166,12 +166,25 @@ def is_player_administrator(username: str) -> bool: def insert_card( - name: str, rarity: int, file_id: str) -> int: + name: str, rarity: int, file_id: str, + power: int =None, charm: int = None, wit: int = None) -> int: '''Inserts a card''' - CURSOR.execute( - 'INSERT INTO cards (name, rarity, file_id) VALUES (?, ?, ?)', - (name, rarity, file_id) - ) + if power is not None and charm is not None and wit is not None: + CURSOR.execute( + ''' + INSERT INTO card (name, rarity, file_id, power, charm, wit) + VALUES (?, ?, ?, ?, ?, ?) + ''', + (name, rarity, file_id, power, charm, wit) + ) + else: + CURSOR.execute( + ''' + INSERT INTO card (name, rarity, file_id) + VALUES (?, ?, ?) + ''', + (name, rarity, file_id) + ) card_id = CURSOR.lastrowid return card_id if card_id else 0