From 7b13cf92eab2da45b301e4892115f011b658cdb3 Mon Sep 17 00:00:00 2001 From: Vlad Date: Sat, 21 Sep 2024 01:21:35 +0300 Subject: [PATCH] =?UTF-8?q?=D0=BE=D1=81=D0=BD=D0=BE=D0=B2=D0=BD=D1=8B?= =?UTF-8?q?=D0=B5=20=D1=84=D1=83=D0=BD=D0=BA=D1=86=D0=B8=D0=B8=20=D0=B7?= =?UTF-8?q?=D0=B0=D0=B2=D0=B5=D1=80=D1=88=D0=B5=D0=BD=D1=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- handlers/__pycache__/keys.cpython-310.pyc | Bin 9110 -> 9025 bytes handlers/keys.py | 74 ++++++++-------------- 2 files changed, 28 insertions(+), 46 deletions(-) diff --git a/handlers/__pycache__/keys.cpython-310.pyc b/handlers/__pycache__/keys.cpython-310.pyc index f452fd24cb1c475621ef2c0cf1a98c38622492fc..cebee691462da035fdc74c4d6f5a4f3ba4e29513 100644 GIT binary patch delta 2957 zcma)8S!^3c7@irg*K2#dUY|)E+qv3M6T*>}t1Ue!v}sF251|MK0FpR7CY)qSa`+EQo4EJa_Fq#TrxyIY!?FvtJHdX%x$grg7D;S^>J`WU9|`EgZR!32nm42 z;Pi>7k9uMa3Q8&%+wtfSPKOZ;gqsm=LD1T7#o28LBM55&N-5ES^K}S!BXl9$j&K*k zc7z@3V(cEeL46t9X5NE>O={~vN^Nq=bX+~2D>z#~SYl)10erYsxrzMhr*OIrV1fcP zK4_K+<0_2lXNlUE$gA7pBkOorQc23;azqArWX{-0I2k3BNK%W=lg5e@2xfFlbi;0O zpE@6Z^8wKZ^O7x=heqtfHyDDBhK|chA<;1~mSUiMcXub+J)Hz#XDsGYgQclMtv~bX z3%9@N=ONUr{aketr`P#ugA{3@{oJVbCW@&^&!`1GbDq|{@dGpa)!9UbZdI=(da9db zpl0)+Wb>@)t?65pAP)gDZuyoMD?2>QBRo3C{4lKIGCW16D7WXRAMxCBLRvbC$#P0Y zc&ru0g9NmvTXC#d%h#vKjR#4k-ymM*^edG3QA%dAsE|7{`Z~G58ftlbE}-7Z_H|`j z%5+lPMp72eCS5x4EBb!z!nG^c-iPs?a4|@EgdPOUnL3}kW9=o@B7do4#U$3*jk_?W zV!!%6)wfe;_WzPy=h$xJc^k#m=5)8a0Tn-i&_?b`y4S?qPLC)zm+6nN2yh`uO=>bD z-~cmg&$#>Pr3KpH#waL4=0@7Q^1rl8>S#xhObmZ{Uhbt44LUXd-W`F4FrS|~|yI1EZqiNUp zIZbs_`ol39NQ}Rd%=)uvurr#FYJYApu?6%)LtZpN|KzFhIu*?i@vpb3%eex($e{P; z)E+CXe#_a_b9Bb?f;H0%HTzR==2sc4SG+;nkM6E;C@^i zn31#!XAf8?fLX!mkU(MsB~QYLl5U1wyaTlkXUm=2LJ6D+%e*}W{#FK{o(t#b=~1ZYls*(Nqj{NX9tul&RfC1T(PA^7x;2#w z^@{!mF970xfNzR8JBrY(0x^oqG6L3liHe8Si-kROLVZ>^MEe%*E^egzR;cF1z|l(M z?t`1_rvuKKB`uEQa;0PcR&}+f->e@X>f4^e_DOuRRln(0`7RnKpd$6PXwl(^Pfw3~ z&x}t#=S^04+cl}~jylkLXRdXSJc&xD5ZbEw9XENrUX-X8D{}f#0K9GVQyb107fKq= z8D>$7y3}I2=!?0pT4nKz>X9;)OcF`UtQ_F9RhYUJ!%a@-!dNMIr9l0*sPB#Xa+wC5LPG{;c)9!o#xzEr zohec=f{n6N6l>HA{V(-whJ_eMID&u|E^z{(iZG%!4h)Xq&DA0<1jGgRY&~2EiXBK& RdkDrb3>(HEj5RRs{2Tv1Kq3GD delta 2977 zcmaJ@O>7(25#D+G!zH;~E|Py;HwBZQn>FVha8p`Qo2f{`AC$b_r&GfA9p9ts1N6t}s5rM{jAf zWj@9$x0Klt4(FBX_~t77O@->_OZpVR_~tDm2XI_YW=or&MQWN|)kFnVigAKH@_;qt z44!9DxO_(5(+|o`YhP(E?b%0=Lg?zvrE80x*iVA*$Saw8+zXir^`NA%gD`JV|hp;FSDj;usu}e@#pqM@eu@4%;obfL@gC@1vl~ zX2cnK@r<0!mi8{v@&v*>AoTypBm=KM;?-PIgo%= z+KF)0XB(_1(|k>wlkcWRPoBhg7S{*wK=BeC8-rlN4dN8%b>iX`~<_=GJj?^^zgm{D-`0?hSCo>SCFZ(88J*g zPj9~GIw0?*pV(h@cMW;SINECwyd)1~DlZ0%yGYLio&^+(Zll-^b`@}YH`2sBE#7G0 zVtIZZ*GR8FWTw7H31|{~*_z25V03}YG-g6fjj0-QSq3u9RoU8vYO@J2*VxytDy}?~ z(dWq`%4h{nNIzSK4koS9e!*AjaZE&{pEe2l?BJx5FI^Do@^qmg%K@Snrb~${jy?M|qW}JhS8a1~-P44d|q~y8(9;p1!NM zVqV5q`X3-WvUMyGKgKh@$+Ne$e$IB9+c3#|`C&9YoWJ>JWJR1<(Q`BB#{*V?%;qaY z*1*Xi^Z9<}<+=Q{af%{#TK+o!GFv_*j~9kyqOg+sl30KL?)~50{|v8B#f$P61zC6w zg{#`p$;-#ZSzK0`pq7PwY;x$S_HNa>lRC6pC-@OUH|i~7(R!DzFNJN3_!H7;N0mqD z#SaO7OyA{s?OM&N%`J)An~OJ=n?dQ%Qtb~|NsX^+eLEKg`QEkWmxZV`yg9N)P{Zp+ zE?>HKsWD#@NAL;x`ABgJPRWbK;r%X>GUIArAQiy|^S`#iP@;Y9GV_17R)8KkWv93= z)oYbq1GU#y=E6o}{DoAWkpC{0+0Pkv#iBfBX4+iyG@q?QMrmmtwhLeNwRII!`XA5e zL_D+4+m^oi4`1&@v8AZ}X3;Y63u7IylNuj@0HWED(u&|cf_Kx4b>bUJ2Wh}f)Q>I0 zk6s1Aku2DXcocQFd#xB*rgxG9zVwPh^G(n3H6%JvA0ah)z!R;w z=WZjvFaSt}7drMMeB+4z+86YT}??FCq1JdHJ!B+KeHmOv6q zpx0D`irvk)oR{|s>#XH?L%xHhqxum)wyw8QeoB1h6;Tq8E9+ZY(7|Dz^J5sltrLj1 z^XQ#cRInxsch#1=i;ZL+gywmrdKuNafq7d>u#1;EDCa}{#9(KS_GHC@Y-fzIlJ&#e zgT8{jsWGZ4-pB`V^Jy&YbkOia<->kDZ20oMa%I{cq$()wQdrs#=pYt9KzJ}hs|5n; zz5(OKXSDv|>PrG|y$OVLgy#XRwu)L@ zqsL$h3Cmedjg1@Oq#}PhRxZuc`yF+f>7Ul`kBvHmX(Y5*em?eO@iqGDC;=y^5%h{P zS($88Q8j4~9UkLg;eYP{0TT>7LQHVLWI^@rE&kn6ZPmp3wXuqG3@bVt`s4uzRqX$SKGIsFzOca=JbtjFpquC=-zN;nLdOt}j3#b&UF zZViA{W83&fdb@V7)AeVYt<+%Vf=MoeyF_M;FiUV9A%U=@9p)?B=aj%en5n9P(m~2l5X_Ik?+1wvjXNd1D+EgfYXszjcw9ah-v@`Jvu7;*mHi0%n^9T- diff --git a/handlers/keys.py b/handlers/keys.py index a3281f4c..e67205ee 100644 --- a/handlers/keys.py +++ b/handlers/keys.py @@ -26,7 +26,7 @@ async def process_callback_view_keys(callback_query: types.CallbackQuery): # Создаем кнопки для каждого ключа buttons = [] for record in records: - key_name = record['email'] # Предполагается, что email - это название ключа + key_name = record['email'] client_id = record['client_id'] button = types.InlineKeyboardButton(text=key_name, callback_data=f'view_key_{key_name}_{client_id}') buttons.append([button]) @@ -100,63 +100,63 @@ async def process_callback_view_key(callback_query: types.CallbackQuery): await callback_query.answer() - # Обработка запроса на удаление ключа @router.callback_query(lambda c: c.data.startswith('delete_key_')) async def process_callback_delete_key(callback_query: types.CallbackQuery): tg_id = callback_query.from_user.id - client_id = callback_query.data.split('_')[2] # Получаем client_id + client_id = callback_query.data.split('_')[2] confirmation_keyboard = types.InlineKeyboardMarkup(inline_keyboard=[ [types.InlineKeyboardButton(text='Да, удалить', callback_data=f'confirm_delete_{client_id}')], [types.InlineKeyboardButton(text='Нет, отменить', callback_data='view_keys')] ]) - await bot.send_message(tg_id, "Вы уверены, что хотите удалить ключ?", reply_markup=confirmation_keyboard) + await bot.edit_message_text("Вы уверены, что хотите удалить ключ?", chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=confirmation_keyboard) await callback_query.answer() @router.callback_query(lambda c: c.data.startswith('confirm_delete_')) async def process_callback_confirm_delete(callback_query: types.CallbackQuery): tg_id = callback_query.from_user.id - client_id = callback_query.data.split('_')[2] # Получаем client_id + client_id = callback_query.data.split('_')[2] try: conn = await asyncpg.connect(DATABASE_URL) try: - # Получаем email для удаления через API record = await conn.fetchrow('SELECT email FROM keys WHERE client_id = $1', client_id) if record: email = record['email'] - # Создаем сессию для API-запросов session = login_with_credentials(ADMIN_USERNAME, ADMIN_PASSWORD) - - # Удаляем клиента через API success = delete_client(session, client_id) if success: - # Удаляем ключ из базы данных await conn.execute('DELETE FROM keys WHERE client_id = $1', client_id) - await bot.send_message(tg_id, "Ключ был успешно удален.") + response_message = "Ключ был успешно удален." else: - await bot.send_message(tg_id, "Ошибка при удалении клиента через API.") + response_message = "Ошибка при удалении клиента через API." else: - await bot.send_message(tg_id, "Ключ не найден или уже удален.") + response_message = "Ключ не найден или уже удален." + + back_button = types.InlineKeyboardButton(text='Назад', callback_data='view_keys') + keyboard = types.InlineKeyboardMarkup(inline_keyboard=[[back_button]]) + + await bot.edit_message_text(response_message, chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=keyboard) finally: await conn.close() except Exception as e: - await bot.send_message(tg_id, f"Ошибка при удалении ключа: {e}") + await bot.edit_message_text(f"Ошибка при удалении ключа: {e}", chat_id=tg_id, message_id=callback_query.message.message_id) await callback_query.answer() + @router.callback_query(lambda c: c.data.startswith('renew_key_')) async def process_callback_renew_key(callback_query: types.CallbackQuery): tg_id = callback_query.from_user.id - client_id = callback_query.data.split('_')[2] # Получаем client_id + client_id = callback_query.data.split('_')[2] try: conn = await asyncpg.connect(DATABASE_URL) @@ -169,10 +169,9 @@ async def process_callback_renew_key(callback_query: types.CallbackQuery): current_time = datetime.utcnow().timestamp() * 1000 if expiry_time <= current_time: - await callback_query.message.answer("Ваш ключ уже истек и не может быть продлен.") + await bot.edit_message_text("Ваш ключ уже истек и не может быть продлен.", chat_id=tg_id, message_id=callback_query.message.message_id) return - # Создаем клавиатуру для выбора плана продления keyboard = types.InlineKeyboardMarkup(inline_keyboard=[ [types.InlineKeyboardButton(text='Продлить на 1 месяц (100 руб.)', callback_data=f'renew_1_month_{client_id}')], [types.InlineKeyboardButton(text='Продлить на 3 месяца (250 руб.)', callback_data=f'renew_3_months_{client_id}')], @@ -184,31 +183,27 @@ async def process_callback_renew_key(callback_query: types.CallbackQuery): f"Баланс: {balance} руб.\n" f"Действующий ключ истекает {datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S')}") - await delete_previous_message(callback_query) - await bot.send_message(tg_id, response_message, parse_mode="HTML", reply_markup=keyboard) + await bot.edit_message_text(response_message, chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=keyboard, parse_mode="HTML") finally: await conn.close() except Exception as e: - await callback_query.message.answer(f"Ошибка при выборе плана: {e}") + await bot.edit_message_text(f"Ошибка при выборе плана: {e}", chat_id=tg_id, message_id=callback_query.message.message_id) await callback_query.answer() - # Обработка выбора плана продления @router.callback_query(lambda c: c.data.startswith('renew_')) async def process_callback_renew_plan(callback_query: types.CallbackQuery): tg_id = callback_query.from_user.id plan, client_id = callback_query.data.split('_')[1], callback_query.data.split('_')[3] # '1' или '3' и client_id - days_to_extend = 30 * int(plan) # 30 дней или 90 дней - print(plan, client_id, days_to_extend) + days_to_extend = 30 * int(plan) try: conn = await asyncpg.connect(DATABASE_URL) try: record = await conn.fetchrow('SELECT email, expiry_time FROM keys WHERE client_id = $1', client_id) - print(record) if record: email = record['email'] @@ -216,58 +211,45 @@ async def process_callback_renew_plan(callback_query: types.CallbackQuery): current_time = datetime.utcnow().timestamp() * 1000 if expiry_time <= current_time: - await callback_query.message.answer("Ваш ключ уже истек и не может быть продлен.") + await bot.edit_message_text("Ваш ключ уже истек и не может быть продлен.", chat_id=tg_id, message_id=callback_query.message.message_id) return - # Рассчитываем новый срок окончания new_expiry_time = int(expiry_time + timedelta(days=days_to_extend).total_seconds() * 1000) - - # Определяем стоимость продления cost = 100 if plan == '1' else 250 - # Проверка баланса balance = await get_balance(tg_id) if balance < cost: replenish_button = types.InlineKeyboardButton(text='Пополнить баланс', callback_data='replenish_balance') back_button = types.InlineKeyboardButton(text='Назад', callback_data='view_keys') keyboard = types.InlineKeyboardMarkup(inline_keyboard=[[replenish_button], [back_button]]) - await callback_query.message.answer("Недостаточно средств для продления ключа.", reply_markup=keyboard) + await bot.edit_message_text("Недостаточно средств для продления ключа.", chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=keyboard) return - # Создаем сессию для API-запросов session = login_with_credentials(ADMIN_USERNAME, ADMIN_PASSWORD) - - # Обновляем ключ через API success = extend_client_key(session, tg_id, client_id, email, new_expiry_time) if success: - await update_balance(tg_id, -cost) # Списание средств с баланса + await update_balance(tg_id, -cost) await conn.execute('UPDATE keys SET expiry_time = $1 WHERE client_id = $2', new_expiry_time, client_id) response_message = f"Ваш ключ был успешно продлен на {days_to_extend // 30} месяц(-)." back_button = types.InlineKeyboardButton(text='Назад', callback_data='view_keys') keyboard = types.InlineKeyboardMarkup(inline_keyboard=[[back_button]]) - await bot.send_message(tg_id, response_message, reply_markup=keyboard) + await bot.edit_message_text(response_message, chat_id=tg_id, message_id=callback_query.message.message_id, reply_markup=keyboard) else: - await bot.send_message(tg_id, "Ошибка при продлении ключа.") + await bot.edit_message_text("Ошибка при продлении ключа.", chat_id=tg_id, message_id=callback_query.message.message_id) else: - await bot.send_message(tg_id, "У вас нет ключей для продления.") + await bot.edit_message_text("У вас нет ключей для продления.", chat_id=tg_id, message_id=callback_query.message.message_id) finally: await conn.close() except Exception as e: - await bot.send_message(tg_id, f"Ошибка при продлении ключа: {e}") + await bot.edit_message_text(f"Ошибка при продлении ключа: {e}", chat_id=tg_id, message_id=callback_query.message.message_id) await callback_query.answer() - -# Удаление предыдущего сообщения -async def delete_previous_message(callback_query: types.CallbackQuery): - if callback_query.message.message_id: - await bot.delete_message(chat_id=callback_query.from_user.id, message_id=callback_query.message.message_id) - # Обработка ошибок async def handle_error(tg_id, callback_query, message): - await delete_previous_message(callback_query) - await bot.send_message(tg_id, message) + await bot.edit_message_text(message, chat_id=tg_id, message_id=callback_query.message.message_id) +