main.py

import os
import logging
import webapp2
import jinja2
from google.appengine.api import users, channel
import base64
import QueueManager, ChannelManager, LabTA
import logging

import LabTAUtils
from LabTA import is_ta

JINJA_ENVIRONMENT = jinja2.Environment(
    loader=jinja2.FileSystemLoader(os.path.dirname(__file__)),
    extensions=['jinja2.ext.autoescape'])
AVERAGE_WORK_TIME = 15  # Number if minutes a TA spends on the typical student

class MainPage(webapp2.RequestHandler):
    def get(self):
        user = users.get_current_user()
        json_queue = QueueManager.get_json_queue()
        token = channel.create_channel(user.email())
        if is_ta(user.email()):
            logging.info("{} is a TA".format(user.email()))
        template_values = {'logout_url': users.create_logout_url('/'),
                           'schedule_url': "http://labta.cs.princeton.edu", 
                           'is_ta': is_ta(user.email()),
                           'curr_user': user.email(),
                           'token': token,
                           'queue': base64.b64encode(json_queue),
                           'active_tas': LabTA.update_active_tas()}
        template = JINJA_ENVIRONMENT.get_template('templates/HelpQueue.html')
        self.response.write(template.render(template_values))

app = webapp2.WSGIApplication([
    ('/', MainPage),
    ('/queue', QueueManager.GetQueue),
    ('/add', QueueManager.AddToQueue),
    ('/mark-helped', QueueManager.MarkAsHelped),
    ('/cancel', QueueManager.CancelFromQueue),
    ('/clear-queue', QueueManager.ClearQueue),
    ('/ack-modal', LabTA.AcknowledgeModal),
    ('/ta-facebook', LabTA.TAFacebook),
    ('/_ah/channel/connected/', ChannelManager.SubscriberConnect),
    ('/_ah/channel/disconnected/', ChannelManager.SubscriberDisconnect),
    ('/_update-tas', LabTAUtils.SetTAsInNDB)
   # ('/_update-schema', LabTAUtils.InQueueSchemaUpdate)
], debug=True)