def get_message(sock, msgtype):
""" Read a message from a socket. msgtype is a subclass of
of protobuf Message.
"""
len_buf = socket_read_n(sock, 4)
msg_len = struct.unpack('>L', len_buf)[0]
msg_buf = socket_read_n(sock, msg_len)
msg = msgtype()
TSerialization.deserialize(msg, msg_buf, TJSONProtocol.TJSONProtocolFactory())
return msg
def send_message(sock, message):
""" Send a serialized message (protobuf Message interface)
to a socket, prepended by its length packed in 4
bytes (big endian).
"""
s = TSerialization.serialize(message, TJSONProtocol.TJSONProtocolFactory())
packed_len = struct.pack('>L', len(s))
sock.sendall(packed_len + s)
from _prof_imports import TBinaryProtocol
from _prof_imports import ProfilerResponse
from _prof_imports import IS_PY3K
if __name__ == '__main__':
file_name = sys.argv[1]
stats = pstats.Stats(file_name)
m = ProfilerResponse(id=0)
statsToResponse(stats.stats, m)
data = TSerialization.serialize(m, TBinaryProtocol.TBinaryProtocolFactory())
# setup stdout to write binary data to it
if IS_PY3K:
out = sys.stdout.buffer
elif sys.platform == 'win32':
import os, msvcrt
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
out = sys.stdout
else:
out = sys.stdout
out.write(data)
out.flush()
from _prof_imports import TJSONProtocol
from _prof_imports import ProfilerResponse
from _prof_imports import IS_PY3K
if __name__ == '__main__':
file_name = sys.argv[1]
stats = pstats.Stats(file_name)
m = ProfilerResponse(id=0)
statsToResponse(stats.stats, m)
data = TSerialization.serialize(m, TJSONProtocol.TJSONProtocolFactory())
if IS_PY3K:
data = data.decode("utf-8")
sys.stdout.write(data)
sys.stdout.flush()
import sys
import pstats
from prof_util import statsToResponse
from _prof_imports import TSerialization
from _prof_imports import TJSONProtocol
from _prof_imports import ProfilerResponse
from _prof_imports import IS_PY3K
if __name__ == '__main__':
file_name = sys.argv[1]
stats = pstats.Stats(file_name)
m = ProfilerResponse(id=0)
statsToResponse(stats.stats, m)
data = TSerialization.serialize(m, TJSONProtocol.TJSONProtocolFactory())
if IS_PY3K:
data = data.decode("utf-8")
sys.stdout.write(data)
sys.stdout.flush()
from _prof_imports import TSerialization
from _prof_imports import TBinaryProtocol
from _prof_imports import ProfilerResponse
from _prof_imports import IS_PY3K
if __name__ == '__main__':
file_name = sys.argv[1]
stats = pstats.Stats(file_name)
m = ProfilerResponse(id=0)
statsToResponse(stats.stats, m)
data = TSerialization.serialize(m,
TBinaryProtocol.TBinaryProtocolFactory())
# setup stdout to write binary data to it
if IS_PY3K:
out = sys.stdout.buffer
elif sys.platform == 'win32':
import os, msvcrt
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
out = sys.stdout
else:
out = sys.stdout
out.write(data)
out.flush()
