sysvitals/web/server.py

#!/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 self.db.create_user(username, password)
		self.create_session(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 self.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()