main_site.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
##
# index.py:
##
# © 2012 Steven Casagrande (scasagrande@galvant.ca) and
#     Christopher E. Granade (cgranade@cgranade.com).
# This file is a part of the ThingPool Server project.
# Licensed under the AGPL version 3.
##
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
##

## IMPORTS #####################################################################

## GAE API ##
from google.appengine.api import users

## WEB FRAMEWORK
import webapp2
import jinja2

## PYTHON STANDARD LIBRARY ##
import os

## THINGPOOL MODULES ##
import api
import config
import index
import dataModels
import security

## CORE SERVER #################################################################

jinja_environment = jinja2.Environment(
    loader=jinja2.FileSystemLoader(os.path.dirname(__file__)))
       
    
class MainPage(webapp2.RequestHandler):

    def get(self):        	    
        self.response.headers['Content-Type'] = 'text/html'	
        
        template_values = {
            'pool_name': config.POOL_NAME,
            'show_requested': False
        }
    
        user = users.get_current_user()
        if user:
            query = dataModels.Person.all().filter('user_account =', user)
            person = query.get()
            
            # Check if we need to file a request.
            if person is None:
                security.request_tp_account(user)
                logged_in = False
                template_values['show_requested'] = True
            else:
                if person.permissions > security.USER_STATUS_REQUESTED:
                    logged_in = True
                else:
                    # FIXME: currently there is no logout button shown,
                    #         so that a user can't change to a valid GAE account.
                    template_values['login_failed'] = True
                    template_values['login_failure_reason'] = {
                        security.USER_STATUS_BANNED: 'This user account has been banned from {}.'.format(config.POOL_NAME),
                        security.USER_STATUS_UNKNOWN: 'This error should not have happened.',
                        security.USER_STATUS_REQUESTED: 'Your account has not yet been approved by a {} administrator.'.format(config.POOL_NAME)
                    }[person.permissions]
                    logged_in = False
        else:
            logged_in = False
        
        # Avoid having to put an if operator in every single value by updating
        # the dictionary differently if we're logged in or not.
        if logged_in:
            template_values.update({
                'login_button': 'Logout',
                'loginout_url': users.create_logout_url(self.request.uri),
                'checked_out_things': 12, # FIXME: this is for mockup purposes ONLY.
                'username': user.nickname()
            })
        else:
            template_values.update({
                'login_button': 'Login',
                'loginout_url': users.create_login_url(self.url_for('main')),
                'checked_out_things': 0,
                'username' : 'Not logged in'
            })
            

        template = jinja_environment.get_template('templates/main/index.html')
        self.response.out.write(template.render(template_values))