Skip to content

Commit 709a303

Browse files
authored
Fix get default session (#226)
1 parent c25d88c commit 709a303

3 files changed

Lines changed: 14 additions & 2 deletions

File tree

pyrogram/methods/auth/disconnect.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,5 +37,6 @@ async def disconnect(
3737

3838
await self.session.stop()
3939
await self.storage.close()
40+
self.session = None
4041
self.is_connected = False
4142
self._is_server_time_synced = False

pyrogram/methods/auth/send_code.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,11 @@ async def send_code(
109109
await self.storage.server_address(dc_option.ip_address)
110110
await self.storage.port(dc_option.port)
111111

112-
self.session = await self.get_session(temporary=True, export_authorization=False)
112+
self.session = await self.get_session(
113+
server_address=await self.storage.server_address(),
114+
port=await self.storage.port(),
115+
export_authorization=False,
116+
temporary=True
117+
)
113118
else:
114119
return types.SentCode._parse(r)

pyrogram/methods/auth/sign_in_bot.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,13 @@ async def sign_in_bot(
6464
await self.storage.server_address(dc_option.ip_address)
6565
await self.storage.port(dc_option.port)
6666

67-
self.session = await self.get_session(temporary=True, export_authorization=False)
67+
self.session = await self.get_session(
68+
server_address=await self.storage.server_address(),
69+
port=await self.storage.port(),
70+
export_authorization=False,
71+
temporary=True
72+
)
73+
6874
else:
6975
await self.storage.user_id(r.user.id)
7076
await self.storage.is_bot(True)

0 commit comments

Comments
 (0)