python---基于memcache的自定义session类

最新推荐文章于 2022-01-31 20:10:17 发布

转载最新推荐文章于 2022-01-31 20:10:17 发布 · 97 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/ssyfj/p/8605451.html

本文介绍了一种基于Memcache的会话管理实现方法。通过使用Python的memcache客户端进行会话数据的存储与读取，并自动生成MD5格式的会话ID确保安全性。文章详细展示了如何初始化会话、设置项、获取项以及删除项的过程。

import config
import hashlib
import time
import memcache
import json

conn = memcache.Client(["127.0.0.1:8081"],debug=True)

class BaseSession:
    @classmethod
    def get_session_handler(cls,handler):
        if config.SESSION_TYPE == "Memery":
            obj = MemerySession(handler)
        elif config.SESSION_TYPE == "Mamcache":
            obj = MemcacheSession(handler)
        elif config.SESSION_TYPE == "Redis":
            obj = RedisSession(handler)
        return obj


class MemcacheSession:
    session_id = "py_session"

    def __init__(self,handler):
        self.handler = handler
        #从客户端获取随机字符串
        client_random_str = self.handler.get_cookie(MemcacheSession.session_id)
        #如果获取到了字符串
        if client_random_str and conn.get(client_random_str):
            self.random_str = client_random_str
        else:
            self.random_str = self.__genarate_random_str()
            conn.set(self.random_str,json.dumps({}),config.SESSION_EXPIRE)
        conn.set(self.random_str,conn.get(self.random_str),config.SESSION_EXPIRE)
        expires_time = time.time() + config.SESSION_EXPIRE
        handler.set_cookie(MemerySession.prex_str, self.random_str, expires=expires_time)

    def __setitem__(self, key, value):
        ret = conn.get(self.random_str)
        ret_dict = json.loads(ret)
        ret_dict[key]=value
        conn.set(self.random_str, json.dumps(ret_dict), config.SESSION_EXPIRE)

    def __getitem__(self, item):
        ret = conn.get(self.random_str)
        ret_dict = json.loads(ret)
        return ret_dict.get(item,None)

    def __delitem__(self, key):
        ret = conn.get(self.random_str)
        ret_dict = json.loads(ret)
        del ret_dict[key]
        conn.set(self.random_str, json.dumps(ret_dict), config.SESSION_EXPIRE)

    def __genarate_random_str(self):
        obj = hashlib.md5()
        obj.update(bytes(str(time.time()),encoding="utf8"))
        random_str=obj.hexdigest()
        return random_str