免费注册 查看新帖 |

Chinaunix

  平台 论坛 博客 文库
最近访问板块 发新帖
查看: 5777 | 回复: 1

[MongoDB] 使用MongoDB来实现web.py的session存储 [复制链接]

论坛徽章:
0
发表于 2012-02-23 18:35 |显示全部楼层
使用MongoDB来实现web.py的session存储






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

  3. class MongoStore(Store):
  4.     def __init__(self, db, collection_name):
  5.         self.collection = db[collection_name]

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

  9.     def __getitem__(self, key):
  10.         now = time.time()
  11.         s = self.collection.find_one({'session_id':key})
  12.         if not s:
  13.             raise KeyError
  14.         else:
  15.             s.update({'attime':now})
  16.             return s

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

  19.         value['attime'] = now

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

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

  29.     def cleanup(self, timeout):
  30.         timeout = timeout/(24.0*60*60) #timedelta takes numdays as arg
  31.         last_allowed_time = time.time() - timeout
  32.         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


论坛徽章:
0
发表于 2012-02-24 17:09 |显示全部楼层
谢谢分享
您需要登录后才可以回帖 登录 | 注册

本版积分规则 发表回复

  

北京盛拓优讯信息技术有限公司. 版权所有 京ICP备16024965号-6 北京市公安局海淀分局网监中心备案编号:11010802020122 niuxiaotong@pcpop.com 17352615567
未成年举报专区
中国互联网协会会员  联系我们:huangweiwei@itpub.net
感谢所有关心和支持过ChinaUnix的朋友们 转载本站内容请注明原作者名及出处

清除 Cookies - ChinaUnix - Archiver - WAP - TOP