X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-web.git/blobdiff_plain/9fcc8cfc4797aafbc674ea78e42e5e51bdb0cf5e..aea62528d39fcdeaf94b440ef3e14992cb79f704:/code/main.py diff --git a/code/main.py b/code/main.py index dd60e96..5c30714 100755 --- a/code/main.py +++ b/code/main.py @@ -12,8 +12,9 @@ import simplejson import sys import time import urllib +import socket +import cherrypy from StringIO import StringIO - def revertStandardError(): """Move stderr to stdout, and return the contents of the old stderr.""" errio = sys.stderr @@ -44,6 +45,17 @@ from invirt.database import Machine, CDROM, session, connect, MachineAccess, Typ from invirt.config import structs as config from invirt.common import InvalidInput, CodeError +from view import View + +class InvirtWeb(View): + def __init__(self): + super(self.__class__,self).__init__() + connect() + + @cherrypy.expose + def helloworld(self): + return "Hello world!" + def pathSplit(path): if path.startswith('/'): path = path[1:] @@ -150,7 +162,7 @@ def parseCreate(username, state, fields): kws = dict([(kw, fields.getfirst(kw)) for kw in 'name description owner memory disksize vmtype cdrom autoinstall'.split()]) validate = validation.Validate(username, state, strict=True, **kws) return dict(contact=username, name=validate.name, description=validate.description, memory=validate.memory, - disksize=validate.disksize, owner=validate.owner, machine_type=validate.vmtype, + disksize=validate.disksize, owner=validate.owner, machine_type=getattr(validate, 'vmtype', Defaults.type), cdrom=getattr(validate, 'cdrom', None), autoinstall=getattr(validate, 'autoinstall', None)) @@ -606,16 +618,17 @@ def info(username, state, path, fields): def unauthFront(_, _2, _3, fields): """Information for unauth'd users.""" - return templates.unauth(searchList=[{'simple' : True}]) + return templates.unauth(searchList=[{'simple' : True, + 'hostname' : socket.getfqdn()}]) def admin(username, state, path, fields): if path == '': return ({'Status': '303 See Other', 'Location': 'admin/'}, "You shouldn't see this message.") - if not username in getAfsGroupMembers(config.web.adminacl, 'athena.mit.edu'): + if not username in getAfsGroupMembers(config.adminacl, 'athena.mit.edu'): raise InvalidInput('username', username, - 'Not in admin group %s.' % config.web.adminacl) + 'Not in admin group %s.' % config.adminacl) newstate = State(username, isadmin=True) newstate.environ = state.environ return handler(username, newstate, path, fields)