mitogen/preamble_size.py

33 lines
825 B
Python

"""
Print the size of a typical SSH command line and the bootstrap code sent to new
contexts.
"""
import inspect
import zlib
import mitogen.master
import mitogen.ssh
import mitogen.sudo
broker = mitogen.master.Broker()
router = mitogen.core.Router(broker)
context = mitogen.master.Context(router, 0)
stream = mitogen.ssh.Stream(router, 0, context.key, hostname='foo')
broker.shutdown()
print 'SSH command size: %s' % (len(' '.join(stream.get_boot_command())),)
print 'Preamble size: %s (%.2fKiB)' % (
len(stream.get_preamble()),
len(stream.get_preamble()) / 1024.0,
)
for mod in (
mitogen.master,
mitogen.ssh,
mitogen.sudo,
):
sz = len(zlib.compress(mitogen.master.minimize_source(inspect.getsource(mod))))
print '%s size: %s (%.2fKiB)' % (mod.__name__, sz, sz / 1024.0)