Tornado 之 Session的处理

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import tornado.web
container = {}

class Session:
    def __init__(self, handler):
        self.handler = handler
        self.random_str = None

    # 创建随机字符串
    def __genarate_random_str(self):
        import hashlib
        import time
        obj = hashlib.md5()
        obj.update(bytes(str(time.time()), encoding='utf-8'))
        random_str = obj.hexdigest()
        return random_str

    def __getitem__(self, item):
        return self.get_value(item)

    def get_value(self, key):
        # 获取客户端的随机字符串
        # 从container中获取专属于我的数据
        # 专属信息【key】
        random_str = self.handler.get_cookie("__kakaka__")
        if not random_str:
            return None
        # 客户端有随机字符串
        user_info_dict = container.get(random_str, None)
        if not user_info_dict:
            return None
        value = user_info_dict.get(key,None)
        return value

    def __setitem__(self, key, value):
        self.set_value(key,value)

    def set_value(self, key, value):
        # 在container中加入随机字符串
        # 定义专属于自己的数据
        # 在客户端中希尔随机字符串
        # 判断，请求的用户名是否已有随机字符串
        if not self.random_str:
            random_str = self.handler.get_cookie('__kakaka__')
            if not random_str:
                random_str = self.__genarate_random_str()
                container[random_str] = {}
            else:
                if random_str in container.keys():
                    pass
                else:
                    random_str = self.__genarate_random_str()
                    container[random_str] = {}
            self.random_str = random_str
        container[self.random_str][key] = value
        self.handler.set_cookie("__kakaka__",self.random_str)

class BaseHandler(tornado.web.RequestHandler):
    def initialize(self):
        self.session = session.Session(self)

class LoginHandler(BaseHandler):
    def get(self):
        if self.get_argument('u',None) in ['alex','eric']:
            self.session['is_login']=True
            self.session['name']= self.get_argument('u',None)
            self.write('已登录')
        else:
            self.write('请登录')

class ManagerHandler(BaseHandler):
    def get(self):
        val = self.session['is_login']
        if val:
            name = self.session['name']
            self.write('成功登录，'+name)
        else:
            self.write('失败')