tornado 里面沒有 session？不，當然有~我知道 github 上肯定有人幫我寫好了~ O(∩_∩)O~
于是乎，找到下面這個項目，用 memcached 實現 tornado 的 session。光會用可不行啊，讓我們看看是怎么寫的~
項目地址：tornado-memcached-sessions
讓我們先從 demo 看起....
app.py 中：
首先可以注意到，這里定義了一個新的 Application 類，繼承于 tornado.web.Application, 在該類的初始化方法中，設定了應用參數 settings, 之后初始化父類和 session_manager.（這是什么？暫時不管它...）
class Application(tornado.web.Application): def __init__(self): settings = dict( # 設定 cookie_secret, 用于 secure_cookie cookie_secret = "e446976943b4e8442f099fed1f3fea28462d5832f483a0ed9a3d5d3859f==78d", # 設定 session_secret 用于生成 session_id session_secret = "3cdcb1f00803b6e78ab50b466a40b9977db396840c28307f428b25e2277f1bcc", # memcached 地址 memcached_address = ["127.0.0.1:11211"], # session 過期時間 session_timeout = 60, template_path = os.path.join(os.path.dirname(__file__), "templates"), static_path = os.path.join(os.path.dirname(__file__), "static"), xsrf_cookies = True, login_url = "/login", ) handlers = [ (r"/", MainHandler), (r"/login", LoginHandler) ] # 初始化父類 tornado.web.Application tornado.web.Application.__init__(self, handlers, **settings) # 初始化該類的 session_manager self.session_manager = session.SessionManager(settings["session_secret"], settings["memcached_address"], settings["session_timeout"])
在下面的 LoginHandler 中我們可以看到 session 的使用：
class LoginHandler(BaseHandler): def get(self): self.render("login.html") def post(self): # 以字典的鍵值對形式存取 self.session["user_name"] = self.get_argument("name") # 修改完要調用 session 的 save, 否則等于沒有修改哦... self.session.save() self.redirect("/") 從使用來看是不是非常簡潔和清晰？那么，細心的你是不是發現現在的 handler 沒有繼承于 tornado.web.RequestHandler？帶著強烈的探（zuo）索（si）精神我們打開了 base.py。天啊，好短....（噢，你想到哪里去了...）
BaseHandler 的方法只是初始化，并重寫了 get_current_user 的用于用戶登錄驗證的方法。
class BaseHandler(tornado.web.RequestHandler): def __init__(self, *argc, **argkw): super(BaseHandler, self).__init__(*argc, **argkw) # 定義 handler 的 session, 注意，根據 HTTP 特點，每次訪問都會初始化一個 Session 實例哦，這對于你后面的理解很重要 self.session = session.Session(self.application.session_manager, self) # 這是干嘛的？用于驗證登錄...請 google 關于 tornado.web.authenticated, 其實就是 tornado 提供的用戶驗證 def get_current_user(self): return self.session.get("user_name") 看到這里，是不是心滿意足？噢，我終于理解了！。。。喂，說好的探（zuo）索（si）精神呢？關鍵在于 session.py ??！你一臉茫然地回過了頭....
首先看看需要的庫：
pickle 一個用于序列化反序列化的庫（聽不懂？你直接看成和 json 一樣作用就行了...）
hmac 和 hashlib 用于生成加密字符串
uuid 用于生成一個唯一 id
memcache Python 的 memcache 客戶端
這里面有三個類，SessionData Session 和 SessionManager。先看最簡單的 SessionData。
SessionData 用于以字典的結構存儲 session 數據，繼承于字典，其實只比字典多了兩個成員變量：
# 繼承字典，因為 session 的存取類似于字典 class SessionData(dict): # 初始化時提供 session id 和 hmac_key def __init__(self, session_id, hmac_key): self.session_id = session_id self.hmac_key = hmac_key
然后就是真正的 Session 類了。Session 類繼承于 SessionData, 注意，它還是十分像內置類型字典，只是重寫了自己的初始化方法，并定義了 save 接口――用于保存修改后的 session 數據。
# 繼承 SessionData 類 class Session(SessionData): # 初始化，綁定 session_manager 和 tornado 的對應 handler def __init__(self, session_manager, request_handler): self.session_manager = session_manager self.request_handler = request_handler try: # 正常是獲取該 session 的所有數據，以 SessionData 的形式保存 current_session = session_manager.get(request_handler) except InvalidSessionException: # 如果是第一次訪問會拋出異常，異常的時候是獲取了一個空的 SessionData 對象,里面沒有數據，但包含新生成的 # session_id 和 hmac_key current_session = session_manager.get() # 取出 current_session 中的數據，以鍵值對的形式迭代存下 for key, data in current_session.iteritems(): self[key] = data # 保存下 session_id self.session_id = current_session.session_id # 以及對應的 hmac_key self.hmac_key = current_session.hmac_key # 定義 save 方法，用于 session 修改后的保存，實際調用 session_manager 的 set 方法 def save(self): self.session_manager.set(self.request_handler, self)
__init__ 方法比較難理解，基本流程是定義自己的 session_manager 和 handler 處理對象。然后通過 session_manager 獲得已有的 session 數據，用這些數據初始化一個訪問的用戶的 session, 如果用戶是第一次訪問，那么他拿到的是一個新的 SessionData 對象，因為有可能是新用戶，所以這里要對 session_id 和 hmac_key（什么鬼） 進行賦值。
而 save 方法是提供了對修改 session 數據后的保存接口，實際是調用 session_manager 的 set 方法，具體實現先不考慮。
看到這兩個類，你就應該對 session 的工作有基本理解，可以從用戶訪問的流程來考慮。注意 BaseHandler 這個入口，每個用戶的訪問都是一次 HTTP 請求。當用戶第一次訪問或者上一次的 session 過期了，這時用戶訪問時 tornado 建立了一個 handler 對象（該 handler 一定繼承于 BaseHandler），并且在初始化時建立了一個 session 對象，因為是新訪問，所以目前 session 里面沒有數據，在之后采用 鍵/值 對的形式讀寫 session（不要忘了 Session 具有字典的所有操作），修改后通過 save 方法保存 session。如果用戶不是新訪問，那么也是按照上述的流程，不過 session 初始化時把 之前的數據取出來保存在該實例中。當用戶結束訪問，HTTP 斷開連接，handler 實例銷毀，session 實例銷毀（注意，是實例銷毀，不是數據銷毀）。
下面準備講 SessionManager 是吧，來～一個一個函數看～
首先是初始化，設置密鑰， memcache 地址，session 超時時間。
# 初始化需要一個用于 session 加密的 secret, memcache 地址, session 的過期時間 def __init__(self, secret, memcached_address, session_timeout): self.secret = secret self.memcached_address = memcached_address self.session_timeout = session_timeout
接著是 _fetch 方法，以 session_id 為鍵從 memcached 中取出數據，并用 pickle 反序列化解析數據：
# 該方法用 session_id 從 memcache 中取出數據 def _fetch(self, session_id): try: # 連接 memcache 服務器 mc = memcache.Client(self.memcached_address, debug=0) # 獲取數據 session_data = raw_data = mc.get(session_id) if raw_data != None: # 為了重新刷新 timeout mc.replace(session_id, raw_data, self.session_timeout, 0) # 反序列化 session_data = pickle.loads(raw_data) # 如果拿到的數據是字典形式，才進行返回 if type(session_data) == type({}): return session_data else: return {} except IOError: return {}
get 經過安全檢查后，以 SessionData 的形式返回 memcached 的數據（調用了 _fetch）方法。
def get(self, request_handler = None): # 獲取對應的 session_id 和 hmac_key if (request_handler == None): session_id = None hmac_key = None else: # session 的基礎還是靠 cookie session_id = request_handler.get_secure_cookie("session_id") hmac_key = request_handler.get_secure_cookie("verification") # session_id 不存在的時候則生成一個新的 session_id 和 hmac_key if session_id == None: session_exists = False session_id = self._generate_id() hmac_key = self._generate_hmac(session_id) else: session_exists = True # 檢查 hmac_key check_hmac = self._generate_hmac(session_id) # 不通過則拋出異常 if hmac_key != check_hmac: raise InvalidSessionException() # 新建 SessionData 對象 session = SessionData(session_id, hmac_key) if session_exists: # 通過 _fetch 方法獲取 memcache 中該 session 的所有數據 session_data = self._fetch(session_id) for key, data in session_data.iteritems(): session[key] = data return session
至于 set 方法，是為了更新 memcached 的數據。
# 設置新的 session,需要設置 handler 的 cookie 和 memcache 客戶端 def set(self, request_handler, session): # 設置瀏覽器的 cookie request_handler.set_secure_cookie("session_id", session.session_id) request_handler.set_secure_cookie("verification", session.hmac_key) # 用 pickle 進行序列化 session_data = pickle.dumps(dict(session.items()), pickle.HIGHEST_PROTOCOL) # 連接 memcache 服務器 mc = memcache.Client(self.memcached_address, debug=0) # 寫入 memcache mc.set(session.session_id, session_data, self.session_timeout, 0)
最后的兩個函數，一個是生成 session_id，另一個用 session_id 與密鑰加密后生成一個加密字符串，用于驗證。
# 生成 session_id def _generate_id(self): new_id = hashlib.sha256(self.secret + str(uuid.uuid4())) return new_id.hexdigest() # 生成 hmac_key def _generate_hmac(self, session_id): return hmac.new(session_id, self.secret, hashlib.sha256).hexdigest()
我們在哪里初始化了 SessionManager 呢？還記得第一篇里面的 Application 類嗎？噢...快回去翻翻。
新聞熱點
疑難解答
圖片精選
