def keeps_previous_items_when_setting_new_ones(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.set('some-object', {'foo': 'bar'})
manager.set('some-object2', {'foo2': 'bar2'})
self.assertEqual(manager.get('some-object')['foo'], 'bar')
self.assertEqual(manager.get('some-object2')['foo2'], 'bar2')
def keeps_previous_items_when_setting_new_ones(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.set('some-object', {'foo': 'bar'})
manager.set('some-object2', {'foo2': 'bar2'})
self.assertEqual(manager.get('some-object')['foo'], 'bar')
self.assertEqual(manager.get('some-object2')['foo2'], 'bar2')
def get(cls, handler, name):
session = SessionManager(handler)
Debug.dprint(text='name: ' + name, type='custom')
Debug.dprint(text='value: ' + str(session.get(name)), type='error')
# print "Session name is: " + name
# print "Do we have a session? " + str(session.get(name))
# if cls.exists(handler, name):
# print "There is a session"
# else:
# print "There is not any sessionn"
return session.get(name)
def get(self, *args, **kwargs):
# redis_instance.set(name='name', value='EHSAN')
# self.write(redis_instance.get(name='name'))
session = SessionManager(self)
# session.set('key', 'OMG')
session.set(name='key', value='OMG')
# session['key'] = 'OMG'
self.write(session.get(self, 'key'))
self.write(str(session.get('key')))
# self.write(str(session.keys()))
self.write(str(session.iterkeys()))
# self.write(session['key'])
self.write('done')
def get(self):
session = SessionManager(self)
k = session.get('catId')
if k != None:
self.render("login.html")
else:
self.redirect('/indexihtml')
def gets_default_value_if_provided_and_not_in_client(self):
handler = StubHandler()
manager = SessionManager(handler)
value = manager.get('foo', 'Default')
self.assertEqual(value, 'Default')
def retrieves_session_with_same_data_as_saved(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.set('some-object', {'foo': 'bar'})
self.assertEqual(manager.get('some-object')['foo'], 'bar')
def gets_default_value_if_provided_and_not_in_client(self):
handler = StubHandler()
manager = SessionManager(handler)
value = manager.get('foo', 'Default')
self.assertEqual(value, 'Default')
def get(self):
sessionid = self.get_current_session()
logging.debug("LoginHandler: sessionid %s" % sessionid)
session = SessionManager(self)
tpl_fields = TemplateFields()
tpl_fields['post'] = False
tpl_fields['server'] = session.get('server')
tpl_fields['username'] = session.get('user')
# using the next field makes it easier across logins
tpl_fields['next'] = self.get_argument('next', '/')
logging.debug("LoginHandler: next: %s" % tpl_fields['next'])
logging.debug('LoginHandler: login page about to be rendered')
self.render("login.html", fields=tpl_fields)
def retrieves_session_with_same_data_as_saved(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.set('some-object', {'foo': 'bar'})
self.assertEqual(manager.get('some-object')['foo'], 'bar')
def still_retrieves_object_if_not_passed_from_expiration(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.set('foo', 'bar')
time.sleep(1)
self.assertEqual(manager.get('foo'), 'bar')
def deletes_item_using_command(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.set('some-object', {'foo': 'bar'})
del manager['some-object']
self.assertIsNone(manager.get('some-object'))
def deletes_item_using_command(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.set('some-object', {'foo': 'bar'})
del manager['some-object']
self.assertIsNone(manager.get('some-object'))
def still_retrieves_object_if_not_passed_from_expiration(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.set('foo', 'bar')
time.sleep(1)
self.assertEqual(manager.get('foo'), 'bar')
class RequestHandler(base.RequestHandler):
def __init__(self, *args, **kwargs):
super(RequestHandler, self).__init__(*args, **kwargs)
self.session = SessionManager(self)
def get_template_path(self):
return settings.base.template_path
@gen.coroutine
def prepare(self):
try:
user_id = self.session.get(settings.redis.USER_KEY)
if user_id:
user = yield self.console_user_model.GetUserFromId(user_id)
if user:
yield self.Login(user)
except:
pass
def _handle_request_exception(self, error):
if self._finished:
return
if isinstance(error, errors.Error):
self.render_error(error)
elif isinstance(error, MissingArgumentError):
self.render_error(errors.ParamInvalidError(error.arg_name))
else:
super(RequestHandler, self)._handle_request_exception(error)
def render_error(self, error):
data = {
'code': u'',
'message': u'操作发生错误'
}
if isinstance(error, errors.PermissionDeny):
data['code'] = error.error_code
data['message'] = error.error_message
self.render('global_error.html', data=data)
@gen.coroutine
def Login(self, user):
if user['status'] in ['locked']:
raise gen.Return(False)
user['privileges'] = []
roles = user['roles']
for r in roles:
role = yield self.role_model.GetRoleFromName(r)
if role:
privileges = role['privileges']
user['privileges'].extend(privileges)
self.session.set(settings.redis.USER_KEY, user['_id'])
self.current_user = user
raise gen.Return(True)
def cannot_retrieve_object_if_passed_from_expiration(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.driver.EXPIRE_SECONDS = 1
manager.set('foo', 'bar')
time.sleep(manager.driver.EXPIRE_SECONDS + 1)
self.assertIsNone(manager.get('foo'))
def cannot_retrieve_object_if_passed_from_expiration(self):
handler = StubHandler()
manager = SessionManager(handler)
manager.driver.EXPIRE_SECONDS = 1
manager.set('foo', 'bar')
time.sleep(manager.driver.EXPIRE_SECONDS + 1)
self.assertIsNone(manager.get('foo'))
def get(self):
sessionid = self.get_current_session()
session = SessionManager(self)
vh = VimHelper()
logging.debug("about to pull clusters using sessionid %s" % sessionid)
clusters = vh.GetClusters(sessionid)
f = TemplateFields()
f['username'] = self.get_current_user()
f['servername'] = session.get('server')
f['servertype'] = vh.ServerType(sessionid)
f['serverapi'] = vh.ApiVersion(sessionid)
f['clusterlist'] = clusters
self.render("listclusters.html", fields=f)
def get(self, vmpath):
logging.debug("ShowVMHandler: %s" % vmpath)
vh = VimHelper()
sessionid = self.get_current_session()
session = SessionManager(self)
f = TemplateFields()
f['username'] = self.get_current_user()
f['servername'] = session.get('server')
f['servertype'] = vh.ServerType(sessionid)
f['serverapi'] = vh.ApiVersion(sessionid)
vm = vh.GetVM(sessionid, vmpath)
f['vmstatus'] = vm.get_status()
f['vmproperties'] = vm.get_properties()
self.render("showvm.html", fields=f)
def get(self):
vh = VimHelper()
sessionid = self.get_current_session()
session = SessionManager(self)
f = TemplateFields()
f['username'] = self.get_current_user()
f['servername'] = session.get('server')
task = Tasks()
task.add('task1 test')
task.add('task2 test')
task.add('task3 test')
ids = task.getids()
mytasks = {}
for id in ids:
mytasks[id] = task.get(id)
f['tasks'] = mytasks
self.render('showtasks.html', fields=f)
def retrieves_none_if_session_object_not_previously_set(self):
handler = StubHandler()
manager = SessionManager(handler)
self.assertIsNone(manager.get('unexistant-object'))
def retrieves_none_if_session_object_not_previously_set(self):
handler = StubHandler()
manager = SessionManager(handler)
self.assertIsNone(manager.get('unexistant-object'))
def exists(cls, handler, name):
session = SessionManager(handler)
# print "Inside exist: " + name + ' ' + str(session.get(name))
return session.get(name) != None
class TornadoRequestBase(tornado.web.RequestHandler):
def __init__(self, application, request, **kwargs):
super(TornadoRequestBase, self).__init__(application, request, **kwargs)
self.session = SessionManager(self)
self.user = {
"name": self.session.get('name', None),
"user": self.session.get('user', None),
"id": self.session.get('id', None),
"picture_address": self.session.get('picture_address', None),
"id_admin": self.session.get('id_admin', None),
"type_user": self.session.get('type_user', None),
"online": self.session.get('online', None),
"password": self.session.get('password', None),
"email": self.session.get('email', None),
"message": self.session.get('message', None),
"count_message": self.session.get('count_message', None)
}
def get_current_user(self):
return self.session.get('online')
class TornadoRequestBase(tornado.web.RequestHandler):
def __init__(self, application, request, **kwargs):
super(TornadoRequestBase, self).__init__(application, request,
**kwargs)
self.session = SessionManager(self)
self.user = {
"name": self.session.get('name', None),
"user": self.session.get('user', None),
"id": self.session.get('id', None),
"picture_address": self.session.get('picture_address', None),
"id_admin": self.session.get('id_admin', None),
"type_user": self.session.get('type_user', None),
"online": self.session.get('online', None),
"password": self.session.get('password', None),
"email": self.session.get('email', None),
"message": self.session.get('message', None),
"count_message": self.session.get('count_message', None)
}
def get_current_user(self):
return self.session.get('online')
def post(self):
session = SessionManager(self)
# if session.get()
noun1 = self.get_argument('noun1')
noun2 = self.get_argument('noun2')
verb = self.get_argument('verb')
noun3 = self.get_argument('noun3')
us = self.get_current_user()
self.render('poem.html', roads=noun1, wood=noun2, made=verb,difference=noun3,user = us,ID=session.get('_id'),INFO=session.get('info'))
