feiyang10086 发表于 2012-02-23 18:35

使用MongoDB来实现web.py的session存储

使用MongoDB来实现web.py的session存储





web.py 是一个python的web开发框架,既然涉及到web开发,那么不可避免的可能使用到session,而传统的session存储基本都是默认在磁盘,当然考虑到web前端机的扩展我们经常需要重写session handler来自定义session的存储。下面就是在web.py框架中使用MongoDB作session存储的例子了。from web.session import Store
import time

class MongoStore(Store):
    def __init__(self, db, collection_name):
      self.collection = db

    def __contains__(self, key):
      data = self.collection.find_one({'session_id':key})
      return bool(data)

    def __getitem__(self, key):
      now = time.time()
      s = self.collection.find_one({'session_id':key})
      if not s:
            raise KeyError
      else:
            s.update({'attime':now})
            return s

    def __setitem__(self, key, value):
      now = time.time()

      value['attime'] = now

      s = self.collection.find_one({'session_id':key})
      if s:
            value = dict(map(lambda x: (str(x), x), [(k,v) for (k,v) in value.iteritems() if k not in ['_id']]))
            s.update(**value)
            self.collection.save(s)
      else:
            self.collection.insert(value)

    def __delitem__(self, key):
      self.collection.remove({'session_id':key})

    def cleanup(self, timeout):
      timeout = timeout/(24.0*60*60) #timedelta takes numdays as arg
      last_allowed_time = time.time() - timeout
      self.collection.remove({'attime' : { '$lt' : last_allowed_time}})然后在app中替换掉使用磁盘进行存储的申明,用

session = web.session.Session(app, MongoStore(db, 'sessions'))替换掉:

session = web.session.Session(app, web.session.DiskStore('sessions'))来源:python.cn


冰释一片天 发表于 2012-02-24 17:09

谢谢分享
页: [1]
查看完整版本: 使用MongoDB来实现web.py的session存储