python将uint型数字转化成bytes类型

前提

这里讨论的数字为uint类型

代码

import array

if hasattr(array.array, "tobytes"):
    def array_tobytes(arry):
        return arry.tobytes()


    def array_frombytes(arry, bs):
        return arry.frombytes(bs)
else:
    def array_tobytes(arry):
        return arry.tostring()


    def array_frombytes(arry, bs):
        return arry.fromstring(bs)


def _varint(i):
    a = array.array("B")
    while (i & ~0x7F) != 0:
        a.append((i & 0x7F) | 0x80)
        i = i >> 7
    a.append(i)
    return array_tobytes(a)


v = _varint(1)
v2 = _varint(513)
print(v, v2)