def post(self):
     '''
     Processes user login
     '''
     self.response.headers['Content-Type'] = 'text/html'
     username = self.request.get('username')
     password = self.request.get('password')
     values = {'username': username, 'password': password}
     cookie = self.request.cookies.get('user_id')
     service = UserService()
     new_cookie = ''
     isError = False
     try:
         new_cookie = service.login(username=username,password=password, cookie=cookie)
     except InvalidCookieError as error:
         isError = True
         self.response.headers.add_header('Set-Cookie', 'user_id=; Path=/')
         self.redirect('/final/signup')
     except LoginError as error:
         isError = True
         messages = error.validation_messages
         logging.info("LoginError msgs: " + str(messages))
         values.update(messages)
         path = os.path.join(os.path.dirname(__file__), '../templates/login.html')
         self.response.out.write(template.render(path, values))
     if not isError:
         self.response.headers.add_header('Set-Cookie', 'user_id=%s; Path=/' % new_cookie)
         self.redirect('/final/welcome')
         
 def post(self):
     '''
     Processes user login
     '''
     self.response.headers['Content-Type'] = 'text/html'
     username = self.request.get('username')
     password = self.request.get('password')
     values = {'username': username, 'password': password}
     cookie = self.request.cookies.get('user_id')
     service = UserService()
     new_cookie = ''
     isError = False
     try:
         new_cookie = service.login(username=username,
                                    password=password,
                                    cookie=cookie)
     except InvalidCookieError as error:
         isError = True
         self.response.headers.add_header('Set-Cookie', 'user_id=; Path=/')
         self.redirect('/final/signup')
     except LoginError as error:
         isError = True
         messages = error.validation_messages
         logging.info("LoginError msgs: " + str(messages))
         values.update(messages)
         path = os.path.join(os.path.dirname(__file__),
                             '../templates/login.html')
         self.response.out.write(template.render(path, values))
     if not isError:
         self.response.headers.add_header('Set-Cookie',
                                          'user_id=%s; Path=/' % new_cookie)
         self.redirect('/final/welcome')
def check_cookie(request_handler):
    try:
        cookie = request_handler.request.cookies.get("user_id")
        user_service = UserService()
        user_service.check_cookie(cookie)
    except InvalidCookieError as error:
        request_handler.response.headers.add_header('Set-Cookie', 'user_id=; Path=/')
        request_handler.redirect('/final/signup')
Beispiel #4
0
def check_cookie(request_handler):
    try:
        cookie = request_handler.request.cookies.get("user_id")
        user_service = UserService()
        user_service.check_cookie(cookie)
    except InvalidCookieError as error:
        request_handler.response.headers.add_header('Set-Cookie',
                                                    'user_id=; Path=/')
        request_handler.redirect('/final/signup')
 def get(self):
     '''
     Handles get requests
     '''
     self.response.headers['Content-Type'] = 'text/html'
     #delete user from data store
     user_id = self.request.cookies.get("user_id")
     if user_id:
         user_id = user_id[0:user_id.find('|')]
         service = UserService()
         logging.info("User %s logged out? %s" %
                      (user_id, str(service.logout(user_id))))
     #clear cookie by setting it to an empty string
     self.response.headers.add_header('Set-Cookie', 'user_id=; Path=/')
     self.redirect('/final/signup')
 def get(self):
     '''
     Handles get requests
     '''
     self.response.headers['Content-Type'] = 'text/html'
     #delete user from data store
     user_id = self.request.cookies.get("user_id")
     if user_id:
         user_id = user_id[0:user_id.find('|')]
         service = UserService()
         logging.info("User %s logged out? %s" % (user_id, str(service.logout(user_id))))
     #clear cookie by setting it to an empty string
     self.response.headers.add_header('Set-Cookie', 'user_id=; Path=/')
     self.redirect('/final/signup')
             
    def get(self):
        '''
        Handles get request
        '''
#        values = {'username':self.request.get('username')}
        values = {}
        cookie = self.request.cookies.get('user_id')        
        service = UserService()
        user = None
        try:
            user = service.welcome(cookie)
            values['username'] = user.username
        except InvalidCookieError as error:
            logging.error('Invalid cookie in UserSignupWelcome: ' + str(error))
            self.redirect('/final/signup')
            
        path = os.path.join(os.path.dirname(__file__), '../templates/welcome.html')
        self.response.out.write(template.render(path, values))
        
 def setUp(self):
     # Setup mock GAE Data Store
     self.testbed = testbed.Testbed()
     self.testbed.activate()
     self.testbed.init_datastore_v3_stub()
     self.testbed.init_memcache_stub()
     self.service = UserService()
     factory = UserDataStoreFactory()
     self.storage = factory.get_storage()
     self.crypt = Crypt()
 def post(self):
     '''
     Handles post requests
     '''
     self.response.headers['Content-Type'] = 'text/html'
     username = self.request.get('username')
     password = self.request.get('password')
     verify = self.request.get('verify')
     email = self.request.get('email')
     values = {'username':username, 'password':password,'verify':verify,'email':email}
     cookie = ''
     service = UserService()
     #Validate that user is not already registered
     isValid = False
     try:
         cookie = service.register(username=username, password=password, verify=verify, email=email)
         isValid = True
     except UserRegistrationError as error:
         path = os.path.join(os.path.dirname(__file__), '../templates/usersignup.html')
         values.update(error.validation_messages)
         self.response.out.write(template.render(path, values))
     if isValid:    
         self.response.headers.add_header('Set-Cookie', 'user_id=%s; Path=/' % cookie)
         self.redirect('/final/welcome')
class Test(unittest.TestCase):


    def setUp(self):
        # Setup mock GAE Data Store
        self.testbed = testbed.Testbed()
        self.testbed.activate()
        self.testbed.init_datastore_v3_stub()
        self.testbed.init_memcache_stub()
        self.service = UserService()
        factory = UserDataStoreFactory()
        self.storage = factory.get_storage()
        self.crypt = Crypt()


    def tearDown(self):
        self.testbed.deactivate()


    def testLogin(self):
        username = '******'
        password = '******'
        hashed_password = self.crypt.crypt_password(username, password)
        user = UserData(username=username, password=hashed_password)
        self.storage.save(user)
        user_id = user.key().id()
        cookie = self.crypt.make_cookie(user_id)
        new_cookie = self.service.login(username, password, cookie)
        self.assertFalse(new_cookie == None)
        self.assertEquals(user.key().id(), 1)

    def testLogin_Failed(self):
        username = '******'
        password = '******'
        hashed_password = self.crypt.crypt_password(username, password)
        user = UserData(username=username, password=hashed_password)
        self.storage.save(user)
        user_id = user.key().id()
        cookie = self.crypt.make_cookie(user_id)
        try: 
            self.service.login(username + 'foo', password, cookie)
            self.fail('Should not get here because LoginError should be raised due to bad usernames')
        except LoginError as error:
            self.assertEquals(error.validation_messages['login_error'], 'Invalid login')
            print error


    def testRegister(self):
        username = '******'
        password = '******'
        verify = password
        cookie = self.service.register(username=username, password=password, verify=verify)
        print cookie
        user = self.storage.fetchByUsername(username)
        self.assertFalse(cookie == None)
        user_id = user.key().id()
        self.assertEquals(user_id, 1)
        cookie_split = cookie.split('|')
        self.assertTrue(cookie_split[0] == str(user_id))
        self.assertFalse(cookie_split[1] == None)
        self.assertFalse(cookie_split[1] == '' )

    def testRegister_failed(self):
        username = '******'
        password = '******'
        verify = password
        user = UserData(username=username, password=password)
        self.storage.save(user)
        try:
            self.service.register(username=username, password=password, verify=verify)
            self.fail('Should not get here because UserRegistrationError should be raised due to the fact that user already exists')
        except UserRegistrationError as error:
            self.assertEquals(error.validation_messages['user_exists_error'], 'That user already exists.')