Change the way int to bytes conversion is used

delivrance · delivrance · commit 4c9d9d84f289 · 2018-09-01T01:27:22.000+02:00
Maybe at some point I should switch to struct
diff --git a/pyrogram/api/core/primitives/bool.py b/pyrogram/api/core/primitives/bool.py
@@ -30,7 +30,7 @@ def read(cls, *args) -> bool:
         return cls.value
 
     def __new__(cls) -> bytes:
-        return int.to_bytes(cls.ID, 4, "little")
+        return cls.ID.to_bytes(4, "little")
 
 
 class BoolTrue(BoolFalse):
diff --git a/pyrogram/api/core/primitives/bytes.py b/pyrogram/api/core/primitives/bytes.py
@@ -48,7 +48,7 @@ def __new__(cls, value: bytes) -> bytes:
         else:
             return (
                 bytes([254])
-                + int.to_bytes(length, 3, "little")
+                + length.to_bytes(3, "little")
                 + value
                 + bytes(-length % 4)
             )
diff --git a/pyrogram/api/core/primitives/int.py b/pyrogram/api/core/primitives/int.py
@@ -29,7 +29,7 @@ def read(cls, b: BytesIO, signed: bool = True) -> int:
         return int.from_bytes(b.read(cls.SIZE), "little", signed=signed)
 
     def __new__(cls, value: int, signed: bool = True) -> bytes:
-        return int.to_bytes(value, cls.SIZE, "little", signed=signed)
+        return value.to_bytes(cls.SIZE, "little", signed=signed)
 
 
 class Long(Int):
diff --git a/pyrogram/api/core/primitives/null.py b/pyrogram/api/core/primitives/null.py
@@ -29,4 +29,4 @@ def read(b: BytesIO, *args) -> None:
         return None
 
     def __new__(cls) -> bytes:
-        return int.to_bytes(cls.ID, 4, "little")
+        return cls.ID.to_bytes(4, "little")
diff --git a/pyrogram/crypto/rsa.py b/pyrogram/crypto/rsa.py
@@ -206,12 +206,8 @@ class RSA:
 
     @classmethod
     def encrypt(cls, data: bytes, fingerprint: int) -> bytes:
-        return int.to_bytes(
-            pow(
-                int.from_bytes(data, "big"),
-                cls.server_public_keys[fingerprint].e,
-                cls.server_public_keys[fingerprint].m
-            ),
-            256,
-            "big"
-        )
+        return pow(
+            int.from_bytes(data, "big"),
+            cls.server_public_keys[fingerprint].e,
+            cls.server_public_keys[fingerprint].m
+        ).to_bytes(256, "big")
diff --git a/pyrogram/session/auth.py b/pyrogram/session/auth.py
@@ -111,8 +111,8 @@ def create(self):
 
                 data = types.PQInnerData(
                     res_pq.pq,
-                    int.to_bytes(p, 4, "big"),
-                    int.to_bytes(q, 4, "big"),
+                    p.to_bytes(4, "big"),
+                    q.to_bytes(4, "big"),
                     nonce,
                     server_nonce,
                     new_nonce,
@@ -131,17 +131,17 @@ def create(self):
                     functions.ReqDHParams(
                         nonce,
                         server_nonce,
-                        int.to_bytes(p, 4, "big"),
-                        int.to_bytes(q, 4, "big"),
+                        p.to_bytes(4, "big"),
+                        q.to_bytes(4, "big"),
                         public_key_fingerprint,
                         encrypted_data
                     )
                 )
 
                 encrypted_answer = server_dh_params.encrypted_answer
 
-                server_nonce = int.to_bytes(server_nonce, 16, "little", signed=True)
-                new_nonce = int.to_bytes(new_nonce, 32, "little", signed=True)
+                server_nonce = server_nonce.to_bytes(16, "little", signed=True)
+                new_nonce = new_nonce.to_bytes(32, "little", signed=True)
 
                 tmp_aes_key = (
                     sha1(new_nonce + server_nonce).digest()
@@ -170,7 +170,7 @@ def create(self):
                 # Step 6
                 g = server_dh_inner_data.g
                 b = int.from_bytes(urandom(256), "big")
-                g_b = int.to_bytes(pow(g, b, dh_prime), 256, "big")
+                g_b = pow(g, b, dh_prime).to_bytes(256, "big")
 
                 retry_id = 0
 
@@ -199,8 +199,8 @@ def create(self):
 
                 # Step 7; Step 8
                 g_a = int.from_bytes(server_dh_inner_data.g_a, "big")
-                auth_key = int.to_bytes(pow(g_a, b, dh_prime), 256, "big")
-                server_nonce = int.to_bytes(server_nonce, 16, "little", signed=True)
+                auth_key = pow(g_a, b, dh_prime).to_bytes(256, "big")
+                server_nonce = server_nonce.to_bytes(16, "little", signed=True)
 
                 # TODO: Handle errors
 
@@ -235,7 +235,7 @@ def create(self):
                 # 3rd message
                 assert nonce == set_client_dh_params_answer.nonce
                 assert server_nonce == set_client_dh_params_answer.server_nonce
-                server_nonce = int.to_bytes(server_nonce, 16, "little", signed=True)
+                server_nonce = server_nonce.to_bytes(16, "little", signed=True)
                 log.debug("Nonce fields check: OK")
 
                 # Step 9

Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ def __new__(cls, value: bytes) -> bytes:`
`48`	`48`	`else:`
`49`	`49`	`return (`
`50`	`50`	`bytes([254])`
`51`		`- + int.to_bytes(length, 3, "little")`
	`51`	`+ + length.to_bytes(3, "little")`
`52`	`52`	`+ value`
`53`	`53`	`+ bytes(-length % 4)`
`54`	`54`	`)`