Python web架構 Tornado（三）自訂session組件

標籤：/usr   value   ==   web架構   style   cache   body   串連   rgs   

　　我們在學習Django架構的過程中，內部封裝了session組件，以方便於我們使用進行驗證。但是Tornado架構是沒有session的，所以如果想使用session的話，就需要我們自己定製相對應的組件。在Tornado架構中是存在cookie機制的，所以我們可以利用加密的cookie以實現驗證機制。通過設定檔決定資料資訊存放的位置。是在緩衝還是在記憶體中。具體代碼如下：

#!/usr/bin/env python# _*_ coding:utf-8 _*_import timeimport jsonimport redisimport hashlibimport importlibimport settingsdef gen_random_str():    """        擷取隨機字串，作為session的key    """    md5 = hashlib.md5()    md5.update(str(time.time()).encode("utf-8"))    return md5.hexdigest()    class CacheSession(object):    """    記憶體中擷取    """    container = {}  # 保證類執行個體化一次，僅有一個session，同一個對象也是一個session    def __init__(self,handler):        self.handler = handler #接收handler對象        self.session_id = settings.SESSION_ID #從設定檔中擷取session的id        self.expires = settings.EXPIRERS #從設定檔中擷取session的逾時時間        self.initial() #初始化    def initial(self):        #先從cookies擷取隨機字串，能拿到並且在container中證明之前訪問過，session 不變，若是不滿足就重建        client_random_str = self.handler.get_cookie(self.session_id) #從cookies中擷取session_id        if client_random_str and client_random_str in self.container: #判斷            self.random_str = client_random_str #存在就賦值        else:#不存在就建立，或是錯誤            self.random_str = gen_random_str() #產生一個新的隨機字串            self.container[self.random_str] = {}       #每次請求來，在逾時時間之內，都需要回寫一次，以延長逾時時間        expires = time.time()+self.expires        self.handler.set_cookie(self.session_id,self.random_str,expires=expires)    def __getitem__(self, item):        return self.container[self.random_str].get(item)    def __setitem__(self, key, value):        self.container[self.random_str][key]=value    def __delitem__(self, key):        if key in self.container[self.random_str]:            del self.container[self.random_str][key]            class RedisSession(object):    """    緩衝中擷取    """    def __init__(self,handler):        self.handler = handler        self.session_id = settings.SESSION_ID        self.expires = settings.EXPIRERS        self.initial()    @property    def conn(self):        """        串連redis        :return:        """        conn = redis.Redis(host=settings.RedisHost,port=settings.RedisPort)        return conn    def initial(self):        """        初始化        :return:        """        client_random_str = self.handler.get_cookie(self.session_id)        if client_random_str and self.conn.exists(client_random_str): #檢測隨機字串及redis中的隨機字串是否存在            self.random_str = client_random_str        else:            self.random_str = gen_random_str()        expires = time.time()+self.expires        self.handler.set_cookie(self.session_id,self.random_str,expires=expires)        self.conn.expire(self.random_str,self.expires) #設定快取資料額度逾時時間    def __getitem__(self, item):        #注意在redis中擷取的資料全是位元組類型，需要轉成字串再json還原序列化        data_str = self.conn.hget(self.random_str,item).decode("utf-8")        data_dict = json.loads(data_str) if data_str else None        return data_dict    def __setitem__(self, key, value):        self.conn.hset(self.random_str,key,json.dumps(value))    def __delitem__(self, key):        self.conn.hdel(self.random_str,key)        class SessionFactory(object):    """    原廠模式，通過匯入的設定檔，執行個體化對應方法的    """    @staticmethod    def get_session():        engine = settings.SESSION_ENGINE        module_path,cls_name = engine.rsplit(‘.‘,maxsplit=1)        md = importlib.import_module(module_path) #以字串的形式匯入        cls = getattr(md,cls_name) #反射驗證，擷取有沒有這個方法        return cls #有的話返回這個類

session_code.py

#!/usr/bin/env python# _*_ coding:utf-8 _*_#設定檔# session 儲存類型# 儲存在記憶體中SESSION_ENGINE = "session_code.CacheSession"# 儲存在redis緩衝中# SESSION_ENGINE = "session_code.RedisSession"# RedisHost = ""# RedisPort = 6379SESSION_ID = "__session__id__"EXPIRERS = 300

settings.py

#!/usr/bin/env python# _*_ coding:utf-8 _*_import tornado.ioloopimport tornado.webfrom tornado.web import RequestHandlerfrom session_code import SessionFactory #匯入自訂session檔案的Factory 方法#建立可以調用session方法的類，讓其他需要使用的類去繼承class SessionHandler(object):    def initialize(self,*args,**kwargs):        """        重寫執行個體化中的鉤子函數，用於自訂實現某些方法        :param args:        :param kwargs:        :return:        """        cls = SessionFactory.get_session() #擷取使用的類方法        self.session = cls(self) #構造方法，把當前的請求類(handler)的對象(self)作為參數傳遞給session__code進行構造session。class LoginHandler(SessionHandler,RequestHandler):    def get(self, *args, **kwargs):        self.render("login.html")    def post(self, *args, **kwargs):                user = self.get_argument("user")        pwd = self.get_argument("pwd")                # 簡單實現使用者登入認證，可以從資料庫中擷取資料        if user=="aaa" and pwd == "123":            # 驗證，在跳轉之前，把資訊寫入session中            self.session["user"]=user # 寫入session            self.redirect("/index") # 跳轉到index頁面        else:            self.render("login.html") # 否則的話，就跳轉到登入頁面class IndexHandler(SessionHandler,RequestHandler):    def get(self, *args, **kwargs):        user = self.session["user"] # 從session中擷取資訊        if user:            self.write("歡迎登入")        else:            self.redirect("/login")set = {    "template_path":"views",}application = tornado.web.Application([    (r"/login",LoginHandler),    (r"/index",IndexHandler),],**set)if __name__ == ‘__main__‘:    application.listen(8888)    tornado.ioloop.IOLoop.instance().start()

app.py（應用）

　　遵循MVC模式使用Tornado架構，建立用於儲存頁面的views檔案夾。在app.py檔案應用中，設定對於的配置資訊。views檔案夾下簡單頁面代碼如下：

<!DOCTYPE html><html lang="en"><head>    <meta charset="UTF-8">    <title>Title</title></head><body>    <h1 class="c1">登入</h1><form method="POST">    <input type="text" name="user">    <input type="text" name="pwd">    <input type="submit" value="提交"></form></body></html>

login.html

<!DOCTYPE html><html lang="en"><head>    <meta charset="UTF-8">    <title>Title</title></head><body></body></html>

index.html

 

Python web架構 Tornado（三）自訂session組件