mumble-django/pyweb/mumble/views.py

from django.shortcuts				import render_to_response, get_object_or_404, get_list_or_404
from django.template				import RequestContext
from django.http					import HttpResponseRedirect
from django.core.urlresolvers		import reverse
from django.contrib.auth.decorators	import login_required

from models	import Mumble, MumbleUser
from mmobjects	import mmServer, mmChannel

class Storage( object ):
	s = list();
	r = None;


def show( request, server ):
	"Displays the channel list for the given Server ID."
	srv, o = createChannelList( server );
	
	return render_to_response(
		'mumble/mumble.htm',
		{ 'DBaseObject': srv, 'ServerObject': o, 'ChannelTable': Storage.s },
		context_instance = RequestContext(request)
		);


def showContent( server, user = None ):
	"Renders and returns the channel list for the given Server ID."
	from django.template import Context, loader
	
	srv, o = createChannelList( server );
	
	mumbleAcc = None;
	if user is not None:
		mmUsers = MumbleUser.objects.filter( owner = user );
		if mmUsers:
			mumbleAcc = mmUsers[0];
	
	t_content = loader.get_template( 'mumble/content.htm' );
	c_content = Context( { 'DBaseObject': srv, 'ServerObject': o, 'ChannelTable': Storage.s, 'user': user, 'mumbleAccount': mumbleAcc } );
	r_content = t_content.render( c_content );
	
	return r_content;


def createChannelList( server ):
	"Renders the channel list."
	srv = get_object_or_404( Mumble, id=server );
	
	o = srv.getServerObject();
	
	Storage.s = list();
	Storage.r = o.channels[0];
	o.channels[0].visit( renderListItem, 0 );
	
	return srv, o;


def renderListItem( item, level ):
	"Stores a line in the channel list."
	if item == Storage.r:
		return;
	
	# Filter channels that don't have players in them and are not a subchannel of root
	if level > 1 and item.playerCount == 0:
		# I used to test if item is an instance of mmChannel here. For some reason, that doesn't work. Dunno why.
		return;
	
	if isinstance( item, mmChannel ):
		Storage.s.append( ( level, item, item.parentChannels() ) );
	else:
		Storage.s.append( ( level, item ) );



@login_required
def register( request, server ):
	# Register the current user with this mumble server, or give them a form to change their registration data.
	srv = Mumble.objects.get( id=server );
	
	if request.user.is_authenticated():
		reg = MumbleUser.objects.get( server=srv, owner=request.user );
	else:
		reg = None;
	
	return render_to_response(
		'mumble/reg.htm',
		{ 'Mumble': srv, 'Reg': reg },
		context_instance = RequestContext(request)
		);

@login_required
def savereg( request ):
	#if not request.user.is_authenticated():
	#	raise Exception, "You need to be logged in to register yourself with Mumble.";
	
	srv = Mumble.objects.get( id=request.POST['id'] );
	reg = MumbleUser.objects.get( server=srv, owner=request.user );
	
	if reg is None:
		reg = MumbleUser( name=request.POST['username'], password=request.POST['password'], server=srv, owner=request.user );
	else:
		reg.name     = request.POST['username'];
		reg.password = request.POST['password'];
	
	reg.save();
	return HttpResponseRedirect( "/mumble/%d" % srv.id );