Reorganize code.

This commit is contained in:
Roman Zeyde
2014-06-19 18:58:13 +03:00
parent 1ad96d9716
commit b15f891aa0
4 changed files with 22 additions and 53 deletions

View File

@@ -22,23 +22,18 @@ scaling = 10000.0
LENGTH_FORMAT = '<I'
def pack(data):
log.info('Sending {} bytes, checksum: {}'.format(len(data), checksum(data)))
log.info('Sending {} bytes'.format(len(data)))
data = struct.pack(LENGTH_FORMAT, len(data)) + data
return data
def unpack(data):
length_size = struct.calcsize(LENGTH_FORMAT)
length, data = data[:length_size], data[length_size:]
length, = struct.unpack(LENGTH_FORMAT, length)
data = data[:length]
log.info('Decoded {} bytes, leftover: {}, checksum: {}'.format(len(data), len(data)-length, checksum(data)))
log.info('Decoded {} bytes, leftover: {}'.format(len(data), len(data)-length))
return data
def checksum(data):
return '\033[0;32m{}\033[0m'.format(hashlib.sha256(data).hexdigest())
def to_bits(chars):
for c in chars:
val = ord(c)