class LoginHandler(tornado.web.RequestHandler):
def initialize(self):
self.dataController = DataController(self.application.con)
def get(self):
userId = getUserIdFromCookie(self)
if userId:
self.redirect("/index", permanent=True)
else:
self.render("login.html", result=None)
"""
username = self.get_argument('username', '')
check = self.get_argument("check", 0)
if (check == 0):
self.render("login.html", result=None)
else:
userId = self.dataController.getUserByName(username)
if userId:
self.write('1')
else:
self.write('0')
"""
def post(self):
username = self.get_argument("username", "")
password = self.get_argument("password", "")
user = self.dataController.getUserByName(username)
if user:
if password == user["password"]:
userId = user["id"]
self.set_secure_cookie("userId", str(userId))
self.redirect("/index", permanent=True)
return
self.render("login.html", result="用户不存在或密码错误！")
class SignUpHandler(tornado.web.RequestHandler):
def initialize(self):
self.dataController = DataController(self.application.con)
def get(self):
username = self.get_argument("userName", "")
if username:
userId = self.dataController.getUserIdByName(username)
self.write(str(userId))
else:
self.render("signup.html", result=None)
def post(self):
username = self.get_argument("username", "")
password = self.get_argument("password", "")
user = self.dataController.getUserByName(username)
if not user:
user = dict()
user["userName"] = username
user["password"] = password
userId = self.dataController.createUser(user)
self.set_secure_cookie("userId", str(userId))
self.redirect("/index", permanent=True)
return
self.render("signup.html", result="用户名已被注册！")
