From 8395c29ae26a268dd7546a8dd5cde39d60d55c40 Mon Sep 17 00:00:00 2001 From: Vladless Date: Wed, 4 Dec 2024 17:25:32 +0300 Subject: [PATCH] fix trial --- handlers/keys/key_management.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/handlers/keys/key_management.py b/handlers/keys/key_management.py index 70701242..fb0f825b 100644 --- a/handlers/keys/key_management.py +++ b/handlers/keys/key_management.py @@ -90,6 +90,10 @@ async def handle_key_creation( if trial_status == 0: expiry_time = current_time + timedelta(days=TRIAL_TIME) logger.info(f"Assigned 1-day trial to user {tg_id}.") + + await session.execute( + "UPDATE connections SET trial = 1 WHERE tg_id = $1", tg_id + ) await create_key(tg_id, expiry_time, state, session, message_or_query) else: builder = InlineKeyboardBuilder() @@ -118,6 +122,7 @@ async def handle_key_creation( await state.set_state(Form.waiting_for_server_selection) + @router.callback_query(F.data.startswith("select_plan_")) async def select_tariff_plan(callback_query: CallbackQuery, state: FSMContext, session: Any): tg_id = callback_query.message.chat.id