add sessionmanager.py more complete aggregate of the session code

python/sessionmanager.py

@@ -0,0 +1,167 @@
+#!/usr/bin/env python
+
+import libjack
+import state
+import subprocess
+from optparse import OptionParser
+from ConfigParser import SafeConfigParser
+import os
+
+SESSION_PATH="/home/torbenh/jackSessions/"
+
+defaults = { "jackclientname": "sessionmanager", "sessiondir": "~/jackSessions" }
+
+class SessionManager( object ):
+    def __init__( self ):
+	self.config = SafeConfigParser( defaults )
+	self.config.read( os.path.expanduser( "~/.jacksessionrc" ) )
+	self.jackname = self.config.get( "DEFAULT", "jackclientname" )
+	self.cl = libjack.JackClient( self.jackname )
+	if self.config.has_section( "infra" ):
+	    self.infra_clients = {}
+	    for inf in self.config.items( "infra" ):
+		self.infra_clients[inf[0]] = inf[1]
+	else:
+	    self.infra_clients = { "a2j": "a2jmidid" }
+	self.implicit_clients = [ "system" ]
+
+	self.sessiondir = os.path.expanduser( self.config.get( "DEFAULT", "sessiondir" ) ) + "/" 
+	if not os.path.exists( self.sessiondir ):
+	    print "Sessiondir %s does not exist. Creating it..."%self.sessiondir
+	    os.mkdir( self.sessiondir )
+
+
+    def load_session( self, name ):
+	if not os.path.exists( self.sessiondir+name+"/session.xml" ):
+	    print "Session %s does not exist"%name
+	    return
+
+	sd = state.SessionDom( self.sessiondir+name+"/session.xml" )
+
+	g=self.cl.get_graph()
+
+	children = []
+
+	for ic in sd.get_infra_clients():
+	    if not ic[0] in g.clients.keys():
+		children.append( subprocess.Popen( ic[1], shell=True ) )
+
+
+	print sd.get_client_names()
+
+	g.ensure_clientnames( sd.get_reg_client_names() )
+
+	# get graph again... renaming isnt prefect yet.
+	g=self.cl.get_graph()
+	# build up list of port connections
+
+	conns = []
+	for p in sd.get_port_names():
+	    for c in sd.get_connections_for_port( p ):
+		conns.append( (p,c) )
+	print conns
+
+	# now fire up the processes
+	for cname in sd.get_reg_client_names():
+	    cmd = sd.get_commandline_for_client( cname )
+	    children.append( subprocess.Popen( cmd, shell=True ) )
+
+	avail_ports = g.get_port_list()
+	# wait for ports to appear, and connect em.
+
+	while len(conns) > 0:
+	    p = self.cl.port_queue.get()
+	    print p[0],p[1]
+	    pname = libjack.port_name(p[0])
+	    for c1 in conns:
+		if c1[0]==pname:
+		    if c1[1] in avail_ports:
+			self.cl.connect( pname, c1[1] )
+
+			conns.remove( c1 )
+			if (c1[1],c1[0]) in conns:
+			    conns.remove( (c1[1],c1[0]) )
+
+		    break
+
+		if c1[1]==pname:
+		    if c1[0] in avail_ports:
+			self.cl.connect( pname, c1[0] )
+
+			conns.remove( c1 )
+			if (c1[1],c1[0]) in conns:
+			    conns.remove( (c1[1],c1[0]) )
+		
+		    break
+
+	    avail_ports.append( pname )
+	    
+
+	print "session restored..."
+
+
+    def save_session( self, name ): 
+	if os.path.exists( self.sessiondir+name ):
+	    print "session %s already exists"
+	    return
+	os.mkdir( self.sessiondir+name )
+	g=self.cl.get_graph()
+	notify = self.cl.session_save( self.sessiondir+name+"/" )
+
+	for n in notify:
+	    c = g.get_client( n.clientname )
+	    c.set_commandline( n.commandline )
+
+	sd = state.SessionDom()
+
+	for c in g.clients.values():
+	    if c.get_commandline() == "":
+		if not c.name in self.infra_clients.keys()+self.implicit_clients:
+		    g.remove_client( c.name )
+		elif c.name in self.implicit_clients:
+		    g.remove_client_only( c.name )
+		else:
+		    c.set_infra( self.infra_clients[c.name] )
+
+
+	for i in g.clients.values():
+	    sd.add_client(i)
+
+	f = file( self.sessiondir+name+"/session.xml", "w" )
+	f.write( sd.get_xml() )
+	f.close()
+
+	print sd.get_xml()
+
+    def exit( self ):
+	self.cl.close()
+
+oparser = OptionParser()
+#oparser.add_option( "--dbus", action="store_true", dest="dbus", default=False,
+#		    help="Use DBUS to issue commands to a running instance" )
+#oparser.add_option( "--save", action="store_true", dest="save", default=False,
+#		    help="Tell SessionManger to save." )
+oparser.add_option( "--saveas", action="store", type="string", dest="saveas",
+		    help="Save Session As <name>" )
+
+#oparser.add_option( "--quit", action="store_true", dest="quit", default=False,
+#		    help="Tell SessionManager to Save And Quit" )
+
+#oparser.add_option( "--quitas", action="store", dest="quitas", type="string",
+#		    help="SaveAs And Quit" )
+oparser.add_option( "--load", action="store", dest="load", type="string",
+		    help="Load Session with <name>" )
+
+(opt,args) = oparser.parse_args()
+
+sm = SessionManager()
+
+if opt.saveas:
+    sm.save_session( opt.saveas )
+
+if opt.load:
+    sm.load_session( opt.load )
+
+sm.exit()
+
+