From cce941471fe9c96f910998c50294350ea032c328 Mon Sep 17 00:00:00 2001 From: w Date: Sun, 22 Jun 2025 01:42:08 -0300 Subject: [PATCH] Enhance insert_card function to support optional power, charm, and wit parameters --- bot/db_utils.py | 23 ++++++++++++++++++----- 1 file changed, 18 insertions(+), 5 deletions(-) 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