mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
Run black over repo
This commit is contained in:
@ -5,14 +5,14 @@ from .constants import HEADER_TAGS
|
||||
|
||||
def encode_uvarint(number):
|
||||
"""Pack `number` into varint bytes"""
|
||||
buf = b''
|
||||
buf = b""
|
||||
while True:
|
||||
towrite = number & 0x7f
|
||||
towrite = number & 0x7F
|
||||
number >>= 7
|
||||
if number:
|
||||
buf += bytes((towrite | 0x80, ))
|
||||
buf += bytes((towrite | 0x80,))
|
||||
else:
|
||||
buf += bytes((towrite, ))
|
||||
buf += bytes((towrite,))
|
||||
break
|
||||
return buf
|
||||
|
||||
@ -22,7 +22,7 @@ def decode_uvarint(buff, index):
|
||||
result = 0
|
||||
while True:
|
||||
i = buff[index]
|
||||
result |= (i & 0x7f) << shift
|
||||
result |= (i & 0x7F) << shift
|
||||
shift += 7
|
||||
if not i & 0x80:
|
||||
break
|
||||
@ -30,19 +30,21 @@ def decode_uvarint(buff, index):
|
||||
|
||||
return result, index + 1
|
||||
|
||||
|
||||
async def decode_uvarint_from_stream(reader, timeout):
|
||||
shift = 0
|
||||
result = 0
|
||||
while True:
|
||||
byte = await asyncio.wait_for(reader.read(1), timeout=timeout)
|
||||
i = struct.unpack('>H', b'\x00' + byte)[0]
|
||||
result |= (i & 0x7f) << shift
|
||||
i = struct.unpack(">H", b"\x00" + byte)[0]
|
||||
result |= (i & 0x7F) << shift
|
||||
shift += 7
|
||||
if not i & 0x80:
|
||||
break
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def get_flag(initiator, action):
|
||||
"""
|
||||
get header flag based on action for mplex
|
||||
|
||||
Reference in New Issue
Block a user