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('/unit6/signup')
except LoginError as error:
isError = True
messages = error.validation_messages
logging.info("LoginErrlr 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('/unit6/welcome')
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 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('/unit6/signup')
path = os.path.join(os.path.dirname(__file__), '../templates/welcome.html')
self.response.out.write(template.render(path, values))
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('/unit6/welcome')
