kevin/neovim
1
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2024-12-19 18:55:14 -07:00
Code Issues Packages Projects Releases Wiki Activity

scripts: Improve shadacat to work with unknown items

Browse Source
This commit is contained in:
ZyX 2015-08-08 05:10:19 +03:00
parent d8b0cd5c17
commit 1e06792099
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
scripts/shadacat.py
View File

@ -1,5 +1,6 @@
#!/usr/bin/env python3.4
import os
import sys
import codecs
@ -58,18 +59,28 @@ def mnormalize(o):
return ctable.get(type(o), idfunc)(o)
with open(sys.argv[1], 'rb') as fp:
fname = sys.argv[1]
poswidth = len(str(os.stat(fname).st_size or 1000))
with open(fname, 'rb') as fp:
unpacker = msgpack.Unpacker(file_like=fp, read_size=1)
max_type = max(typ.value for typ in EntryTypes)
while True:
try:
pos = fp.tell()
typ = EntryTypes(unpacker.unpack())
typ = unpacker.unpack()
except msgpack.OutOfData:
break
else:
timestamp = unpacker.unpack()
time = datetime.fromtimestamp(timestamp)
length = unpacker.unpack()
entry = unpacker.unpack()
print('{0:4} {1:13} {2} {3:5} {4!r}'.format(
pos, typ.name, time.isoformat(), length, mnormalize(entry)))
if typ > max_type:
entry = fp.read(length)
typ = EntryTypes.Unknown
else:
entry = unpacker.unpack()
typ = EntryTypes(typ)
print('%*u %13s %s %5u %r' % (
poswidth, pos, typ.name, time.isoformat(), length, mnormalize(entry)))