Refactor wallet attachment URL in send_transaction method.

nessshon · Jan 24, 2024 · 03d19b4 · 03d19b4
1 parent e5ef89d
commit 03d19b4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/aiogram_tonconnect/manager.py b/aiogram_tonconnect/manager.py
@@ -15,7 +15,7 @@
 from aiogram.utils.markdown import hide_link
 from redis.asyncio import Redis
 
-from .pytonconnect.exceptions import (
+from pytonconnect.exceptions import (
     UserRejectsError,
     WalletNotConnectedError,
 )
@@ -132,9 +132,7 @@ async def connect_wallet(
             await self.send_message(self.__emoji)
 
         if self.tonconnect.connected:
-            with suppress(WalletNotConnectedError):
-                await self.tonconnect.restore_connection()
-                await self.tonconnect.disconnect()
+            await self.disconnect_wallet()
 
         if callbacks:
             await self.connect_wallet_callbacks_storage.add(callbacks)
@@ -228,7 +226,9 @@ async def send_transaction(
         )
         universal_url = self.user.app_wallet.universal_url
         if self.user.app_wallet.app_name == "telegram-wallet":
-            universal_url = universal_url.replace("attach=wallet", "startattach=tonconnect")
+            universal_url = universal_url.replace(
+                "attach=wallet", "startattach=tonconnect-ret__back"
+            )
         reply_markup = self.__inline_keyboard.send_transaction(
             self.user.app_wallet.name, universal_url,
         )