mux_client.py

#! /usr/bin/env python

import sys
import socket
import fdpexpect
import atexit

_default_host = 'localhost'
_default_port = 23200

class MuxClient(object):
	def __init__(self, host = _default_host, port = _default_port):
		self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
		self.host = host
		self.port = port
		self.term = None

	def run(self):
		self.sock.connect_ex((self.host, self.port))
		self.term = fdpexpect.fdspawn(self.sock)
		print >>sys.stderr, 'MUX > Connected to %s:%d' % (self.host, self.port)
		return self.term

	def interact(self):
		print >>sys.stderr, 'MUX > Use ctrl+] to stop...\n'
		self.term.interact()

	def close(self):
		print >>sys.stderr, '\nMUX > Closing...'

		if self.term is None:
			self.sock.close()
		else:
			self.term.close() # Closes sock too

		print >>sys.stderr, 'MUX > Done! =)'


if __name__ == '__main__':
	import optparse

	# Option parsing, duh
	parser = optparse.OptionParser()
	parser.add_option('-p', '--port',
					help = 'Host port',
					dest = 'port',
					type = 'int',
					default = _default_port)
	(opts, args) = parser.parse_args()

	client = MuxClient(port=opts.port)
	client.run()
	client.interact()

	if not sys.flags.interactive:
		client.close()