31 lines
1.7 KiB
Python
31 lines
1.7 KiB
Python
import tornado.web
|
||
from pycket.session import SessionMixin
|
||
from .MongoDB import *
|
||
|
||
|
||
class AuthBaseHandler(tornado.web.RequestHandler,SessionMixin):
|
||
def get_current_user(self): #重写get_current_user()方法
|
||
return self.session.get('username',None) #session是一种会话状态，跟数据库的session可能不一样
|
||
|
||
#添加装饰器,装饰需要验证的请求
|
||
class IndexHandler(AuthBaseHandler):
|
||
|
||
@tornado.web.authenticated #@tornado.web.authenticated装饰器包裹get方法时，表示这个方法只有在用户合法时才会调用，authenticated装饰器会调用get_current_user()方法获取current_user的值，若值为False，则重定向到登录url装饰器判断有没有登录，如果没有则跳转到配置的路由下去，但是要在app.py里面设置login_url
|
||
def get(self,*args,**kwargs):
|
||
username = self.get_current_user()
|
||
self.render('index.html',user=username)
|
||
|
||
class TestHandler(AuthBaseHandler):
|
||
|
||
@tornado.web.authenticated #@tornado.web.authenticated装饰器包裹get方法时，表示这个方法只有在用户合法时才会调用，authenticated装饰器会调用get_current_user()方法获取current_user的值，若值为False，则重定向到登录url装饰器判断有没有登录，如果没有则跳转到配置的路由下去，但是要在app.py里面设置login_url
|
||
def post(self,*args,**kwargs):
|
||
username = self.get_current_user()
|
||
message = self.get_argument('message')
|
||
|
||
print('TestHandler: '+username+' '+message)
|
||
Mongo.connect(DataBase='example',Collection=username)
|
||
Mongo.update(behavior=message,tags='test')
|
||
|
||
self.write("success")
|
||
|