|
|
@@ -0,0 +1,100 @@
|
|
|
+#!/usr/bin/env python3
|
|
|
+
|
|
|
+import os
|
|
|
+import urllib.parse
|
|
|
+
|
|
|
+import cleancss
|
|
|
+import tornado.gen
|
|
|
+import tornado.escape
|
|
|
+import tornado.httpclient
|
|
|
+import tornado.ioloop
|
|
|
+import tornado.web
|
|
|
+
|
|
|
+from config import web as config
|
|
|
+import db
|
|
|
+
|
|
|
+class BaseHandler(tornado.web.RequestHandler):
|
|
|
+ def render(self, *args, **kwargs):
|
|
|
+ kwargs['host'] = config.host
|
|
|
+ return super(BaseHandler, self).render(*args, **kwargs)
|
|
|
+
|
|
|
+ def render_string(self, *args, **kwargs):
|
|
|
+ s = super(BaseHandler, self).render_string(*args, **kwargs)
|
|
|
+ return s.replace(b'\n', b'') # this is like Django's {% spaceless %}
|
|
|
+
|
|
|
+ def get_current_user(self):
|
|
|
+ return self.get_secure_cookie('username')
|
|
|
+
|
|
|
+ @tornado.gen.coroutine
|
|
|
+ def create_session(self, username):
|
|
|
+ self.set_secure_cookie('username', username)
|
|
|
+
|
|
|
+ @property
|
|
|
+ def db(self):
|
|
|
+ return self.application.db
|
|
|
+
|
|
|
+class MainHandler(BaseHandler):
|
|
|
+ @tornado.gen.coroutine
|
|
|
+ def get(self):
|
|
|
+ self.render('home.html')
|
|
|
+
|
|
|
+class RegisterHandler(BaseHandler):
|
|
|
+ @tornado.gen.coroutine
|
|
|
+ def get(self):
|
|
|
+ self.render('register.html')
|
|
|
+
|
|
|
+ @tornado.gen.coroutine
|
|
|
+ def post(self):
|
|
|
+ username = self.get_argument("username", "")
|
|
|
+ password = self.get_argument("password", "")
|
|
|
+ yield db.create_user(username, password)
|
|
|
+ self.create_session(user['username'])
|
|
|
+ self.redirect("/")
|
|
|
+
|
|
|
+class LoginHandler(BaseHandler):
|
|
|
+ @tornado.gen.coroutine
|
|
|
+ def get(self):
|
|
|
+ self.render('login.html')
|
|
|
+
|
|
|
+ @tornado.gen.coroutine
|
|
|
+ def post(self):
|
|
|
+ username = self.get_argument("username", "")
|
|
|
+ password = self.get_argument("password", "")
|
|
|
+ user = yield db.check_user(username, password)
|
|
|
+ if not user:
|
|
|
+ self.render("login.html")
|
|
|
+ else:
|
|
|
+ self.create_session(user['username'])
|
|
|
+ self.redirect("/")
|
|
|
+
|
|
|
+class LogoutHandler(BaseHandler):
|
|
|
+ def get(self):
|
|
|
+ self.clear_all_cookies()
|
|
|
+ self.redirect('/')
|
|
|
+
|
|
|
+class CSSHandler(tornado.web.RequestHandler):
|
|
|
+ def get(self, css_path):
|
|
|
+ css_path = os.path.join(os.path.dirname(__file__), 'static', css_path) + '.ccss'
|
|
|
+ with open(css_path, 'r') as f:
|
|
|
+ self.set_header('Content-Type', 'text/css')
|
|
|
+ self.write(cleancss.convert(f))
|
|
|
+
|
|
|
+if __name__ == '__main__':
|
|
|
+ app = tornado.web.Application(
|
|
|
+ handlers=[
|
|
|
+ (r'/', MainHandler),
|
|
|
+ (r'/register', RegisterHandler),
|
|
|
+ (r'/login', LoginHandler),
|
|
|
+ (r'/logout', LogoutHandler),
|
|
|
+ (r'/(css/.+)\.css', CSSHandler),
|
|
|
+ ],
|
|
|
+ template_path=os.path.join(os.path.dirname(__file__), 'templates'),
|
|
|
+ static_path=os.path.join(os.path.dirname(__file__), 'static'),
|
|
|
+ login_url='/login',
|
|
|
+ cookie_secret=config.cookie_secret,
|
|
|
+ debug=config.debug,
|
|
|
+ )
|
|
|
+ app.listen(config.port)
|
|
|
+ app.db = db.MomokoDB()
|
|
|
+ print('listening on :%d' % config.port)
|
|
|
+ tornado.ioloop.IOLoop.instance().start()
|