Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Add type annotations to scram util functions (sasl)

Browse source
This commit is contained in:
jesopo 2019-02-06 21:05:20 +00:00
parent 1895ac34d7
commit 90b540d5dc
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
modules/sasl/scram.py
View file

@ -1,12 +1,12 @@
import base64, enum, hashlib, hmac, typing, uuid import base64, enum, hashlib, hmac, typing, uuid
def _scram_nonce(): def _scram_nonce() -> bytes:
return uuid.uuid4().hex.encode("utf8") return uuid.uuid4().hex.encode("utf8")
def _scram_escape(s): def _scram_escape(s: bytes) -> bytes:
return s.replace(b"=", b"=3D").replace(b",", b"=2C") return s.replace(b"=", b"=3D").replace(b",", b"=2C")
def _scram_unescape(s): def _scram_unescape(s: bytes) -> bytes:
return s.replace(b"=3D", b"=").replace(b"=2C", b",") return s.replace(b"=3D", b"=").replace(b"=2C", b",")
def _scram_xor(s1, s2): def _scram_xor(s1: bytes, s2: bytes) -> bytes:
return bytes(a ^ b for a, b in zip(s1, s2)) return bytes(a ^ b for a, b in zip(s1, s2))
class SCRAMState(enum.Enum): class SCRAMState(enum.Enum):