Fix ref
This commit is contained in:
-29
@@ -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(
|
||||
|
||||
+1
-9
@@ -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}"
|
||||
|
||||
Reference in New Issue
Block a user