diff --git a/database.py b/database.py index c78db0fe..a60d10d6 100644 --- a/database.py +++ b/database.py @@ -461,17 +461,6 @@ async def upsert_user( language_code: str = None, is_bot: bool = False, ): - """ - Создает или обновляет информацию о пользователе в базе данных. - - Args: - tg_id (int): Уникальный идентификатор пользователя в Telegram - username (str, optional): Никнейм пользователя - first_name (str, optional): Имя пользователя - last_name (str, optional): Фамилия пользователя - language_code (str, optional): Код языка пользователя - is_bot (bool, optional): Флаг, указывающий является ли пользователь ботом - """ conn = await asyncpg.connect(DATABASE_URL) try: await conn.execute( @@ -494,29 +483,11 @@ async def upsert_user( language_code, is_bot, ) - - # Создаем запись в connections, если ее еще нет - await conn.execute( - """ - INSERT INTO connections (tg_id, balance, trial) - VALUES ($1, 0.0, 0) - ON CONFLICT (tg_id) DO NOTHING - """, - tg_id, - ) finally: await conn.close() async def add_payment(tg_id: int, amount: float, payment_system: str): - """ - Создает запись о платеже в базе данных. - - Args: - tg_id (int): Уникальный идентификатор пользователя в Telegram - amount (float): Сумма платежа - payment_system (str): Система, через которую был совершен платеж - """ conn = await asyncpg.connect(DATABASE_URL) try: await conn.execute( diff --git a/handlers/start.py b/handlers/start.py index 17cd9224..7b296e8f 100644 --- a/handlers/start.py +++ b/handlers/start.py @@ -69,31 +69,23 @@ async def start_command(message: Message, admin: bool = False): logger.info(f"ID пригласившего пользователя: {referrer_tg_id}") except ValueError: logger.error("Ошибка парсинга реферального ID.") - await message.answer("Некорректный реферальный код.") return connection_exists = await check_connection_exists(message.from_user.id) logger.info( f"Результат проверки подключения для user_id {message.from_user.id}: {connection_exists}" ) - if not connection_exists: logger.info( f"Добавляем подключение для пользователя: {message.from_user.id}" ) await add_connection(message.from_user.id) - logger.info( f"Добавляем реферал для пользователя {message.from_user.id}, приглашённым {referrer_tg_id}" ) await add_referral(message.from_user.id, referrer_tg_id) - - await message.answer("Вас пригласил друг, добро пожаловать!") else: - logger.warning( - f"Пользователь {message.from_user.id} уже зарегистрирован." - ) - await message.answer("Вы уже зарегистрированы в системе!") + logger.info(f"Пользователь {message.from_user.id} уже зарегистрирован.") logger.info( f"Проверяем статус пробного периода для user_id {message.from_user.id}"