Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

✨ Auth: Add type hint for send result

Browse Source
This commit is contained in:
Sam 2022-01-19 22:18:17 +08:00
parent 7178b4b1db
commit 6c078c38d0
Signed by: sam01101
GPG Key ID: 42D7B6D13FF5E611
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pyrogram/session/auth.py
View File

@ -28,6 +28,7 @@ from pyrogram import raw
from pyrogram.connection import Connection
from pyrogram.crypto import aes, rsa, prime
from pyrogram.errors import SecurityCheckMismatch
from pyrogram.raw import types
from pyrogram.raw.core import TLObject, Long, Int
from .internals import MsgId
@ -87,7 +88,7 @@ class Auth:
# Step 1; Step 2
nonce = int.from_bytes(urandom(16), "little", signed=True)
log.debug(f"Send req_pq: {nonce}")
res_pq = await self.send(raw.functions.ReqPqMulti(nonce=nonce))
res_pq: types.ResPQ = await self.send(raw.functions.ReqPqMulti(nonce=nonce))
log.debug(f"Got ResPq: {res_pq.server_nonce}")
log.debug(f"Server public key fingerprints: {res_pq.server_public_key_fingerprints}")
@ -131,7 +132,7 @@ class Auth:
# Step 5. TODO: Handle "server_DH_params_fail". Code assumes response is ok
log.debug("Send req_DH_params")
server_dh_params = await self.send(
server_dh_params: types.ServerDHParamsOk = await self.send(
raw.functions.ReqDHParams(
nonce=nonce,
server_nonce=server_nonce,
@ -191,7 +192,7 @@ class Auth:
encrypted_data = aes.ige256_encrypt(data_with_hash, tmp_aes_key, tmp_aes_iv)
log.debug("Send set_client_DH_params")
set_client_dh_params_answer = await self.send(
set_client_dh_params_answer: types.DhGenOk = await self.send(
raw.functions.SetClientDHParams(
nonce=nonce,
server_nonce=server_nonce,