main.py

#!/usr/bin/env python
import hashlib 
import logging
import os
import random
import simplejson as json 
import urllib
import urllib2

from datetime import datetime
from google.appengine.ext import webapp
from google.appengine.ext.webapp import util
from google.appengine.ext.webapp import template
from google.appengine.api import users

from consts  import *
from cookies import LilCookies
from emailer import Emailer
from helpers import *
from models  import User, Question, Troupe, get_question

####################################################################################
# Decorators
####################################################################################
def login_required( fn ):
    def check( self ):
        user = self.get_user( )
        if user:
            fn( self, user )
        else:
            # Go log in
            self.redirect( '/' ) #users.create_login_url( '/question' ) )
            return
    
    return check

def data_required( fn ):
    def check( self, user ):
        if user.first_name == '' or user.last_name == '' or user.email == '':
            self.redirect( '/account' )
            return
        else:
            fn( self, user )
    return check
    
def once_per_day( fn ):
    def check( self, user ):
        if user.played_today:
            self.redirect( '/result' )
            return
        else:
            fn( self, user )
    return check

####################################################################################
# Handlers
####################################################################################
class URIHandler( webapp.RequestHandler ):

    def __init__(self):
        # For simple caching purposes. Do not directly access this. Use self.get_user() instead.
        self.user = None

    def get_user(self):
        if self.user:
            return self.user

        #uuid = 'g66c8dmdNl8QIyzC05LisbfBH2Bp6mkzleTIi7SW6ltvQ2BySDS05gC3mcw6bW96BgRTO9oGRj38rF0AHSMzQPbAFSVu0WAzoGKZFrSfsaUDI1c3D'
        
        #cookieutil = LilCookies(self, COOKIE_SECRET)
        #cookieutil.set_secure_cookie(name = COOKIE_NAME, value = uuid, expires_days = 365)

        # Look at cookies
        cookieutil = LilCookies(self, COOKIE_SECRET)
        uuid       = cookieutil.get_secure_cookie(name = COOKIE_NAME) 
        
        logging.error("UUID: %s" % uuid)

        if uuid:
            self.user = User.get_or_create_by_uid( uuid )
            """
            # Make sure the user has purchased this offer.
            params = { 'offer' : OFFER_ID,
                       'user'  : uuid }
            result = json.loads( urllib2.urlopen( '%s%s' % (API_URL, WILLET_CHECK_URL), urllib.urlencode( params )).read() ) 
            if not result['answer'] == 'yes':
                return None
            """
        else:
            logging.error("CANT FIND PERSON")
        
        return self.user
    
    def render_page(self, template_file_name, content_template_values):
        """This re-renders the full page with the specified template."""

        main_path = os.path.join('templates/index.html')
        content_path = os.path.join('templates/' + template_file_name )

        user = self.get_user()

        content_values = {
            'user' : user
        }
        merged_content_values = dict(content_values)
        merged_content_values.update(content_template_values)

        content = template.render( content_path, merged_content_values )

        msgs = user.messages
        msgs.reverse()

        show_messages = False
        if len( user.messages ) != 0:
            if 'result' in template_file_name or 'account' in template_file_name or 'about' in template_file_name:
                show_messages = True
                user.messages = []
                user.put()

        template_values = {
            'CONTENT': content,
            'LOGOUT_URL' : '/logout',
            'user' : user,
            'show_messages' : show_messages,
            'msgs' : msgs
        }
        merged_values = dict(template_values)
        merged_values.update(content_template_values)

        return template.render(main_path, merged_values)

#########
## Shows
#########
class ShowAccountHandler( URIHandler ):
    
    @login_required
    def get( self, user ):
        
        if user.first_name == '' or user.last_name == '' or user.email == '':
            html_page    = 'first_account.html'
            troupe_mates = []
        else:
            html_page    = 'account.html'
            troupe_mates = user.get_troupe_mates ()
            while len(troupe_mates) <= 5:
                troupe_mates.append( '%d. -' % (len(troupe_mates) + 1) )

      #  logging.info("got %d" % len(troupe_mates))
        
        template_values = { 
            'CSS_FILE' : 'account',
            'JS_FILE' : 'account',
            'daily_email' : user.daily_email,
            'troupe_mates' : troupe_mates[0:5], # Top 5 people
        }

        self.response.out.write( self.render_page( html_page, template_values ) )
    
    
class ShowAboutHandler( URIHandler ):
    
    @login_required
    def get( self, user ):

        template_values = {
            'CSS_FILE' : 'about',
            'JS_FILE' : 'about',
        }

        self.response.out.write( self.render_page( 'about.html', template_values ) )

class ShowQuestionHandler( URIHandler ):
    
    @login_required
    @data_required
    @once_per_day
    def get( self, user ):
        q  = get_question()

        template_values = {
            'JS_FILE' : 'question',
            'CSS_FILE' : 'question',
            'TODAY' : datetime.date( datetime.today() ) .strftime( '%A %B %d, %Y' ),
            'PASS' : PASS,
            'q' : q
        }

        self.response.out.write( self.render_page( 'question.html', template_values ) )
    
    def post( self ):
        
        message      = self.request.get( 'message' )
        message_key  = self.request.get( 'message_key' ) 
        message_hash = self.request.get( 'message_hash' ) 
        
        #foo = '{"message" : {"user_id" : "g66c8dmdNl8QIyzC05LisbfBH2Bp6mkzleTIi7SW6ltvQ2BySDS05gC3mcw6bW96BgRTO9oGRj38rF0AHSMzQPbAFSVu0WAzoGKZFrSfsaUDI1c3D", "description" : "adf", "title" : "asdf", "timestamp" : "1303356071.71", "price" : 86, "message_type" : "NEW", "offer_id" : "8f62704e12794e8d"}, "message_hash" : "39e643c11e558fdb6bf71165a6ab92721413edc1", "message_key" : "600c7d7e2f1f4aa7a91b7f995823ccc4"}'

        logging.error('DATA %s' % self.request)

        #data         = json.loads( self.request.get('data') )
        #message      = data[ 'message' ]
        #message_key  = data[ 'message_key' ]
        #message_hash = data[ 'message_hash' ]

        logging.error("Message: %s Key %s Hash %s" % (message, message_key, message_hash))

        computed_message_hash = hashlib.sha1(message + CLIENT_SECRET_KEY + message_key).hexdigest() 
        logging.error("MY %s their %s" % (computed_message_hash, message_hash))
        
        #computed_message_hash = "39e643c11e558fdb6bf71165a6ab92721413edc1"

        if not message_hash == computed_message_hash: 
            self.redirect( '/' )
        else: 
            uuid = json.loads( message )['user_id']

            user = User.get_or_create_by_uid( uuid )
            self.user = user
            
            cookieutil = LilCookies(self, COOKIE_SECRET)
            cookieutil.set_secure_cookie(name = COOKIE_NAME, value = uuid, expires_days = 1)
            logging.error("SETTING A COOKIE")
            
            self.redirect( '/question' )

class ShowHowHandler( webapp.RequestHandler ):
    
    def get(self):
        path = os.path.join(os.path.dirname(__file__), 'templates/how.html')
        template_values = {
            'offer_id' : OFFER_ID
        }
        
        self.response.out.write( template.render( path, template_values ) )

class ShowLandingHandler( webapp.RequestHandler ):
    
    def get(self):
        path = os.path.join(os.path.dirname(__file__), 'templates/landing.html')
        template_values = {
            'offer_id' : OFFER_ID
        }

        self.response.out.write( template.render( path, template_values ) )

class ShowResultHandler( URIHandler ):

    @login_required
    @data_required
    def get( self, user ):
        if not user.played_today:
            self.redirect( '/question' )
            return

        score = self.request.get( 'score' )
        if not score:
            score = 0

        place = self.request.get( 'place' )
        if not place:
            place = 0

        already_played = False
        if not (score == '1' or score == '2' or score == '-1'):
            if user.last_played:
                already_played = True
       
        troupe_mates = user.get_troupe_mates ()
        while len(troupe_mates) <= 5:
            troupe_mates.append( '%d. -' % (len(troupe_mates) + 1) )

        template_values = {
            'CSS_FILE' : 'result',
            'q_score' : int(score),
            'score' : user.score,
            'question' : get_question().question,
            'answer' : get_question().answer,
            'already_played' : already_played,
            'place' : place,
            'troupe_mates' : troupe_mates[0:5] # Top 5 people
        }

        self.response.out.write( self.render_page( 'result.html', template_values ) )

#######
## Do's
#######

class DoInviteHandler( URIHandler ):

    @login_required
    def post( self, user ):
        email = self.request.get( 'email' )
        
        if is_good_email( email ) and not user.first_name == '':
            Emailer.invite( email, user )
        else:
            self.response.set_status( 400 )
        
class DoUpdateAccountHandler( URIHandler ):
    
    @login_required
    def post( self, user ):

        first_name  = self.request.get( 'first_name' )
        last_name   = self.request.get( 'last_name' )
        email       = self.request.get( 'email' )
        daily_email = self.request.get( 'daily_email' )

        if is_good_name( first_name ):
            user.first_name = first_name.capitalize()
        else:
            self.response.set_status( 400 )
            return
        
        if is_good_name( last_name ):
            user.last_name  = last_name.capitalize()
        else:
            self.response.set_status( 401 )
            return

        if is_good_email( email ):
            user.email      = email
        else:
            self.response.set_status( 402 )
            return

        if daily_email == 'first':
            user.daily_email = True
            troupe           = self.request.get( 'troupe' )

            if is_good_name( troupe ) and not troupe == "Know your troupe? (optional)":
                user.switch_troupe( troupe )
            else:
                user.put()
            
            self.response.set_status( 403 )
        else:
            user.daily_email = True if daily_email == 'on' else False
            user.put()
            self.redirect( '/account' )

class DoUpdateTroupeHandler( URIHandler ):
    
    @login_required
    def post( self, user ):
        
        troupe_name = self.request.get( 'troupe_name' )

        if is_good_name( troupe_name ):
            user.switch_troupe( troupe_name )
        else:
            self.response.set_status( 400 )
        
        self.redirect( '/account' )

class DoSubmitFeedbackHandler( URIHandler ):
    
    @login_required
    def post( self, user ):
        question    = self.request.get( 'question' )
        correct     = self.request.get( 'correct' )
        cat         = self.request.get( 'category' )
        diff        = self.request.get( 'difficulty' )
        incorrect_1 = self.request.get( 'incorrect_1' )
        incorrect_2 = self.request.get( 'incorrect_2' )
        incorrect_3 = self.request.get( 'incorrect_3' )

        if not question == 'Question':
            q = Question( question=question, opt_1=correct, opt_2=incorrect_1, opt_3=incorrect_2, opt_4=incorrect_3, answer=correct, state='not_approved', category=cat, difficulty=diff, submitter=user )
            q.put()

        Emailer.feedback ( question, correct, incorrect_1, incorrect_2, incorrect_3, user )

        self.redirect( '/about' )

class DoAnswerQuestionHandler( URIHandler ):

    @login_required
    @data_required
    @once_per_day
    def post( self, user ):
        # Grab the user's answer
        ans = self.request.get( 'q1' )

        # Grab today's question
        question = get_question( )
    
        # Check if the user is correct:
        score = user.update_score( question, ans )

        # Update time
        user.last_played  = triv_today( True )
        user.played_today = True
        user.put()

        # Go to Results screen
        self.redirect( '/result?score=%s' % score )

class DoLogoutHandler( URIHandler ):

    @login_required
    def post( self, user ):
        logging.error("LOGOUT")
        cookieutil = LilCookies( self, COOKIE_SECRET )
        cookieutil.clear_cookie( name = COOKIE_NAME )
        logging.error("DELETING COOKIE")

def main():
    application = webapp.WSGIApplication([
                                          ('/about', ShowAboutHandler),
                                          ('/account', ShowAccountHandler),
                                          ('/answerQuestion', DoAnswerQuestionHandler),
                                          ('/how', ShowHowHandler),
                                          ('/invite', DoInviteHandler),
                                          ('/logout', DoLogoutHandler),
                                          ('/question', ShowQuestionHandler),
                                          ('/result', ShowResultHandler),
                                          ('/submitFeedback', DoSubmitFeedbackHandler),
                                          ('/updateAccount', DoUpdateAccountHandler),
                                          ('/updateTroupe', DoUpdateTroupeHandler),
                                          ('/.*', ShowLandingHandler),
                                          ],
                                         debug=True)
    util.run_wsgi_app(application)


if __name__ == '__main__':
    main()