Run black over repo

This commit is contained in:
Alex Stokes
2019-07-31 15:00:12 -07:00
parent a2133d8c7c
commit 0ae9840928
69 changed files with 791 additions and 1095 deletions

View File

@ -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