1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
|
import struct
from .varint import write_varint
MAX_UINT64 = (1 << 64) - 1
MAX_INT64 = (1 << 63) - 1
def _byte(b):
return bytes((b, ))
def write_binary_str(text, buf):
text = text.encode('utf-8')
write_binary_bytes(text, buf)
def write_binary_bytes(text, buf):
write_varint(len(text), buf)
buf.write(text)
def write_binary_int(number, buf, fmt):
"""
Writes int from buffer with provided format.
"""
fmt = '<' + fmt
buf.write(struct.pack(fmt, number))
def write_binary_int8(number, buf):
write_binary_int(number, buf, 'b')
def write_binary_int16(number, buf):
write_binary_int(number, buf, 'h')
def write_binary_int32(number, buf):
write_binary_int(number, buf, 'i')
def write_binary_int64(number, buf):
write_binary_int(number, buf, 'q')
def write_binary_uint8(number, buf):
write_binary_int(number, buf, 'B')
def write_binary_uint16(number, buf):
write_binary_int(number, buf, 'H')
def write_binary_uint32(number, buf):
write_binary_int(number, buf, 'I')
def write_binary_uint64(number, buf):
write_binary_int(number, buf, 'Q')
def write_binary_uint128(number, buf):
fmt = '<QQ'
packed = struct.pack(fmt, (number >> 64) & MAX_UINT64, number & MAX_UINT64)
buf.write(packed)
|