webui: Implement login support (password authentication)

1 files changed, 30 insertions, 0 deletions

diff --git a/fg21sim/webui/handlers/base.py b/fg21sim/webui/handlers/base.py
new file mode 100644
index 0000000..5a6e3a9
--- /dev/null
+++ b/fg21sim/webui/handlers/base.py
@@ -0,0 +1,30 @@
+# Copyright (c) 2016 Weitian LI <liweitianux@live.com>
+# MIT license
+
+"""
+Base handler for other handlers
+"""
+
+
+import tornado.web
+from tornado.options import options
+
+
+class BaseRequestHandler(tornado.web.RequestHandler):
+    def get_current_user(self):
+        """
+        Override the ``get_current_user()`` method to implement user
+        authentication.
+
+        Determine the current user based on the value of a cookie.
+
+        References
+        ----------
+        - Tornado: Authentication and security
+          http://www.tornadoweb.org/en/stable/guide/security.html
+        """
+        if (options.password is None) or (options.password == ""):
+            # Password not set, then all accesses are allowed
+            return True
+        else:
+            return self.get_secure_cookie("user")